1// RUN: mlir-opt %s --test-kernel-to-cubin -split-input-file | FileCheck %s 2 3// CHECK: attributes {nvvm.cubin = "CUBIN"} 4gpu.module @foo { 5 llvm.func @kernel(%arg0 : !llvm.float, %arg1 : !llvm.ptr<float>) 6 // CHECK: attributes {gpu.kernel} 7 attributes { gpu.kernel } { 8 llvm.return 9 } 10} 11 12// ----- 13 14gpu.module @bar { 15 // CHECK: func @kernel_a 16 llvm.func @kernel_a() 17 attributes { gpu.kernel } { 18 llvm.return 19 } 20 21 // CHECK: func @kernel_b 22 llvm.func @kernel_b() 23 attributes { gpu.kernel } { 24 llvm.return 25 } 26} 27