constant-op-to-llvm.mlir
2.01 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
// RUN: mlir-opt -convert-spirv-to-llvm %s | FileCheck %s
//===----------------------------------------------------------------------===//
// spv.constant
//===----------------------------------------------------------------------===//
func @bool_constant_scalar() {
// CHECK: {{.*}} = llvm.mlir.constant(true) : !llvm.i1
%0 = spv.constant true
// CHECK: {{.*}} = llvm.mlir.constant(false) : !llvm.i1
%1 = spv.constant false
return
}
func @bool_constant_vector() {
// CHECK: {{.*}} = llvm.mlir.constant(dense<[true, false]> : vector<2xi1>) : !llvm<"<2 x i1>">
%0 = constant dense<[true, false]> : vector<2xi1>
// CHECK: {{.*}} = llvm.mlir.constant(dense<false> : vector<3xi1>) : !llvm<"<3 x i1>">
%1 = constant dense<false> : vector<3xi1>
return
}
func @integer_constant_scalar() {
// CHECK: {{.*}} = llvm.mlir.constant(0 : i8) : !llvm.i8
%0 = spv.constant 0 : i8
// CHECK: {{.*}} = llvm.mlir.constant(-5 : i64) : !llvm.i64
%1 = spv.constant -5 : si64
// CHECK: {{.*}} = llvm.mlir.constant(10 : i16) : !llvm.i16
%2 = spv.constant 10 : ui16
return
}
func @integer_constant_vector() {
// CHECK: {{.*}} = llvm.mlir.constant(dense<[2, 3]> : vector<2xi32>) : !llvm<"<2 x i32>">
%0 = spv.constant dense<[2, 3]> : vector<2xi32>
// CHECK: {{.*}} = llvm.mlir.constant(dense<-4> : vector<2xi32>) : !llvm<"<2 x i32>">
%1 = spv.constant dense<-4> : vector<2xsi32>
// CHECK: {{.*}} = llvm.mlir.constant(dense<[2, 3, 4]> : vector<3xi32>) : !llvm<"<3 x i32>">
%2 = spv.constant dense<[2, 3, 4]> : vector<3xui32>
return
}
func @float_constant_scalar() {
// CHECK: {{.*}} = llvm.mlir.constant(5.000000e+00 : f16) : !llvm.half
%0 = spv.constant 5.000000e+00 : f16
// CHECK: {{.*}} = llvm.mlir.constant(5.000000e+00 : f64) : !llvm.double
%1 = spv.constant 5.000000e+00 : f64
return
}
func @float_constant_vector() {
// CHECK: {{.*}} = llvm.mlir.constant(dense<[2.000000e+00, 3.000000e+00]> : vector<2xf32>) : !llvm<"<2 x float>">
%0 = spv.constant dense<[2.000000e+00, 3.000000e+00]> : vector<2xf32>
return
}