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. 13 Jul, 2022 1 commit
  5. 09 Jul, 2022 1 commit
  6. 30 Jun, 2022 1 commit
  7. 29 Jun, 2022 1 commit
  8. 31 May, 2022 1 commit
  9. 25 Mar, 2022 1 commit
  10. 24 Mar, 2022 1 commit
  11. 21 Mar, 2022 2 commits
  12. 11 Mar, 2022 25 commits
  13. 10 Mar, 2022 3 commits