[Fix] Fix the conflict issue between AMP and weight prepack (#2657)

intel · Apr 8, 2024 · faf8925 · faf8925
1 parent 49d1136
commit faf8925
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/itex/core/graph/auto_mixed_precision/auto_mixed_precision_lists.h b/itex/core/graph/auto_mixed_precision/auto_mixed_precision_lists.h
@@ -172,6 +172,8 @@ class AutoMixedPrecisionLists {
       // Should be remove in future.
       "MklAUGRU",
       "MklGRU",
+      // from Intel-TF bert-large, should be remove in future.
+      "_FusedMatMul",
       // Below ops are fusion ops.
       "_ITEXConv2DBackpropFilterWithBias",
       "_ITEXConv2DBackpropInputWithSlice",