colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add explicit dialect registration for mlir-hlo-opt 

This is fixing the build after upstream changes.

PiperOrigin-RevId: 327738263
This commit is contained in:
Mehdi Amini 2020-08-20 18:50:13 -07:00 committed by TensorFlow MLIR Team
parent 1912b90d25
commit f39466cc2d
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tools/mlir-hlo-opt/mlir-hlo-opt.cpp
View File

@ -13,18 +13,25 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "mlir-hlo/Dialect/mhlo/IR/register.h" #include "mlir-hlo/Dialect/mhlo/IR/chlo_ops.h"
#include "mlir-hlo/Dialect/mhlo/IR/hlo_ops.h"
#include "mlir-hlo/Dialect/mhlo/IR/lhlo_ops.h"
#include "mlir-hlo/Dialect/mhlo/transforms/register_passes.h" #include "mlir-hlo/Dialect/mhlo/transforms/register_passes.h"
#include "mlir/InitAllDialects.h" #include "mlir/InitAllDialects.h"
#include "mlir/InitAllPasses.h" #include "mlir/InitAllPasses.h"
#include "mlir/Support/MlirOptMain.h" #include "mlir/Support/MlirOptMain.h"
int main(int argc, char **argv) { int main(int argc, char **argv) {
mlir::DialectRegistry registry;
mlir::registerAllDialects(registry);
mlir::registerAllPasses(); mlir::registerAllPasses();
mlir::mhlo::registerAllMhloPasses(); mlir::mhlo::registerAllMhloPasses();
mlir::lmhlo::registerAllLmhloPasses(); mlir::lmhlo::registerAllLmhloPasses();
mlir::DialectRegistry registry;
mlir::registerAllDialects(registry);
registry.insert<mlir::mhlo::MhloDialect>();
registry.insert<mlir::chlo::HloClientDialect>();
registry.insert<mlir::lmhlo::LmhloDialect>();
return failed( return failed(
mlir::MlirOptMain(argc, argv, "MLIR HLO pass driver\n", registry)); mlir::MlirOptMain(argc, argv, "MLIR HLO pass driver\n", registry));
} }