• B
    【AMP OP&Test】unit test for accuracy_op (#51009) · 8c61a95a
    Bo Zhang 提交于
    * test_accuracy_op
    
    * add create_test_fp/bf16_class
    
    * cast after calculation
    
    * change convert_uint16_to_float_ifneed
    
    * delete TestAccuracyOpFp32 according to PR comment
    
    * fix the rtol setting rules in bfloat16 forward
    8c61a95a
accuracy_kernel.cu 5.3 KB