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