d31c9e5a46
Having the input dumped on failure seems like a better default: I debugged FileCheck tests for a while without knowing about this option, which really helps to understand failures. Remove `-dump-input-on-failure` and the environment variable FILECHECK_DUMP_INPUT_ON_FAILURE which are now obsolete. Differential Revision: https://reviews.llvm.org/D81422
35 lines
1.9 KiB
MLIR
35 lines
1.9 KiB
MLIR
// RUN: mlir-opt -convert-affine-for-to-gpu="gpu-block-dims=0 gpu-thread-dims=1" %s | FileCheck --check-prefix=CHECK-THREADS %s
|
|
// RUN: mlir-opt -convert-affine-for-to-gpu="gpu-block-dims=1 gpu-thread-dims=0" %s | FileCheck --check-prefix=CHECK-BLOCKS %s
|
|
|
|
// CHECK-THREADS-LABEL: @one_d_loop
|
|
// CHECK-BLOCKS-LABEL: @one_d_loop
|
|
func @one_d_loop(%A : memref<?xf32>, %B : memref<?xf32>) {
|
|
// Bounds of the loop, its range and step.
|
|
// CHECK-THREADS-NEXT: %{{.*}} = constant 0 : index
|
|
// CHECK-THREADS-NEXT: %{{.*}} = constant 42 : index
|
|
// CHECK-THREADS-NEXT: %[[BOUND:.*]] = subi %{{.*}}, %{{.*}} : index
|
|
// CHECK-THREADS-NEXT: %{{.*}} = constant 1 : index
|
|
// CHECK-THREADS-NEXT: %[[ONE:.*]] = constant 1 : index
|
|
//
|
|
// CHECK-BLOCKS-NEXT: %{{.*}} = constant 0 : index
|
|
// CHECK-BLOCKS-NEXT: %{{.*}} = constant 42 : index
|
|
// CHECK-BLOCKS-NEXT: %[[BOUND:.*]] = subi %{{.*}}, %{{.*}} : index
|
|
// CHECK-BLOCKS-NEXT: %{{.*}} = constant 1 : index
|
|
// CHECK-BLOCKS-NEXT: %[[ONE:.*]] = constant 1 : index
|
|
|
|
// CHECK-THREADS-NEXT: gpu.launch blocks(%[[B0:.*]], %[[B1:.*]], %[[B2:.*]]) in (%{{.*}} = %[[ONE]], %{{.*}} = %[[ONE]], %{{.*}}0 = %[[ONE]]) threads(%[[T0:.*]], %[[T1:.*]], %[[T2:.*]]) in (%{{.*}} = %[[BOUND]], %{{.*}} = %[[ONE]], %{{.*}} = %[[ONE]])
|
|
// CHECK-BLOCKS-NEXT: gpu.launch blocks(%[[B0:.*]], %[[B1:.*]], %[[B2:.*]]) in (%{{.*}} = %[[BOUND]], %{{.*}} = %[[ONE]], %{{.*}}0 = %[[ONE]]) threads(%[[T0:.*]], %[[T1:.*]], %[[T2:.*]]) in (%{{.*}} = %[[ONE]], %{{.*}} = %[[ONE]], %{{.*}} = %[[ONE]])
|
|
affine.for %i = 0 to 42 {
|
|
// CHECK-THREADS-NEXT: %[[INDEX:.*]] = addi %{{.*}}, %[[T0]]
|
|
// CHECK-THREADS-NEXT: load %{{.*}}[%[[INDEX]]]
|
|
// CHECK-BLOCKS-NEXT: %[[INDEX:.*]] = addi %{{.*}}, %[[B0]]
|
|
// CHECK-BLOCKS-NEXT: load %{{.*}}[%[[INDEX]]]
|
|
%0 = load %A[%i] : memref<?xf32>
|
|
store %0, %B[%i] : memref<?xf32>
|
|
// CHECK-THREADS: gpu.terminator
|
|
// CHECK-BLOCKS: gpu.terminator
|
|
}
|
|
return
|
|
}
|
|
|