1// RUN: mlir-opt %s --gpu-to-llvm | FileCheck %s 2 3module 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} 11