115 lines · plain
1// RUN: mlir-opt --split-input-file --spirv-lower-abi-attrs --verify-diagnostics %s \2// RUN: | FileCheck %s3 4module attributes {5 spirv.target_env = #spirv.target_env<6 #spirv.vce<v1.0, [Shader], [SPV_KHR_storage_buffer_storage_class]>, #spirv.resource_limits<>>7} {8 9// CHECK-LABEL: spirv.module10spirv.module Logical GLSL450 {11 // CHECK-DAG: spirv.GlobalVariable [[VAR0:@.*]] bind(0, 0) : !spirv.ptr<!spirv.struct<(f32 [0])>, StorageBuffer>12 // CHECK-DAG: spirv.GlobalVariable [[VAR1:@.*]] bind(0, 1) : !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32, stride=4> [0])>, StorageBuffer>13 // CHECK: spirv.func [[FN:@.*]]()14 // We cannot generate SubgroupSize execution mode for Shader capability -- leave it alone.15 // CHECK-SAME: #spirv.entry_point_abi<subgroup_size = 64>16 spirv.func @kernel(17 %arg0: f3218 {spirv.interface_var_abi = #spirv.interface_var_abi<(0, 0), StorageBuffer>},19 %arg1: !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32>)>, StorageBuffer>20 {spirv.interface_var_abi = #spirv.interface_var_abi<(0, 1)>}) "None"21 attributes {spirv.entry_point_abi = #spirv.entry_point_abi<workgroup_size = [32, 1, 1], subgroup_size = 64>} {22 // CHECK: [[ADDRESSARG0:%.*]] = spirv.mlir.addressof [[VAR0]]23 // CHECK: [[CONST0:%.*]] = spirv.Constant 0 : i3224 // CHECK: [[ARG0PTR:%.*]] = spirv.AccessChain [[ADDRESSARG0]]{{\[}}[[CONST0]]25 // CHECK: [[ARG0:%.*]] = spirv.Load "StorageBuffer" [[ARG0PTR]]26 // CHECK: [[ARG1:%.*]] = spirv.mlir.addressof [[VAR1]]27 // CHECK: spirv.Return28 spirv.Return29 }30 // CHECK: spirv.EntryPoint "GLCompute" [[FN]]31 // CHECK: spirv.ExecutionMode [[FN]] "LocalSize", 32, 1, 132} // end spirv.module33 34} // end module35 36// -----37 38module attributes {39 spirv.target_env = #spirv.target_env<40 #spirv.vce<v1.0, [VulkanMemoryModel, Shader, Int8, TensorsARM, GraphARM], [SPV_ARM_tensors, SPV_ARM_graph, SPV_KHR_vulkan_memory_model]>, #spirv.resource_limits<>>41} {42 43// CHECK-LABEL: spirv.module44spirv.module Logical Vulkan {45 // CHECK-DAG: spirv.GlobalVariable [[VARARG0:@.*]] bind(0, 0) : !spirv.ptr<!spirv.arm.tensor<1x16x16x16xi8>, UniformConstant>46 // CHECK-DAG: spirv.GlobalVariable [[VARRES0:@.*]] bind(0, 1) : !spirv.ptr<!spirv.arm.tensor<1x16x16x16xi8>, UniformConstant>47 48 // CHECK: spirv.ARM.GraphEntryPoint [[GN:@.*]], [[VARARG0]], [[VARRES0]]49 // CHECK: spirv.ARM.Graph [[GN]]([[ARG0:%.*]]: !spirv.arm.tensor<1x16x16x16xi8>) -> !spirv.arm.tensor<1x16x16x16xi8> attributes {entry_point = true}50 spirv.ARM.Graph @main(%arg0: !spirv.arm.tensor<1x16x16x16xi8> {spirv.interface_var_abi = #spirv.interface_var_abi<(0, 0)>})51 -> (!spirv.arm.tensor<1x16x16x16xi8> {spirv.interface_var_abi = #spirv.interface_var_abi<(0, 1)>}) attributes {entry_point = true} {52 spirv.ARM.GraphOutputs %arg0 : !spirv.arm.tensor<1x16x16x16xi8>53 }54} // end spirv.module55 56} // end module57 58// -----59 60module {61// expected-error@+1 {{'spirv.module' op missing SPIR-V target env attribute}}62spirv.module Logical GLSL450 {}63} // end module64 65// -----66 67// CHECK-LABEL: spirv.module68// Test case with SPIRV version 1.4: all the interface's storage variables are passed to OpEntryPoint69spirv.module Logical GLSL450 attributes {spirv.target_env = #spirv.target_env<#spirv.vce<v1.4, [Shader], [SPV_KHR_storage_buffer_storage_class]>, #spirv.resource_limits<>>} {70 // CHECK-DAG: spirv.GlobalVariable [[VAR0:@.*]] bind(0, 0) : !spirv.ptr<!spirv.struct<(f32 [0])>, StorageBuffer>71 // CHECK-DAG: spirv.GlobalVariable [[VAR1:@.*]] bind(0, 1) : !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32, stride=4> [0])>, StorageBuffer>72 // CHECK: spirv.func [[FN:@.*]]()73 // CHECK-SAME: #spirv.entry_point_abi<subgroup_size = 64>74 spirv.func @kernel(75 %arg0: f3276 {spirv.interface_var_abi = #spirv.interface_var_abi<(0, 0), StorageBuffer>},77 %arg1: !spirv.ptr<!spirv.struct<(!spirv.array<12 x f32>)>, StorageBuffer>78 {spirv.interface_var_abi = #spirv.interface_var_abi<(0, 1)>}) "None"79 attributes {spirv.entry_point_abi = #spirv.entry_point_abi<workgroup_size = [32, 1, 1], subgroup_size = 64>} {80 // CHECK: [[ADDRESSARG0:%.*]] = spirv.mlir.addressof [[VAR0]]81 // CHECK: [[CONST0:%.*]] = spirv.Constant 0 : i3282 // CHECK: [[ARG0PTR:%.*]] = spirv.AccessChain [[ADDRESSARG0]]{{\[}}[[CONST0]]83 // CHECK: [[ARG0:%.*]] = spirv.Load "StorageBuffer" [[ARG0PTR]]84 // CHECK: [[ARG1:%.*]] = spirv.mlir.addressof [[VAR1]]85 // CHECK: spirv.Return86 spirv.Return87 }88 // CHECK: spirv.EntryPoint "GLCompute" [[FN]], [[VAR0]], [[VAR1]]89 // CHECK: spirv.ExecutionMode [[FN]] "LocalSize", 32, 1, 190} // end spirv.module91 92// -----93 94module {95 spirv.module Logical GLSL450 attributes {spirv.target_env = #spirv.target_env<#spirv.vce<v1.6, [Shader, Sampled1D], []>, #spirv.resource_limits<>>} {96 // CHECK-DAG: spirv.GlobalVariable @[[IMAGE_GV:.*]] bind(0, 0) : !spirv.ptr<!spirv.sampled_image<!spirv.image<f32, Dim1D, DepthUnknown, NonArrayed, SingleSampled, NeedSampler, R32f>>, UniformConstant>97 // CHECK: spirv.func @read_image98 spirv.func @read_image(%arg0: !spirv.ptr<!spirv.sampled_image<!spirv.image<f32, Dim1D, DepthUnknown, NonArrayed, SingleSampled, NeedSampler, R32f>>, UniformConstant> {spirv.interface_var_abi = #spirv.interface_var_abi<(0, 0)>}, %arg1: !spirv.ptr<!spirv.struct<(!spirv.array<1 x f32, stride=4> [0])>, StorageBuffer> {spirv.interface_var_abi = #spirv.interface_var_abi<(0, 1)>}) "None" attributes {spirv.entry_point_abi = #spirv.entry_point_abi<workgroup_size = [1, 1, 1]>} {99 // CHECK: %[[IMAGE_ADDR:.*]] = spirv.mlir.addressof @[[IMAGE_GV]] : !spirv.ptr<!spirv.sampled_image<!spirv.image<f32, Dim1D, DepthUnknown, NonArrayed, SingleSampled, NeedSampler, R32f>>, UniformConstant>100 %cst0_i32 = spirv.Constant 0 : i32101 // CHECK: spirv.Load "UniformConstant" %[[IMAGE_ADDR]]102 %0 = spirv.Load "UniformConstant" %arg0 : !spirv.sampled_image<!spirv.image<f32, Dim1D, DepthUnknown, NonArrayed, SingleSampled, NeedSampler, R32f>>103 %1 = spirv.Image %0 : !spirv.sampled_image<!spirv.image<f32, Dim1D, DepthUnknown, NonArrayed, SingleSampled, NeedSampler, R32f>>104 %2 = spirv.ImageFetch %1, %cst0_i32 : !spirv.image<f32, Dim1D, DepthUnknown, NonArrayed, SingleSampled, NeedSampler, R32f>, i32 -> vector<4xf32>105 %3 = spirv.CompositeExtract %2[0 : i32] : vector<4xf32>106 %cst0_i32_0 = spirv.Constant 0 : i32107 %cst0_i32_1 = spirv.Constant 0 : i32108 %cst1_i32 = spirv.Constant 1 : i32109 %4 = spirv.AccessChain %arg1[%cst0_i32_0, %cst0_i32] : !spirv.ptr<!spirv.struct<(!spirv.array<1 x f32, stride=4> [0])>, StorageBuffer>, i32, i32 -> !spirv.ptr<f32, StorageBuffer>110 spirv.Store "StorageBuffer" %4, %3 : f32111 spirv.Return112 }113 }114}115