• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1// RUN: mlir-opt %s --test-kernel-to-hsaco -split-input-file | FileCheck %s
2
3// CHECK: attributes {rocdl.hsaco = "HSACO"}
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