1. 11 Jan, 2023 1 commit
  2. 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. 23 Jul, 2022 1 commit
  4. 09 Jul, 2022 1 commit
  5. 21 Mar, 2022 1 commit
  6. 02 Mar, 2021 1 commit
  7. 11 Feb, 2020 1 commit