minor refactor
This commit is contained in:
@@ -146,7 +146,7 @@ void ONNXToSpatialPass::runOnOperation() {
|
|||||||
encapsulateGlobalInstruction(*entryFunc);
|
encapsulateGlobalInstruction(*entryFunc);
|
||||||
|
|
||||||
// Dump to file for debug
|
// Dump to file for debug
|
||||||
dumpModule(moduleOp, "spatial");
|
dumpModule(moduleOp, "spatial0");
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
|
|||||||
@@ -420,7 +420,7 @@ void GraphDCP::to_dot() {
|
|||||||
std::string outputDir = onnx_mlir::getOutputDir();
|
std::string outputDir = onnx_mlir::getOutputDir();
|
||||||
if (outputDir.empty())
|
if (outputDir.empty())
|
||||||
return;
|
return;
|
||||||
std::string graphDir = outputDir + "/DCPGraph";
|
std::string graphDir = outputDir + "/dcp_graph";
|
||||||
onnx_mlir::createDirectory(graphDir);
|
onnx_mlir::createDirectory(graphDir);
|
||||||
std::fstream file(graphDir + "/graph_" + std::to_string(index++) + ".dot", std::ios::out);
|
std::fstream file(graphDir + "/graph_" + std::to_string(index++) + ".dot", std::ios::out);
|
||||||
file << "digraph G {\n";
|
file << "digraph G {\n";
|
||||||
|
|||||||
@@ -133,7 +133,7 @@ public:
|
|||||||
computeNodetoRemove.erase();
|
computeNodetoRemove.erase();
|
||||||
}
|
}
|
||||||
func::FuncOp func = getOperation();
|
func::FuncOp func = getOperation();
|
||||||
dumpModule(cast<ModuleOp>(func->getParentOp()), "SpatialDCPMerged");
|
dumpModule(cast<ModuleOp>(func->getParentOp()), "spatial1_dcp_merged");
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
|||||||
Reference in New Issue
Block a user