Skip to content

GitLab

Switch branch/tag
  1. 24 Nov, 2022 1 commit
  3. 03 Oct, 2022 1 commit
    • Jeff Daily's avatar
      enable ROCm build; add BF16 for ROCm and CUDA (#325) · 9a651d91
      Jeff Daily authored 
      * first step, everything compiles

* fix rebuilds; skip cuda version check for rocm

* use macro for __shfl_up_sync __shfl_down_sync

* add BFloat16 support for ROCm and CUDA

* add USE_ROCM definition to setup.py

* flake8 fixes
      9a651d91
  5. 25 Mar, 2022 1 commit
  7. 22 Oct, 2021 1 commit
  9. 07 Sep, 2021 1 commit
  11. 05 Aug, 2021 1 commit
  13. 28 Jul, 2021 1 commit
  15. 05 Jun, 2021 1 commit
  17. 02 Mar, 2021 3 commits
  19. 19 Jan, 2021 1 commit
  21. 12 Jan, 2021 1 commit
  23. 03 Jan, 2021 1 commit
  25. 11 Dec, 2020 1 commit
  27. 18 Jul, 2020 2 commits
  29. 24 Jun, 2020 1 commit
  31. 23 Apr, 2020 1 commit
  33. 01 Apr, 2020 1 commit
  35. 22 Feb, 2020 1 commit
  37. 18 Feb, 2020 2 commits
  39. 14 Feb, 2020 2 commits
  41. 13 Feb, 2020 8 commits
  43. 11 Feb, 2020 7 commits