8c213ef122
[Feature] Enable bfloat16 convert functions in Python API ( #5760 )
2023-07-31 18:09:32 +08:00
ce378327c7
fix bf16 tests ( #5089 )
2022-12-29 21:36:39 +08:00
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
05d9d4969b
Update ( #4277 )
...
Co-authored-by: Ubuntu <ubuntu@ip-172-31-53-142.us-west-2.compute.internal >
2022-07-21 13:27:42 +08:00
3ca52b1e59
[Fix] Various Fix after Bug Bash ( #2620 )
...
* update
* Misc fix
* Update
2021-02-05 17:34:16 +08:00
7bab1365e2
[feature] Supporting half precision floating data type (fp16). ( #2552 )
...
* add tvm as submodule
* compilation is ok but calling fails
* can call now
* pack multiple modules, change names
* upd
* upd
* upd
* fix cmake
* upd
* upd
* upd
* upd
* fix
* relative path
* upd
* upd
* upd
* singleton
* upd
* trigger
* fix
* upd
* count reducible
* upd
* upd
* upd
* upd
* upd
* upd
* upd
* upd
* upd
* only keep related files
* upd
* upd
* upd
* upd
* lint
* lint
* lint
* lint
* pylint
* upd
* upd
* compilation
* fix
* upd
* upd
* upd
* upd
* upd
* upd
* upd doc
* refactor
* fix
* upd number
Co-authored-by: Zhi Lin <linzhilynn@gmail.com >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-42-78.us-east-2.compute.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-21-156.us-east-2.compute.internal >
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2021-01-28 11:21:58 +08:00
Ilia Taraban