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/GPUToNVVM | |
| 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/GPUToNVVM')
| -rw-r--r-- | mlir/test/Conversion/GPUToNVVM/gpu-to-nvvm.mlir | 58 |
1 files changed, 28 insertions, 30 deletions
diff --git a/mlir/test/Conversion/GPUToNVVM/gpu-to-nvvm.mlir b/mlir/test/Conversion/GPUToNVVM/gpu-to-nvvm.mlir index 02637376622..cf8e7ed1113 100644 --- a/mlir/test/Conversion/GPUToNVVM/gpu-to-nvvm.mlir +++ b/mlir/test/Conversion/GPUToNVVM/gpu-to-nvvm.mlir @@ -1,37 +1,35 @@ // RUN: mlir-opt %s -lower-gpu-ops-to-nvvm-ops | FileCheck %s -module attributes {gpu.kernel_module} { - // CHECK-LABEL: func @gpu_index_ops() - func @gpu_index_ops() - attributes { gpu.kernel } { - // CHECK: = nvvm.read.ptx.sreg.tid.x : !llvm.i32 - %tIdX = "gpu.thread_id"() {dimension = "x"} : () -> (index) - // CHECK: = nvvm.read.ptx.sreg.tid.y : !llvm.i32 - %tIdY = "gpu.thread_id"() {dimension = "y"} : () -> (index) - // CHECK: = nvvm.read.ptx.sreg.tid.z : !llvm.i32 - %tIdZ = "gpu.thread_id"() {dimension = "z"} : () -> (index) +// CHECK-LABEL: func @gpu_index_ops() +func @gpu_index_ops() + attributes { gpu.kernel } { + // CHECK: = nvvm.read.ptx.sreg.tid.x : !llvm.i32 + %tIdX = "gpu.thread_id"() {dimension = "x"} : () -> (index) + // CHECK: = nvvm.read.ptx.sreg.tid.y : !llvm.i32 + %tIdY = "gpu.thread_id"() {dimension = "y"} : () -> (index) + // CHECK: = nvvm.read.ptx.sreg.tid.z : !llvm.i32 + %tIdZ = "gpu.thread_id"() {dimension = "z"} : () -> (index) - // CHECK: = nvvm.read.ptx.sreg.ntid.x : !llvm.i32 - %bDimX = "gpu.block_dim"() {dimension = "x"} : () -> (index) - // CHECK: = nvvm.read.ptx.sreg.ntid.y : !llvm.i32 - %bDimY = "gpu.block_dim"() {dimension = "y"} : () -> (index) - // CHECK: = nvvm.read.ptx.sreg.ntid.z : !llvm.i32 - %bDimZ = "gpu.block_dim"() {dimension = "z"} : () -> (index) + // CHECK: = nvvm.read.ptx.sreg.ntid.x : !llvm.i32 + %bDimX = "gpu.block_dim"() {dimension = "x"} : () -> (index) + // CHECK: = nvvm.read.ptx.sreg.ntid.y : !llvm.i32 + %bDimY = "gpu.block_dim"() {dimension = "y"} : () -> (index) + // CHECK: = nvvm.read.ptx.sreg.ntid.z : !llvm.i32 + %bDimZ = "gpu.block_dim"() {dimension = "z"} : () -> (index) - // CHECK: = nvvm.read.ptx.sreg.ctaid.x : !llvm.i32 - %bIdX = "gpu.block_id"() {dimension = "x"} : () -> (index) - // CHECK: = nvvm.read.ptx.sreg.ctaid.y : !llvm.i32 - %bIdY = "gpu.block_id"() {dimension = "y"} : () -> (index) - // CHECK: = nvvm.read.ptx.sreg.ctaid.z : !llvm.i32 - %bIdZ = "gpu.block_id"() {dimension = "z"} : () -> (index) + // CHECK: = nvvm.read.ptx.sreg.ctaid.x : !llvm.i32 + %bIdX = "gpu.block_id"() {dimension = "x"} : () -> (index) + // CHECK: = nvvm.read.ptx.sreg.ctaid.y : !llvm.i32 + %bIdY = "gpu.block_id"() {dimension = "y"} : () -> (index) + // CHECK: = nvvm.read.ptx.sreg.ctaid.z : !llvm.i32 + %bIdZ = "gpu.block_id"() {dimension = "z"} : () -> (index) - // CHECK: = nvvm.read.ptx.sreg.nctaid.x : !llvm.i32 - %gDimX = "gpu.grid_dim"() {dimension = "x"} : () -> (index) - // CHECK: = nvvm.read.ptx.sreg.nctaid.y : !llvm.i32 - %gDimY = "gpu.grid_dim"() {dimension = "y"} : () -> (index) - // CHECK: = nvvm.read.ptx.sreg.nctaid.z : !llvm.i32 - %gDimZ = "gpu.grid_dim"() {dimension = "z"} : () -> (index) + // CHECK: = nvvm.read.ptx.sreg.nctaid.x : !llvm.i32 + %gDimX = "gpu.grid_dim"() {dimension = "x"} : () -> (index) + // CHECK: = nvvm.read.ptx.sreg.nctaid.y : !llvm.i32 + %gDimY = "gpu.grid_dim"() {dimension = "y"} : () -> (index) + // CHECK: = nvvm.read.ptx.sreg.nctaid.z : !llvm.i32 + %gDimZ = "gpu.grid_dim"() {dimension = "z"} : () -> (index) - std.return - } + std.return } |
