Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • 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
build_dgl.sh 1.63 KB