200 lines · plain
1// RUN: mlir-opt -split-input-file -allow-unregistered-dialect -map-memref-spirv-storage-class='client-api=vulkan' -verify-diagnostics %s -o - | FileCheck %s --check-prefix=VULKAN2// RUN: mlir-opt -split-input-file -allow-unregistered-dialect -map-memref-spirv-storage-class='client-api=opencl' -verify-diagnostics %s -o - | FileCheck %s --check-prefix=OPENCL3// RUN: mlir-opt -split-input-file -allow-unregistered-dialect -map-memref-spirv-storage-class -verify-diagnostics %s -o - | FileCheck %s4 5// Vulkan Mappings:6// 0 -> StorageBuffer7// 1 -> Generic8// 2 -> [null]9// 3 -> Workgroup10// 4 -> Uniform11// 5 -> Private12// 6 -> Function13// 7 -> PushConstant14// 8 -> UniformConstant15// 9 -> Input16// 10 -> Output17// 11 -> PhysicalStorageBuffer18// 12 -> Image19 20// OpenCL Mappings:21// 0 -> CrossWorkgroup22// 1 -> Generic23// 2 -> [null]24// 3 -> Workgroup25// 4 -> UniformConstant26// 5 -> Private27// 6 -> Function28// 7 -> Image29 30// VULKAN-LABEL: func @operand_result31// OPENCL-LABEL: func @operand_result32func.func @operand_result() {33 // VULKAN: memref<f32, #spirv.storage_class<StorageBuffer>>34 // OPENCL: memref<f32, #spirv.storage_class<CrossWorkgroup>>35 %0 = "dialect.memref_producer"() : () -> (memref<f32>)36 // VULKAN: memref<4xi32, #spirv.storage_class<Generic>>37 // OPENCL: memref<4xi32, #spirv.storage_class<Generic>>38 %1 = "dialect.memref_producer"() : () -> (memref<4xi32, 1>)39 // VULKAN: memref<?x4xf16, #spirv.storage_class<Workgroup>>40 // OPENCL: memref<?x4xf16, #spirv.storage_class<Workgroup>>41 %2 = "dialect.memref_producer"() : () -> (memref<?x4xf16, 3>)42 // VULKAN: memref<*xf16, #spirv.storage_class<Uniform>>43 // OPENCL: memref<*xf16, #spirv.storage_class<UniformConstant>>44 %3 = "dialect.memref_producer"() : () -> (memref<*xf16, 4>)45 // VULKAN: memref<*xf16, #spirv.storage_class<Private>>46 // OPENCL: memref<*xf16, #spirv.storage_class<Private>>47 %4 = "dialect.memref_producer"() : () -> (memref<*xf16, 5>)48 // VULKAN: memref<*xf16, #spirv.storage_class<Function>>49 // OPENCL: memref<*xf16, #spirv.storage_class<Function>>50 %5 = "dialect.memref_producer"() : () -> (memref<*xf16, 6>)51 // VULKAN: memref<*xf16, #spirv.storage_class<PushConstant>>52 // OPENCL: memref<*xf16, #spirv.storage_class<Image>>53 %6 = "dialect.memref_producer"() : () -> (memref<*xf16, 7>)54 55 56 "dialect.memref_consumer"(%0) : (memref<f32>) -> ()57 // VULKAN: memref<4xi32, #spirv.storage_class<Generic>>58 // OPENCL: memref<4xi32, #spirv.storage_class<Generic>>59 "dialect.memref_consumer"(%1) : (memref<4xi32, 1>) -> ()60 // VULKAN: memref<?x4xf16, #spirv.storage_class<Workgroup>>61 // OPENCL: memref<?x4xf16, #spirv.storage_class<Workgroup>>62 "dialect.memref_consumer"(%2) : (memref<?x4xf16, 3>) -> ()63 // VULKAN: memref<*xf16, #spirv.storage_class<Uniform>>64 // OPENCL: memref<*xf16, #spirv.storage_class<UniformConstant>>65 "dialect.memref_consumer"(%3) : (memref<*xf16, 4>) -> ()66 // VULKAN: memref<*xf16, #spirv.storage_class<Private>>67 // OPENCL: memref<*xf16, #spirv.storage_class<Private>>68 "dialect.memref_consumer"(%4) : (memref<*xf16, 5>) -> ()69 // VULKAN: memref<*xf16, #spirv.storage_class<Function>>70 // OPENCL: memref<*xf16, #spirv.storage_class<Function>>71 "dialect.memref_consumer"(%5) : (memref<*xf16, 6>) -> ()72 // VULKAN: memref<*xf16, #spirv.storage_class<PushConstant>>73 // OPENCL: memref<*xf16, #spirv.storage_class<Image>>74 "dialect.memref_consumer"(%6) : (memref<*xf16, 7>) -> ()75 76 return77}78 79// -----80 81// VULKAN-LABEL: func @type_attribute82// OPENCL-LABEL: func @type_attribute83func.func @type_attribute() {84 // VULKAN: attr = memref<i32, #spirv.storage_class<Generic>>85 // OPENCL: attr = memref<i32, #spirv.storage_class<Generic>>86 "dialect.memref_producer"() { attr = memref<i32, 1> } : () -> ()87 return88}89 90// -----91 92// VULKAN-LABEL: func.func @function_io93// OPENCL-LABEL: func.func @function_io94func.func @function_io95 // VULKAN-SAME: (%{{.+}}: memref<f64, #spirv.storage_class<Generic>>, %{{.+}}: memref<4xi32, #spirv.storage_class<Workgroup>>)96 // OPENCL-SAME: (%{{.+}}: memref<f64, #spirv.storage_class<Generic>>, %{{.+}}: memref<4xi32, #spirv.storage_class<Workgroup>>)97 (%arg0: memref<f64, 1>, %arg1: memref<4xi32, 3>)98 // VULKAN-SAME: -> (memref<f64, #spirv.storage_class<Generic>>, memref<4xi32, #spirv.storage_class<Workgroup>>)99 // OPENCL-SAME: -> (memref<f64, #spirv.storage_class<Generic>>, memref<4xi32, #spirv.storage_class<Workgroup>>)100 -> (memref<f64, 1>, memref<4xi32, 3>) {101 return %arg0, %arg1: memref<f64, 1>, memref<4xi32, 3>102}103 104// -----105 106gpu.module @kernel {107// VULKAN-LABEL: gpu.func @function_io108// OPENCL-LABEL: gpu.func @function_io109// VULKAN-SAME: memref<8xi32, #spirv.storage_class<StorageBuffer>>110// OPENCL-SAME: memref<8xi32, #spirv.storage_class<CrossWorkgroup>>111gpu.func @function_io(%arg0 : memref<8xi32>) kernel { gpu.return }112}113 114// -----115 116// VULKAN-LABEL: func.func @region117// OPENCL-LABEL: func.func @region118func.func @region(%cond: i1, %arg0: memref<f32, 1>) {119 scf.if %cond {120 // VULKAN: "dialect.memref_consumer"(%{{.+}}) {attr = memref<i64, #spirv.storage_class<Workgroup>>}121 // OPENCL: "dialect.memref_consumer"(%{{.+}}) {attr = memref<i64, #spirv.storage_class<Workgroup>>}122 // VULKAN-SAME: (memref<f32, #spirv.storage_class<Generic>>) -> memref<f32, #spirv.storage_class<Generic>>123 // OPENCL-SAME: (memref<f32, #spirv.storage_class<Generic>>) -> memref<f32, #spirv.storage_class<Generic>>124 %0 = "dialect.memref_consumer"(%arg0) { attr = memref<i64, 3> } : (memref<f32, 1>) -> (memref<f32, 1>)125 }126 return127}128 129// -----130 131// VULKAN-LABEL: func @non_memref_types132// OPENCL-LABEL: func @non_memref_types133func.func @non_memref_types(%arg: f32) -> f32 {134 // VULKAN: "dialect.op"(%{{.+}}) {attr = 16 : i64} : (f32) -> f32135 // OPENCL: "dialect.op"(%{{.+}}) {attr = 16 : i64} : (f32) -> f32136 %0 = "dialect.op"(%arg) { attr = 16 } : (f32) -> (f32)137 return %0 : f32138}139 140// -----141 142func.func @missing_mapping() {143 // expected-error @+1 {{failed to legalize}}144 %0 = "dialect.memref_producer"() : () -> (memref<f32, 2>)145 return146}147 148// -----149 150/// Checks memory maps to OpenCL mapping if Kernel capability is enabled.151module attributes { spirv.target_env = #spirv.target_env<#spirv.vce<v1.0, [Kernel], []>, #spirv.resource_limits<>> } {152func.func @operand_result() {153 // CHECK: memref<f32, #spirv.storage_class<CrossWorkgroup>>154 %0 = "dialect.memref_producer"() : () -> (memref<f32>)155 // CHECK: memref<4xi32, #spirv.storage_class<Generic>>156 %1 = "dialect.memref_producer"() : () -> (memref<4xi32, 1>)157 // CHECK: memref<?x4xf16, #spirv.storage_class<Workgroup>>158 %2 = "dialect.memref_producer"() : () -> (memref<?x4xf16, 3>)159 // CHECK: memref<*xf16, #spirv.storage_class<UniformConstant>>160 %3 = "dialect.memref_producer"() : () -> (memref<*xf16, 4>)161 162 163 "dialect.memref_consumer"(%0) : (memref<f32>) -> ()164 // CHECK: memref<4xi32, #spirv.storage_class<Generic>>165 "dialect.memref_consumer"(%1) : (memref<4xi32, 1>) -> ()166 // CHECK: memref<?x4xf16, #spirv.storage_class<Workgroup>>167 "dialect.memref_consumer"(%2) : (memref<?x4xf16, 3>) -> ()168 // CHECK: memref<*xf16, #spirv.storage_class<UniformConstant>>169 "dialect.memref_consumer"(%3) : (memref<*xf16, 4>) -> ()170 171 return172}173}174 175// -----176 177/// Checks memory maps to Vulkan mapping if Shader capability is enabled.178module attributes { spirv.target_env = #spirv.target_env<#spirv.vce<v1.0, [Shader], []>, #spirv.resource_limits<>> } {179func.func @operand_result() {180 // CHECK: memref<f32, #spirv.storage_class<StorageBuffer>>181 %0 = "dialect.memref_producer"() : () -> (memref<f32>)182 // CHECK: memref<4xi32, #spirv.storage_class<Generic>>183 %1 = "dialect.memref_producer"() : () -> (memref<4xi32, 1>)184 // CHECK: memref<?x4xf16, #spirv.storage_class<Workgroup>>185 %2 = "dialect.memref_producer"() : () -> (memref<?x4xf16, 3>)186 // CHECK: memref<*xf16, #spirv.storage_class<Uniform>>187 %3 = "dialect.memref_producer"() : () -> (memref<*xf16, 4>)188 189 190 "dialect.memref_consumer"(%0) : (memref<f32>) -> ()191 // CHECK: memref<4xi32, #spirv.storage_class<Generic>>192 "dialect.memref_consumer"(%1) : (memref<4xi32, 1>) -> ()193 // CHECK: memref<?x4xf16, #spirv.storage_class<Workgroup>>194 "dialect.memref_consumer"(%2) : (memref<?x4xf16, 3>) -> ()195 // CHECK: memref<*xf16, #spirv.storage_class<Uniform>>196 "dialect.memref_consumer"(%3) : (memref<*xf16, 4>) -> ()197 return198}199}200