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

Register the new branch op rewrite patterns in hlo to lhlo conversion 

for good measure.

PiperOrigin-RevId: 344512196
This commit is contained in:
Stephan Herhut 2020-11-27 03:43:21 -08:00 committed by TensorFlow MLIR Team
parent d617f3edb6
commit f183e95e69
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/Dialect/mhlo/transforms/hlo_legalize_to_lhlo.cc
View File

@ -581,6 +581,9 @@ struct HloLegalizeToLhlo
populateHLOToLHLOConversionPattern(&context, &converter, &patterns); populateHLOToLHLOConversionPattern(&context, &converter, &patterns);
populateFuncOpTypeConversionPattern(patterns, &context, converter); populateFuncOpTypeConversionPattern(patterns, &context, converter);
populateCallOpTypeConversionPattern(patterns, &context, converter); populateCallOpTypeConversionPattern(patterns, &context, converter);
populateBranchOpInterfaceAndReturnOpTypeConversionPattern(
patterns, &context, converter);
populateShapeStructuralTypeConversionsAndLegality(&context, converter, populateShapeStructuralTypeConversionsAndLegality(&context, converter,
patterns, target); patterns, target);
if (failed(applyPartialConversion(getOperation(), target, if (failed(applyPartialConversion(getOperation(), target,