HELSON a1ce02d740 [zero] test gradient accumulation (#1964) ...

* [zero] fix memory leak for zero2

* [zero] test gradient accumulation

* [zero] remove grad clip test

2022-11-29 13:00:30 +08:00

..

__init__.py

[test] skip tests when not enough GPUs are detected (#1090)

2022-06-09 17:19:13 +08:00

comparison.py

[NFC] polish colossalai/testing/comparison.py code style. (#1558)

2022-09-08 22:11:04 +08:00

pytest_wrapper.py

[unittest] added doc for the pytest wrapper (#1704)

2022-10-14 10:56:17 +08:00

random.py

[zero] test gradient accumulation (#1964)

2022-11-29 13:00:30 +08:00

utils.py

[unittest] supported condititonal testing based on env var (#1701)

2022-10-13 19:38:45 +08:00