llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

changed name option to decompose-complex-ops

pull/822/head snapshot-20220505.432
Kristof Denolf 2022-05-04 15:09:55 -07:00 committed by Sean Silva
parent 5243638e33
commit e682b1d0f3
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/torch-mlir/Dialect/Torch/Transforms/Passes.h
View File

@ -35,8 +35,8 @@ struct TorchLoweringPipelineOptions
// If this option is false, decompose complex operations.
// If this option is true, skip decomposition of complex operations.
Option<bool> nodecompose{*this, "nodecompose", llvm::cl::desc("No decomposition."),
llvm::cl::init(false)};
Option<bool> decompose{*this, "decompose-complex-ops", llvm::cl::desc("Decompose complex operations."),
llvm::cl::init(true)};
};
/// Creates a pipeline that lowers the object graph IR that is produced by

2
lib/Dialect/Torch/Transforms/Passes.cpp
View File

@ -155,7 +155,7 @@ void mlir::torch::Torch::createTorchFunctionToTorchBackendPipeline(
pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());
}
if (!options.nodecompose)
if (options.decompose)
pm.addNestedPass<func::FuncOp>(Torch::createDecomposeComplexOpsPass());
// TODO: VerifyTorchBackendContractPass.