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

[mlir][linalg] Delete unused vars if there are shaped-only operands. 

Reviewed By: stella.stamenova

Differential Revision: https://reviews.llvm.org/D97851
This commit is contained in:
Hanhan Wang 2021-03-03 09:35:54 -08:00
parent 5d7e0a23c6
commit 497b7b8c00
2 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlir/test/mlir-linalg-ods-gen/test-linalg-ods-gen.tc
View File

@ -194,6 +194,8 @@ def test8(A: f32(M, K), B: f32(K)) -> (C: f32(M))
// IMPL: auto map0 = AffineMap::get(2, 2, {d0, d1}, context);
// IMPL: auto map1 = AffineMap::get(2, 2, {d1}, context);
// IMPL: auto map2 = AffineMap::get(2, 2, {d0}, context);
// IMPL-LABEL: void Test9Op::regionBuilder(Block &block, ValueRange captures) {
// IMPL: Value [[a:.*]](args[0]), [[c:.*]](args[2]);
ods_def<Test9Op>:
def test9(A: f32(M, K), B: f32(K)) -> (C: f32(M))
{

13
mlir/tools/mlir-linalg-ods-gen/mlir-linalg-ods-gen.cpp
View File

@ -35,6 +35,7 @@
#include "llvm/Support/ToolOutputFile.h"
#include <map>
#include <set>
#define DEBUG_TYPE "linalg-ods-gen"
@ -2341,14 +2342,14 @@ void TCParser::printRegionBuilder(llvm::raw_ostream &os, StringRef cppOpName,
(linalg_yield(ValueRange{ {3} }));
})FMT";
unsigned idx = 0;
std::string valueHandleStr;
llvm::raw_string_ostream valueHandleStringStream(valueHandleStr);
llvm::interleaveComma(
llvm::seq<int>(0, state.numArgs), valueHandleStringStream, [&](auto) {
valueHandleStringStream << "_" << idx << "(args[" << idx << "])";
idx++;
});
std::set<unsigned> usedTensorId;
for (const auto &iter : state.orderedTensorArgs)
usedTensorId.insert(iter.second);
llvm::interleaveComma(usedTensorId, valueHandleStringStream, [&](auto idx) {
valueHandleStringStream << "_" << idx << "(args[" << idx << "])";
});
std::string expressionsStr;
llvm::raw_string_ostream expressionStringStream(expressionsStr);