Commit 54f4090
authored
Patching refactor (#1450)
* New tests for patching utilities in patching.py
Signed-off-by: Charlelie Laurent <claurent@nvidia.com>
* Refactored patching.py
Signed-off-by: Charlelie Laurent <claurent@nvidia.com>
* Fixed global_index reference in patching.py
Signed-off-by: Charlelie Laurent <claurent@nvidia.com>
* Added docstring examples + recompile tests for patching.py
Signed-off-by: Charlelie Laurent <claurent@nvidia.com>
* Changed tests to fullgraph=True, and some cosmetic imrovements in patching.py
Signed-off-by: Charlelie Laurent <claurent@nvidia.com>
* Added TODO comment in patching.py
Signed-off-by: Charlelie Laurent <claurent@nvidia.com>
---------
Signed-off-by: Charlelie Laurent <claurent@nvidia.com>1 parent 04b15d8 commit 54f4090
File tree
2 files changed
+402
-405
lines changed- physicsnemo/diffusion/multi_diffusion
- test/diffusion
2 files changed
+402
-405
lines changed
0 commit comments