• C
    [prim] move batch_norm prim test to op_test (#54458) · 58b4c60f
    cyber-pioneer 提交于
    * move batch_norm prim test to op_test
    
    * fix optest bug
    
    * add test to cmake
    
    * add cinn test case
    
    * fix batch_norm prim grad bf16
    
    * fix code
    
    * add cuda check
    
    * fix batch_norm bfloat16
    
    * fix cpu bfloat16 bug
    
    * skip non-bfloat16-supported platform
    
    * fix code
    
    * fix cinn rtol and atol in bfloat16
    
    * fix name
    
    * fix config
    58b4c60f
test_batch_norm_op_prim.py 15.1 KB