Wenxuan Tan ec73f1b5e2 [CI] Cleanup Dist Optim tests with shared helper funcs (#6125) ...

* Refractor and cleanup using common helper funcs. Tests passed

* Update comments

* Fix relative import

* Fix param fetching bug

2025-02-12 13:42:34 +08:00

..

_utils.py

[CI] Cleanup Dist Optim tests with shared helper funcs (#6125)

2025-02-12 13:42:34 +08:00

test_adam_kernel.py

[misc] refactor launch API and tensor constructor (#5666)

2024-04-29 10:40:11 +08:00

test_adam_optim.py

[CI] Cleanup Dist Optim tests with shared helper funcs (#6125)

2025-02-12 13:42:34 +08:00

test_dist_adafactor.py

[CI] Cleanup Dist Optim tests with shared helper funcs (#6125)

2025-02-12 13:42:34 +08:00

test_dist_came.py

[CI] Cleanup Dist Optim tests with shared helper funcs (#6125)

2025-02-12 13:42:34 +08:00

test_dist_galore.py

[CI] Cleanup Dist Optim tests with shared helper funcs (#6125)

2025-02-12 13:42:34 +08:00

test_dist_lamb.py

[CI] Cleanup Dist Optim tests with shared helper funcs (#6125)

2025-02-12 13:42:34 +08:00

test_lr_scheduler.py

[lr-scheduler] fix load state dict and add test (#5369)

2024-02-06 14:23:32 +08:00

test_nvme.py

[devops] remove post commit ci (#5566)

2024-04-08 15:09:40 +08:00