"...en/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "5551506b295708b55636862aa5bad0fd64fa3f50"
  • Xin Yao's avatar
    [Feature] Add bfloat16 (bf16) support (#4648) · 96297fb8
    Xin Yao authored
    * add bf16 specializations
    
    * remove SWITCH_BITS
    
    * enable amp for bf16
    
    * remove SWITCH_BITS for cpu kernels
    
    * enbale bf16 based on CUDART
    
    * fix compiling for sm<80
    
    * fix cpu build
    
    * enable unit tests
    
    * update doc
    
    * disable test for CUDA < 11.0
    
    * address comments
    
    * address comments
    96297fb8
array_scatter.cu 2.19 KB