This is an archive of the discontinued LLVM Phabricator instance.

Differential D101956

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

Authored by mravishankar on May 5 2021, 4:06 PM.

Details

Reviewers
nicolasvasilache
Commits
rG4b2d7ef3ea81: [mlir][Linalg] Fix test to use new reshape op form.

Diff Detail

Event Timeline

mravishankar created this revision.May 5 2021, 4:06 PM
Herald added subscribers: dcaballe, cota, teijeong and 16 others. · View Herald TranscriptMay 5 2021, 4:06 PM
mravishankar requested review of this revision.May 5 2021, 4:06 PM
Herald added a reviewer: nicolasvasilache. · View Herald TranscriptMay 5 2021, 4:06 PM
Herald added a project: Restricted Project. · View Herald Transcript
Herald added subscribers: limo1996, stephenneuendorffer, nicolasvasilache. · View Herald Transcript
This revision was not accepted when it landed; it landed in state Needs Review.May 5 2021, 4:07 PM
This revision was landed with ongoing or failed builds.
Closed by commit rG4b2d7ef3ea81: [mlir][Linalg] Fix test to use new reshape op form. (authored by mravishankar). · Explain Why
This revision was automatically updated to reflect the committed changes.
mravishankar added a commit: rG4b2d7ef3ea81: [mlir][Linalg] Fix test to use new reshape op form..
Harbormaster completed remote builds in B102874: Diff 343226.May 5 2021, 4:53 PM

Revision Contents

PathSize
mlir/
test/
Dialect/
Linalg/
3 lines

Diff 343227

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

Show First 20 Lines • Show All 90 Lines • ▼ Show 20 Lines
// ----- // -----
func @type_correctness(%arg0 : tensor<6x5xi32>, %arg1 : tensor<5xf32>, func @type_correctness(%arg0 : tensor<6x5xi32>, %arg1 : tensor<5xf32>,
%arg2 : tensor<5xf32>) -> tensor<2x3x5xf32> { %arg2 : tensor<5xf32>) -> tensor<2x3x5xf32> {
%cst_6 = constant 1.000000e+00 : f32 %cst_6 = constant 1.000000e+00 : f32
%cst_7 = constant 7.000000e+00 : f32 %cst_7 = constant 7.000000e+00 : f32
%cst_8 = constant 1.1920929E-7 : f32 %cst_8 = constant 1.1920929E-7 : f32
%25 = linalg.tensor_reshape %arg0 %25 = linalg.tensor_reshape %arg0 [[0, 1], [2]]
[affine_map<(d0, d1, d2) -> (d0, d1)>, affine_map<(d0, d1, d2) -> (d2)>]
: tensor<6x5xi32> into tensor<2x3x5xi32> : tensor<6x5xi32> into tensor<2x3x5xi32>
%26 = linalg.init_tensor [2, 3, 5] : tensor<2x3x5xf32> %26 = linalg.init_tensor [2, 3, 5] : tensor<2x3x5xf32>
%28 = linalg.generic { %28 = linalg.generic {
indexing_maps = [affine_map<(d0, d1, d2) -> (d0, d1, d2)>, indexing_maps = [affine_map<(d0, d1, d2) -> (d0, d1, d2)>,
affine_map<(d0, d1, d2) -> (d2)>, affine_map<(d0, d1, d2) -> (d2)>,
affine_map<(d0, d1, d2) -> (d2)>, affine_map<(d0, d1, d2) -> (d2)>,
affine_map<(d0, d1, d2) -> (d0, d1, d2)>], affine_map<(d0, d1, d2) -> (d0, d1, d2)>],
iterator_types = ["parallel", "parallel", "parallel"]} iterator_types = ["parallel", "parallel", "parallel"]}
Show All 19 Lines