forked from OSchip/llvm-project
27 lines
539 B
MLIR
27 lines
539 B
MLIR
// RUN: mlir-opt %s --test-kernel-to-cubin -split-input-file | FileCheck %s
|
|
|
|
// CHECK: attributes {nvvm.cubin = "CUBIN"}
|
|
gpu.module @foo {
|
|
llvm.func @kernel(%arg0 : !llvm.float, %arg1 : !llvm<"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
|
|
}
|
|
}
|