Tkurth/flexible sharding (#22)

* working distributed fwd SHT with flexible sharding and without padding

* fixing distributed sht with new logic

* fixed distributed SHT and ISHT with flexible padding

* working unittest for distributed fwd SHT

* distributed tests converted to unit tests

* bumping version number up to 0.6.4

* fixing small splitting bug in th distributed

* updated changeloc, removed tests folder

Author: azrael417

Changelog.md

@@ -2,6 +2,11 @@
 
 ## Versioning
 
+### 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
+
 ### v0.6.3
 
 * Adding gradient check in unit tests
@@ -62,4 +67,4 @@
 ### v0.1.0
 
 * Single GPU forward and backward transform
-* Minimal code example and notebook
+* Minimal code example and notebook

Dockerfile

@@ -30,9 +30,10 @@
 # build after cloning in directoy torch_harmonics via
 # docker build . -t torch_harmonics
 
-FROM nvcr.io/nvidia/pytorch:22.08-py3
+FROM nvcr.io/nvidia/pytorch:23.11-py3
 
 COPY . /workspace/torch_harmonics
 
-RUN pip install --use-feature=in-tree-build /workspace/torch_harmonics
+RUN pip install parameterized
+RUN pip install /workspace/torch_harmonics