You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
27 lines
540 B
27 lines
540 B
7 months ago
|
// 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.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
|
||
|
|
}
|
||
|
|
}
|