diff options
| author | Uday Bondhugula <uday@polymagelabs.com> | 2019-12-20 09:43:34 -0800 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-12-20 09:44:03 -0800 |
| commit | e5691c512f7335fe90093bb9989da1760d80701b (patch) | |
| tree | aa9054cc7b1b09c157729a7cc0c62b5f8ee6c089 /mlir/test/Transforms | |
| parent | 42d46b4efa9d7b596c74c101a777f4464a8f1fd0 (diff) | |
| download | bcm5719-llvm-e5691c512f7335fe90093bb9989da1760d80701b.tar.gz bcm5719-llvm-e5691c512f7335fe90093bb9989da1760d80701b.zip | |
fix isValidDim for block arg case
- a block argument associated with an arbitrary op can't be a valid
dimensional identifier; it has to be the block argument of either
a function op or an affine.for.
Signed-off-by: Uday Bondhugula <uday@polymagelabs.com>
Closes tensorflow/mlir#331
COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/mlir/pull/331 from bondhugula:valid_dim 3273b4fcbaa31fb7b6671d93c9e42a6b2a6a4e4c
PiperOrigin-RevId: 286593693
Diffstat (limited to 'mlir/test/Transforms')
| -rw-r--r-- | mlir/test/Transforms/slicing-utils.mlir | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/mlir/test/Transforms/slicing-utils.mlir b/mlir/test/Transforms/slicing-utils.mlir index 49410dbbf9e..8c6fb01e1f7 100644 --- a/mlir/test/Transforms/slicing-utils.mlir +++ b/mlir/test/Transforms/slicing-utils.mlir @@ -222,13 +222,12 @@ func @slicing_test() { // FWDBWD-LABEL: slicing_test_2 func @slicing_test_2() { %c0 = constant 0 : index - %c1 = constant 1 : index %c2 = constant 2 : index %c16 = constant 16 : index - loop.for %i0 = %c0 to %c16 step %c1 { + affine.for %i0 = %c0 to %c16 { affine.for %i1 = (i)[] -> (i)(%i0) to 10 { // BWD: matched: %[[b:.*]] {{.*}} backward static slice: - // BWD: loop.for {{.*}} + // BWD: affine.for {{.*}} // affine.for appears in the body of loop.for // BWD: affine.for {{.*}} @@ -238,7 +237,7 @@ func @slicing_test_2() { %b = "slicing-test-op"(%i1): (index) -> index // BWD: matched: %[[c:.*]] {{.*}} backward static slice: - // BWD: loop.for {{.*}} + // BWD: affine.for {{.*}} // affine.for appears in the body of loop.for // BWD-NEXT: affine.for {{.*}} |
