Releases: NVIDIA/torch-harmonics
Releases · NVIDIA/torch-harmonics
v0.7.2
v0.7.1
v0.7.0
- CUDA-accelerated DISCO convolutions
- Updated DISCO convolutions to support even number of collocation points across the diameter
- Distributed DISCO convolutions
- Fused quadrature into multiplication with the Psi tensor to lower memory footprint
- Removed DISCO convolution in the plane to focus on the sphere
- Updated unit tests which now include tests for the distributed convolutions
v0.6.5
- Discrete-continuous (DISCO) convolutions on the sphere and in two dimensions
- DISCO supports isotropic and anisotropic kernel functions parameterized as hat functions
- Supports regular and transpose convolutions
- Experimental accelerated spherical DISCO convolutions on GPU via Triton Kernel
- Unittests for DISCO convolutions in tests/test_convolution.py
v0.6.4
- reworking distributed to allow for uneven split tensors, effectively removing the necessity of padding the transformed tensors
- distributed SHT tests are now using unittest. Test extended to vector SHT versions. Tests are defined in torch_harmonics/distributed/distributed_tests.py
- base pytorch container version bumped up to 23.11 in Dockerfile