blob: 5a7178030843cd7e750d89456334249f7ae8e723 (
plain)
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
|
// RUN: mlir-opt %s | FileCheck %s
func @rocdl_special_regs() -> !llvm.i32 {
// CHECK-LABEL: rocdl_special_regs
// CHECK: rocdl.workitem.id.x : !llvm.i32
%0 = rocdl.workitem.id.x : !llvm.i32
// CHECK: rocdl.workitem.id.y : !llvm.i32
%1 = rocdl.workitem.id.y : !llvm.i32
// CHECK: rocdl.workitem.id.z : !llvm.i32
%2 = rocdl.workitem.id.z : !llvm.i32
// CHECK: rocdl.workgroup.id.x : !llvm.i32
%3 = rocdl.workgroup.id.x : !llvm.i32
// CHECK: rocdl.workgroup.id.y : !llvm.i32
%4 = rocdl.workgroup.id.y : !llvm.i32
// CHECK: rocdl.workgroup.id.z : !llvm.i32
%5 = rocdl.workgroup.id.z : !llvm.i32
// CHECK: rocdl.workgroup.dim.x : !llvm.i32
%6 = rocdl.workgroup.dim.x : !llvm.i32
// CHECK: rocdl.workgroup.dim.y : !llvm.i32
%7 = rocdl.workgroup.dim.y : !llvm.i32
// CHECK: rocdl.workgroup.dim.z : !llvm.i32
%8 = rocdl.workgroup.dim.z : !llvm.i32
// CHECK: rocdl.grid.dim.x : !llvm.i32
%9 = rocdl.grid.dim.x : !llvm.i32
// CHECK: rocdl.grid.dim.y : !llvm.i32
%10 = rocdl.grid.dim.y : !llvm.i32
// CHECK: rocdl.grid.dim.z : !llvm.i32
%11 = rocdl.grid.dim.z : !llvm.i32
llvm.return %0 : !llvm.i32
}
|
