"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "a5fc62f81957c739b1d4a8fd99bf551a2949dc3c"
  • 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
utils.h 7.91 KB