// RUN: mlir-opt %s \ // RUN: | mlir-opt -gpu-lower-to-nvvm-pipeline="cubin-format=%gpu_compilation_format" \ // RUN: | mlir-cpu-runner \ // RUN: --shared-libs=%mlir_cuda_runtime \ // RUN: --shared-libs=%mlir_runner_utils \ // RUN: --entry-point-result=void \ // RUN: | FileCheck %s // CHECK: Hello from 0, 2, 3.000000 // CHECK: Hello from 1, 2, 3.000000 module attributes {gpu.container_module} { gpu.module @kernels { gpu.func @hello() kernel { %0 = gpu.thread_id x %csti8 = arith.constant 2 : i8 %cstf32 = arith.constant 3.0 : f32 gpu.printf "Hello from %lld, %d, %f\n" %0, %csti8, %cstf32 : index, i8, f32 gpu.return } } func.func @main() { %c2 = arith.constant 2 : index %c1 = arith.constant 1 : index gpu.launch_func @kernels::@hello blocks in (%c1, %c1, %c1) threads in (%c2, %c1, %c1) return } }