11 lines · plain
1// RUN: mlir-opt %s --gpu-to-llvm | FileCheck %s2 3module attributes {gpu.container_module} {4 // CHECK-LABEL: func @set_default_device5 func.func @set_default_device(%arg0: i32) {6 // CHECK: mgpuSetDefaultDevice7 gpu.set_default_device %arg08 return9 }10}11