GitLab

fixing bug in quadrature weights for full attention. Adding better unit tests for attention. Cleanup in the cuda code.

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

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