Skip to content

Releases: NVIDIA/torch-harmonics

v0.7.2

01 Oct 16:34
663bea1
Compare
Choose a tag to compare
  • Added resampling modules for convenience
  • Changing behavior of distributed SHT to use dim=-3 as channel dimension
  • Fixing SHT unittests to test SHT and ISHT individually, rather than the roundtrip
  • Changing the way custom CUDA extensions are handled

v0.7.1

27 Aug 21:47
5d7e9b0
Compare
Choose a tag to compare
  • Hotfix to AMP in SFNO example

v0.7.0

27 Aug 16:49
1bfda53
Compare
Choose a tag to compare
  • 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

30 Jan 14:08
54502a1
Compare
Choose a tag to compare
  • 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

07 Dec 12:47
eab72f0
Compare
Choose a tag to compare
  • 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

v0.6.3

25 Nov 17:28
13aa492
Compare
Choose a tag to compare
  • Updated SFNO example
  • Fixes NCCL issues with distributed SHT
  • other bugixes

v0.6.2

08 Aug 09:47
Compare
Choose a tag to compare
  • Adding github CI
  • Changed SHT modules to convert dtype dynamically when computing the SHT/ISHT
  • Bugfixes to fix importing examples

v0.6.1

08 Jun 14:36
Compare
Choose a tag to compare
  • Minor bugfixes to export SFNO code
  • Readme should now render correctly in PyPI

v0.6.0

07 Jun 17:51
Compare
Choose a tag to compare
  • Added SFNO example
  • Added Shallow Water Equations Dataset for SFNO training
  • Cleanup of the repository and added PyPI
  • Updated Readme