1. 05 Feb, 2024 1 commit
    • Thorsten Kurth's avatar
      Tkurth/distributed disco (#30) · 214fa40a
      Thorsten Kurth authored
      
      
      * initial implementation of distributed DISCO layer
      
      * working distributed convolution
      
      * working refactored serial conv transpose with torch kernel
      
      * working distributed conv and transposed conv when using the python kernel
      
      * working distributed convolution with torch kernel
      
      * fixed triton kernel tests
      
      * adding print statement to debug CI
      
      * adjusting tolerances in local convolution unittest
      
      ---------
      Co-authored-by: default avatarBoris Bonev <bbonev@nvidia.com>
      214fa40a
  2. 07 Dec, 2023 1 commit
    • Thorsten Kurth's avatar
      Tkurth/flexible sharding (#22) · eab72f04
      Thorsten Kurth authored
      * working distributed fwd SHT with flexible sharding and without padding
      
      * fixing distributed sht with new logic
      
      * fixed distributed SHT and ISHT with flexible padding
      
      * working unittest for distributed fwd SHT
      
      * distributed tests converted to unit tests
      
      * bumping version number up to 0.6.4
      
      * fixing small splitting bug in th distributed
      
      * updated changeloc, removed tests folder
      eab72f04
  3. 18 Oct, 2023 1 commit
  4. 26 Apr, 2023 1 commit
  5. 17 Feb, 2023 1 commit