- 05 Feb, 2024 1 commit
-
-
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:Boris Bonev <bbonev@nvidia.com>
-
- 07 Dec, 2023 1 commit
-
-
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
-
- 18 Oct, 2023 1 commit
-
-
Thorsten Kurth authored
* fixing contig in distributed sht * make it normally contiguous before all to all * commented the code to be a WAR
-
- 26 Apr, 2023 1 commit
-
-
Boris Bonev authored
-
- 17 Feb, 2023 1 commit
-
-
Boris Bonev authored
-