Skip to content

Commit 47645c1

Browse files
committed
amend! Fix gpu.set_default_device rewriting.
Fix gpu.set_default_device rewriting to properly use the new call. Smoke test included.
1 parent 08045e6 commit 47645c1

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed
Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
// RUN: mlir-opt %s --gpu-to-llvm | FileCheck %s
2+
3+
module attributes {gpu.container_module} {
4+
// CHECK-LABEL: func @set_default_device
5+
func.func @set_default_device(%arg0: i32) {
6+
// CHECK: mgpuSetDefaultDevice
7+
gpu.set_default_device %arg0
8+
return
9+
}
10+
}

0 commit comments

Comments
 (0)