• S
    support pure bfloat16 for more ops (#46364) · b7b231a6
    sneaxiy 提交于
    * support pure bfloat16
    
    * support bf16 linear
    
    * update PR to pass CI
    
    * tiny fix where_grad_kernel.cu
    
    * add bfloat16 to selu_grad to pass CI
    
    * fix selu grad compilation error
    b7b231a6
pixel_shuffle_grad_kernel.cu 1.1 KB