fix amp pass for en_table_structure,mask_rcnn_r50_1x_coco,rec_mtb_nrt…

…r in auto_mixed_precision_pass.cc
PaddlePaddle · Sep 23, 2024 · efe2c0e · efe2c0e
1 parent e458f8c
commit efe2c0e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/paddle/fluid/pir/transforms/general/auto_mixed_precision_pass.cc b/paddle/fluid/pir/transforms/general/auto_mixed_precision_pass.cc
@@ -103,6 +103,7 @@ class AutoMixedPrecisionPass : public pir::Pass {
         ProcessBlock(&block, builder);
       }
     }
+    cached_cast_ops_.clear();
   }
 
   bool CanApplyOn(pir::Operation* op) const override {