maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

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

Differential Revision: https://reviews.llvm.org/D101956
This commit is contained in:
MaheshRavishankar 2021-05-05 16:05:44 -07:00
parent 6251b2f7f6
commit 4b2d7ef3ea
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mlir/test/Dialect/Linalg/fusion-push-reshape.mlir
View File

@ -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 {