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
  2. 23 Jul, 2022 1 commit
  3. 24 Mar, 2022 1 commit
  4. 22 Oct, 2021 1 commit
  5. 28 Jul, 2021 1 commit
  6. 19 Jul, 2021 2 commits
  7. 30 May, 2021 1 commit
  8. 23 Nov, 2020 2 commits
  9. 24 Aug, 2020 1 commit
  10. 16 Jul, 2020 1 commit
  11. 16 Apr, 2020 1 commit
  12. 04 Feb, 2020 1 commit
  13. 03 Feb, 2020 1 commit
  14. 30 Jan, 2020 3 commits
  15. 29 Jan, 2020 2 commits