lower-rocdl-kernel-to-hsaco.mlir
541 Bytes
// RUN: mlir-opt %s --test-kernel-to-hsaco -split-input-file | FileCheck %s
// CHECK: attributes {rocdl.hsaco = "HSACO"}
gpu.module @foo {
llvm.func @kernel(%arg0 : !llvm.float, %arg1 : !llvm.ptr<float>)
// CHECK: attributes {gpu.kernel}
attributes { gpu.kernel } {
llvm.return
}
}
// -----
gpu.module @bar {
// CHECK: func @kernel_a
llvm.func @kernel_a()
attributes { gpu.kernel } {
llvm.return
}
// CHECK: func @kernel_b
llvm.func @kernel_b()
attributes { gpu.kernel } {
llvm.return
}
}