Skip to content

GitLab

Switch branch/tag
  1. 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
  3. 29 Jun, 2019 1 commit
  5. 04 Aug, 2018 1 commit
  7. 19 Apr, 2018 1 commit
    • rusty1s's avatar
      typo · 3411455c
      rusty1s authored
      3411455c
  9. 07 Apr, 2018 1 commit
  11. 29 Jan, 2018 1 commit
  13. 27 Dec, 2017 1 commit
  15. 22 Dec, 2017 3 commits
  17. 18 Dec, 2017 2 commits
  19. 16 Dec, 2017 3 commits