xref: /llvm-project/mlir/test/Target/SPIRV/global-variable.mlir (revision bdf00e2216280edef1ec91ccc07987db92197b59) 
1// RUN: mlir-translate -no-implicit-module -test-spirv-roundtrip -split-input-file %s | FileCheck %s
2
3// CHECK:      spirv.GlobalVariable @var0 bind(1, 0) : !spirv.ptr<f32, Input>
4// CHECK-NEXT: spirv.GlobalVariable @var1 bind(0, 1) : !spirv.ptr<f32, Output>
5// CHECK-NEXT: spirv.GlobalVariable @var2 built_in("GlobalInvocationId") : !spirv.ptr<vector<3xi32>, Input>
6// CHECK-NEXT: spirv.GlobalVariable @var3 built_in("GlobalInvocationId") : !spirv.ptr<vector<3xi32>, Input>
7
8spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader], []> {
9  spirv.GlobalVariable @var0 bind(1, 0) : !spirv.ptr<f32, Input>
10  spirv.GlobalVariable @var1 bind(0, 1) : !spirv.ptr<f32, Output>
11  spirv.GlobalVariable @var2 {built_in = "GlobalInvocationId"} : !spirv.ptr<vector<3xi32>, Input>
12  spirv.GlobalVariable @var3 built_in("GlobalInvocationId") : !spirv.ptr<vector<3xi32>, Input>
13}
14
15// -----
16
17spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader], []> {
18  // CHECK:         spirv.GlobalVariable @var1 : !spirv.ptr<f32, Input>
19  // CHECK-NEXT:    spirv.GlobalVariable @var2 initializer(@var1) bind(1, 0) : !spirv.ptr<f32, Input>
20  spirv.GlobalVariable @var1 : !spirv.ptr<f32, Input>
21  spirv.GlobalVariable @var2 initializer(@var1) bind(1, 0) : !spirv.ptr<f32, Input>
22}
23
24// -----
25
26spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader], []> {
27  // CHECK:         spirv.SpecConstant @sc = 1 : i8
28  // CHECK-NEXT:    spirv.GlobalVariable @var initializer(@sc) : !spirv.ptr<i8, Uniform>
29  spirv.SpecConstant @sc = 1 : i8
30
31  spirv.GlobalVariable @var initializer(@sc) : !spirv.ptr<i8, Uniform>
32}
33
34// -----
35
36spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader], []> {
37  // CHECK:         spirv.SpecConstantComposite @scc (@sc0, @sc1, @sc2) : !spirv.array<3 x i8>
38  // CHECK-NEXT:    spirv.GlobalVariable @var initializer(@scc) : !spirv.ptr<!spirv.array<3 x i8>, Uniform>
39  spirv.SpecConstant @sc0 = 1 : i8
40  spirv.SpecConstant @sc1 = 2 : i8
41  spirv.SpecConstant @sc2 = 3 : i8
42
43  spirv.SpecConstantComposite @scc (@sc0, @sc1, @sc2) : !spirv.array<3 x i8>
44
45  spirv.GlobalVariable @var initializer(@scc) : !spirv.ptr<!spirv.array<3 x i8>, Uniform>
46}
47
48// -----
49
50spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader], []> {
51  spirv.GlobalVariable @globalInvocationID built_in("GlobalInvocationId") : !spirv.ptr<vector<3xi32>, Input>
52  spirv.func @foo() "None" {
53    // CHECK: %[[ADDR:.*]] = spirv.mlir.addressof @globalInvocationID : !spirv.ptr<vector<3xi32>, Input>
54    %0 = spirv.mlir.addressof @globalInvocationID : !spirv.ptr<vector<3xi32>, Input>
55    %1 = spirv.Constant 0: i32
56    // CHECK: spirv.AccessChain %[[ADDR]]
57    %2 = spirv.AccessChain %0[%1] : !spirv.ptr<vector<3xi32>, Input>, i32 -> !spirv.ptr<i32, Input>
58    spirv.Return
59  }
60}
61
62// -----
63
64spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader, Linkage], []> {
65  // CHECK: linkage_attributes = #spirv.linkage_attributes<linkage_name = "outSideGlobalVar1", linkage_type = <Import>>
66  spirv.GlobalVariable @var1 {
67    linkage_attributes=#spirv.linkage_attributes<
68      linkage_name="outSideGlobalVar1",
69      linkage_type=<Import>
70    >
71  } : !spirv.ptr<f32, Private>
72}
73