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

for good measure. PiperOrigin-RevId: 344512196
2020-11-27 03:43:21 -08:00 · 2020-11-27 03:43:21 -08:00 · f183e95e69
parent d617f3edb6
commit f183e95e69
1 changed files with 3 additions and 0 deletions
--- a/lib/Dialect/mhlo/transforms/hlo_legalize_to_lhlo.cc
+++ b/lib/Dialect/mhlo/transforms/hlo_legalize_to_lhlo.cc
@ -581,6 +581,9 @@ struct HloLegalizeToLhlo
    populateHLOToLHLOConversionPattern(&context, &converter, &patterns);
    populateFuncOpTypeConversionPattern(patterns, &context, converter);
    populateCallOpTypeConversionPattern(patterns, &context, converter);
+    populateBranchOpInterfaceAndReturnOpTypeConversionPattern(
+        patterns, &context, converter);
+
    populateShapeStructuralTypeConversionsAndLegality(&context, converter,
                                                      patterns, target);
    if (failed(applyPartialConversion(getOperation(), target,