• Z
    [AMP] Support amp for Intermediate_dygraph (#40623) · c12f7d48
    zhangbo9674 提交于
    * approve amp for intermediate_dygraph
    
    * add amp_utils for intermediate_dygraph
    
    * add amp needcast check for mlu & npu
    
    * test unittest
    
    * add SetGradNode for set_stop_gradient && add checktensor for GradientHooks
    
    * refine code
    
    * refien unittest of imperative_amp for new dygraph
    
    * inplace api skip amp
    
    * add test_imperative_qat_amp for intermediate amp
    
    * refine code
    
    * refine test_amp ci strategy
    
    * refine unittest code
    
    * refine amp_utils code
    
    * refine amp getpromotetype for some special op
    
    * refine unittest code
    c12f7d48
grad_node_info.h 10.5 KB