GitLab

This reverts commit 82881276.

* setting imaginary parts of DCT and nyquist frequency to zero in IRSHT variants

* small fix

* making einsum result contiguous

* adding zero frequency to distributed sht

* small hotfix for lobatto grid precomputation routine

* adding lobatto grid to tests

* adding caching

* replacing many numpy calls with torch calls

* bumping up version number to 0.7.6

* Improved computation of Morlet filter basis and switched to a Hann window.

* addresses #064 and some cleanup

switched psi tensor computation to double precision and implemented a fudge factor for theta_cutoff to avoid aliasing issues with the grid width

updated resampling module to handle corner cases where the input signal needs to be extended towards the pole

Adding FilterBasis datastructure to contain information about the type of basis used in disco convolution

adding local SNO architecture and updating trraining script for SWE prediction with powerspectrum computation

* safeguarding all custom CUDA and C++ routines via the _cuda_extension_available flag

* bumping up version number

* changing distributeds SHT to use dim=-3 as the channel dimension for distributed transpose

* Fixed formatting in tests

* Adding bash script for running test suite

* Replaced asserts regarding number of dimensions in tensor with checks

* Adding spherical upsampling routine and example notebook for resampling

- doing manual type conversion in custom autograd
- fixing stride issues in backward pass by making some output tensors contiguous

* AMP hotfix

* Bumping up version to 0.7.1

* reverting deploy script

* moving setuptools install

* removing requirements

* bumping up version

* bumping up the torch-harmonics version to 0.7.0

* updating amp to new torch.amp
* using amp autocrats to FP32 for disco convolution kernels
* implemented reduce_scatter routines but disabled those because of memory fluctuations which can cause OOM on big networks