• C
    [cherry-pick]Fix cuda12 test (#54622) · bd1a240b
    Charles-hit 提交于
    * [AMP Prim OP]support some prim ops for bf16 dtype part3 (#54368)
    
    * support some prim ops bf16 dtype
    
    * fix cmake
    
    * [AMP Prim OP]support some prim ops for bf16 dtype part5 (#54422)
    
    * support some prim ops for bf16 dtype
    
    * remove useless code
    
    * support some prim ops bf16 dtype (#54399)
    bd1a240b
test_top_k_v2_op.py 16.4 KB