- 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>
-
- 30 Jan, 2024 1 commit
-
-
Boris Bonev authored
* Cleaned up DISCO convolutions
-
- 22 Dec, 2023 1 commit
-
-
Boris Bonev authored
Changed the code to only implicitly use sparse tensors in the modules, in order to enable compatibility with DDP
-
- 20 Dec, 2023 1 commit
-
-
Boris Bonev authored
* Moved convolutions and exposed them directly * Added transposition to the unit test * Minor bugfix in CPU version of DISCO transpose code * Adding convolution tests to CI * Added gradient check * Checking the weight grad as well * Added test for anisotropic kernels
-
- 15 Dec, 2023 1 commit
-
-
Boris Bonev authored
-
- 14 Dec, 2023 1 commit
-
-
Boris Bonev authored
* Fixing the precomputation of the Psi matrix * moving tests to a parent folder * removing tqdm * deleting deprecated distributed tests * Moving distributed test * adapting workfolow * Added some more comments to make the code more understandable * more detailed explanation for the derivation of the rotation angles * added another comment --------- Co-authored-by:Thorsten Kurth <tkurth@nvidia.com>
-
- 07 Dec, 2023 1 commit
-
-
Boris Bonev authored
* fixing banding issues with corrected computation of kernel size * fixing Parameters * changed normalization of isotropic kernels --------- Co-authored-by:
-
- 05 Dec, 2023 1 commit
-
-
Boris Bonev authored
* Adding prototype implementation of disco convolutions * adding s2convolutions.py * Somewhat functional first implementation in Triton * Somewhat functional first implementation in Triton * Made batched execution work on GPU and refactored precomputation of kernel values and supports. * Adding the Lobatto grid to the grid selection method in quadrature * fixing bug in triton kernel when iterating over non-zeros * Updating with working custom triton implementation of DISCO convolution * Merged both forward and backward DISCO contraction into a single kernel * Some cleanup and minor bugfix * bugfixes to the reference implementation * adding weights to s2conv * removing unnecessary imports * suggestion for torch harmonics, math should be checked thoroughly * Intermediate working reference implementation for the transpose DISCO convolution. Fixed normalization of kernels * adjusting cutoff frequency in disco convolution * fixing transpose DISCO contraction * moving triton to install_requires * enabling triton by default --------- Co-authored-by:
-
- 02 Dec, 2023 1 commit
-
-
Boris Bonev authored
* Adding prototype implementation of disco convolutions * adding s2convolutions.py * Somewhat functional first implementation in Triton * Somewhat functional first implementation in Triton * Made batched execution work on GPU and refactored precomputation of kernel values and supports. * Adding the Lobatto grid to the grid selection method in quadrature * fixing bug in triton kernel when iterating over non-zeros * Updating with working custom triton implementation of DISCO convolution * Merged both forward and backward DISCO contraction into a single kernel * Some cleanup and minor bugfix * bugfixes to the reference implementation * adding weights to s2conv * removing unnecessary imports * suggestion for torch harmonics, math should be checked thoroughly * Intermediate working reference implementation for the transpose DISCO convolution. Fixed normalization of kernels * adjusting cutoff frequency in disco convolution * fixing transpose DISCO contraction * moving triton to install_requires --------- Co-authored-by:
-
