Xin Yao 96297fb8fd [Feature] Add bfloat16 (bf16) support (#4648) ...

* 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

2022-11-06 19:48:36 +08:00

..

dgl

[Feature] Add bfloat16 (bf16) support (#4648)

2022-11-06 19:48:36 +08:00

setup.py

[Misc] Black auto fix. (#4691)

2022-10-10 10:53:51 +08:00

update_version.py

[Misc] Black auto fix. (#4691)

2022-10-10 10:53:51 +08:00