[Mlir-commits] [mlir] 4b2d7ef - [mlir][Linalg] Fix test to use new reshape op form.

llvmlistbot at llvm.org llvmlistbot at llvm.org
Wed May 5 16:07:33 PDT 2021


Author: MaheshRavishankar
Date: 2021-05-05T16:06:58-07:00
New Revision: 4b2d7ef3ea81d0d6746e609b46f38bfceff23838

URL: https://github.com/llvm/llvm-project/commit/4b2d7ef3ea81d0d6746e609b46f38bfceff23838
DIFF: https://github.com/llvm/llvm-project/commit/4b2d7ef3ea81d0d6746e609b46f38bfceff23838.diff

LOG: [mlir][Linalg] Fix test to use new reshape op form.

Differential Revision: https://reviews.llvm.org/D101956

Added: 
    

Modified: 
    mlir/test/Dialect/Linalg/fusion-push-reshape.mlir

Removed: 
    


################################################################################
diff  --git a/mlir/test/Dialect/Linalg/fusion-push-reshape.mlir b/mlir/test/Dialect/Linalg/fusion-push-reshape.mlir
index cc46bd3c273f..5f0ec829978d 100644
--- a/mlir/test/Dialect/Linalg/fusion-push-reshape.mlir
+++ b/mlir/test/Dialect/Linalg/fusion-push-reshape.mlir
@@ -96,8 +96,7 @@ func @type_correctness(%arg0 : tensor<6x5xi32>, %arg1 : tensor<5xf32>,
   %cst_6 = constant 1.000000e+00 : f32
   %cst_7 = constant 7.000000e+00 : f32
   %cst_8 = constant 1.1920929E-7 : f32
-  %25 = linalg.tensor_reshape %arg0
-      [affine_map<(d0, d1, d2) -> (d0, d1)>, affine_map<(d0, d1, d2) -> (d2)>]
+  %25 = linalg.tensor_reshape %arg0 [[0, 1], [2]]
       : tensor<6x5xi32> into tensor<2x3x5xi32>
   %26 = linalg.init_tensor [2, 3, 5] : tensor<2x3x5xf32>
   %28 = linalg.generic {


        


More information about the Mlir-commits mailing list