forked from OSchip/llvm-project
27 lines
541 B
MLIR
27 lines
541 B
MLIR
// 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
|
|
}
|
|
}
|