35 lines · plain
1// First test various sets of invalid arguments2// RUN: not mlir-opt -allow-unregistered-dialect %s -pass-pipeline='builtin.module(func.func(test-scf-parallel-loop-collapsing))' 2>&1 | FileCheck %s --check-prefix=CL03// CL0: No collapsed-indices were specified. This pass is only for testing and does not automatically collapse all parallel loops or similar4 5// RUN: not mlir-opt -allow-unregistered-dialect %s -pass-pipeline='builtin.module(func.func(test-scf-parallel-loop-collapsing{collapsed-indices-1=1}))' 2>&1 | FileCheck %s --check-prefix=CL16// CL1: collapsed-indices-1 specified but not collapsed-indices-07 8// RUN: not mlir-opt -allow-unregistered-dialect %s -pass-pipeline='builtin.module(func.func(test-scf-parallel-loop-collapsing{collapsed-indices-0=1 collapsed-indices-2=2}))' 2>&1 | FileCheck %s --check-prefix=CL29// CL2: collapsed-indices-2 specified but not collapsed-indices-110 11// RUN: not mlir-opt -allow-unregistered-dialect %s -pass-pipeline='builtin.module(func.func(test-scf-parallel-loop-collapsing{collapsed-indices-0=1 collapsed-indices-1=2}))' 2>&1 | FileCheck %s --check-prefix=NON-ZERO12// NON-ZERO: collapsed-indices arguments must include all values [0,N).13 14// RUN: not mlir-opt -allow-unregistered-dialect %s -pass-pipeline='builtin.module(func.func(test-scf-parallel-loop-collapsing{collapsed-indices-0=0 collapsed-indices-1=2}))' 2>&1 | FileCheck %s --check-prefix=NON-CONTIGUOUS15// NON-CONTIGUOUS: collapsed-indices arguments must include all values [0,N).16 17 18// Then test for invalid combinations of argument+input-ir19// RUN: mlir-opt -allow-unregistered-dialect %s -pass-pipeline='builtin.module(func.func(test-scf-parallel-loop-collapsing{collapsed-indices-0=0,1}))' -verify-diagnostics20func.func @too_few_iters(%arg0: index, %arg1: index, %arg2: index) {21 // expected-error @+1 {{op has 1 iter args while this limited functionality testing pass was configured only for loops with exactly 2 iter args.}}22 scf.parallel (%arg3) = (%arg0) to (%arg1) step (%arg2) {23 scf.reduce24 }25 return26}27 28func.func @too_many_iters(%arg0: index, %arg1: index, %arg2: index) {29 // expected-error @+1 {{op has 3 iter args while this limited functionality testing pass was configured only for loops with exactly 2 iter args.}}30 scf.parallel (%arg3, %arg4, %arg5) = (%arg0, %arg0, %arg0) to (%arg1, %arg1, %arg1) step (%arg2, %arg2, %arg2) {31 scf.reduce32 }33 return34}35