mlir/test/Conversion/GPUToCUDA/lower-nvvm-kernel-to-cubin.mlir


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

// RUN: mlir-opt %s --test-kernel-to-cubin -split-input-file | FileCheck %s

// CHECK: attributes {gpu.kernel_module, nvvm.cubin = "CUBIN"}
module @foo attributes {gpu.kernel_module} {
  llvm.func @kernel(%arg0 : !llvm.float, %arg1 : !llvm<"float*">)
    // CHECK: attributes  {gpu.kernel}
    attributes  { gpu.kernel } {
    llvm.return
  }
}

// -----

module @bar attributes {gpu.kernel_module} {
  // 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
  }
}