diff options
author | George Karpenkov <cheshire@google.com> | 2019-09-19 14:33:54 -0700 |
---|---|---|
committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-09-19 14:34:30 -0700 |
commit | 2df646bef6e7665fdb8523613d82e7d4a5013217 (patch) | |
tree | 3bcbbee989db665c23fa9d9fef9131cfaf549e85 /mlir/test/Conversion/GPUToCUDA | |
parent | c8961d408ee0f48e3156e3c4248bba9a43579f1f (diff) | |
download | bcm5719-llvm-2df646bef6e7665fdb8523613d82e7d4a5013217.tar.gz bcm5719-llvm-2df646bef6e7665fdb8523613d82e7d4a5013217.zip |
Automated rollback of commit 5684a12434f923d03b6870f2aa16226bfb0b38b6
PiperOrigin-RevId: 270126672
Diffstat (limited to 'mlir/test/Conversion/GPUToCUDA')
-rw-r--r-- | mlir/test/Conversion/GPUToCUDA/insert-cubin-getter.mlir | 9 | ||||
-rw-r--r-- | mlir/test/Conversion/GPUToCUDA/lower-nvvm-kernel-to-cubin.mlir | 34 |
2 files changed, 10 insertions, 33 deletions
diff --git a/mlir/test/Conversion/GPUToCUDA/insert-cubin-getter.mlir b/mlir/test/Conversion/GPUToCUDA/insert-cubin-getter.mlir index 9e0907f7477..d2e291f57e7 100644 --- a/mlir/test/Conversion/GPUToCUDA/insert-cubin-getter.mlir +++ b/mlir/test/Conversion/GPUToCUDA/insert-cubin-getter.mlir @@ -2,14 +2,9 @@ // CHECK: llvm.mlir.global constant @[[global:.*]]("CUBIN") -module attributes {gpu.kernel_module} { - func @kernel(!llvm.float, !llvm<"float*">) - attributes {nvvm.cubin = "CUBIN"} -} - func @kernel(!llvm.float, !llvm<"float*">) -// CHECK: attributes {gpu.kernel, nvvm.cubingetter = @[[getter:.*]]} - attributes {gpu.kernel} +// CHECK: attributes {gpu.kernel, nvvm.cubin = "CUBIN", nvvm.cubingetter = @[[getter:.*]]} + attributes {gpu.kernel, nvvm.cubin = "CUBIN"} // CHECK: func @[[getter]]() -> !llvm<"i8*"> // CHECK: %[[addressof:.*]] = llvm.mlir.addressof @[[global]] diff --git a/mlir/test/Conversion/GPUToCUDA/lower-nvvm-kernel-to-cubin.mlir b/mlir/test/Conversion/GPUToCUDA/lower-nvvm-kernel-to-cubin.mlir index b6e19989203..8ddfc1996ef 100644 --- a/mlir/test/Conversion/GPUToCUDA/lower-nvvm-kernel-to-cubin.mlir +++ b/mlir/test/Conversion/GPUToCUDA/lower-nvvm-kernel-to-cubin.mlir @@ -1,26 +1,8 @@ -// RUN: mlir-opt %s --test-kernel-to-cubin -split-input-file | FileCheck %s - -module attributes {gpu.kernel_module} { - func @kernel(%arg0 : !llvm.float, %arg1 : !llvm<"float*">) - // CHECK: attributes {gpu.kernel, nvvm.cubin = "CUBIN"} - attributes { gpu.kernel } { - // CHECK-NOT: llvm.return - llvm.return - } -} - -// ----- - -module attributes {gpu.kernel_module} { - // CHECK: func @kernel_a - func @kernel_a() - attributes { gpu.kernel } { - llvm.return - } - - // CHECK: func @kernel_b - func @kernel_b() - attributes { gpu.kernel } { - llvm.return - } -} +// RUN: mlir-opt %s --test-kernel-to-cubin | FileCheck %s + +func @kernel(%arg0 : !llvm.float, %arg1 : !llvm<"float*">) +// CHECK: attributes {gpu.kernel, nvvm.cubin = "CUBIN"} + attributes { gpu.kernel } { +// CHECK-NOT: llvm.return + llvm.return +}
\ No newline at end of file |