[mlir][py] Enable loading only specified dialects during creation. (#121421)
[llvm-project.git] / mlir / test / Target / SPIRV / undef.mlir
blobb9044fe8b40af05438db15ea2f0d08dde4392068
1 // RUN: mlir-translate -no-implicit-module -split-input-file -test-spirv-roundtrip %s | FileCheck %s
3 spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader], []> {
4   spirv.func @foo() -> () "None" {
5     // CHECK: {{%.*}} = spirv.Undef : f32
6     // CHECK-NEXT: {{%.*}} = spirv.Undef : f32
7     %0 = spirv.Undef : f32
8     %1 = spirv.Undef : f32
9     %2 = spirv.FAdd %0, %1 : f32
10     // CHECK: {{%.*}} = spirv.Undef : vector<4xi32>
11     %3 = spirv.Undef : vector<4xi32>
12     %4 = spirv.CompositeExtract %3[1 : i32] : vector<4xi32>
13     // CHECK: {{%.*}} = spirv.Undef : !spirv.array<4 x !spirv.array<4 x i32>>
14     %5 = spirv.Undef : !spirv.array<4x!spirv.array<4xi32>>
15     %6 = spirv.CompositeExtract %5[1 : i32, 2 : i32] : !spirv.array<4x!spirv.array<4xi32>>
16     // CHECK: {{%.*}} = spirv.Undef : !spirv.ptr<!spirv.struct<(f32)>, StorageBuffer>
17     %7 = spirv.Undef : !spirv.ptr<!spirv.struct<(f32)>, StorageBuffer>
18     %8 = spirv.Constant 0 : i32
19     %9 = spirv.AccessChain %7[%8] : !spirv.ptr<!spirv.struct<(f32)>, StorageBuffer>, i32 -> !spirv.ptr<f32, StorageBuffer>
20     spirv.Return
21   }
24 // -----
26 spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader], []> {
27   // CHECK: spirv.func {{@.*}}
28   spirv.func @ignore_unused_undef() -> () "None" {
29     // CHECK-NEXT: spirv.Return
30     %0 = spirv.Undef : f32
31     spirv.Return
32   }