DeepSpeed/tests
Ramya Ramineni d24629f4fd
[ROCm] Enable TestCUDABackward::test_backward unit tests (#3849)
* Workaround to pass unit/ops/accelerators/test_accelerator_backward.py unit tests on ROCm

* Rearranged is_rocm_pytorch()

* Introduced is_rocm_pytorch() for ROCm

* Fixed formatting errors

* Function call

* formatting fix

---------

Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
Co-authored-by: Logan Adams <loadams@microsoft.com>
2023-07-07 00:56:19 +00:00
..
accelerator Update DeepSpeed copyright license to Apache 2.0 (#3111) 2023-03-30 17:14:38 -07:00
benchmarks Update DeepSpeed copyright license to Apache 2.0 (#3111) 2023-03-30 17:14:38 -07:00
hybrid_engine DeepSpeed Chat (#3186) 2023-04-11 11:53:38 -07:00
lightning Update DeepSpeed copyright license to Apache 2.0 (#3111) 2023-03-30 17:14:38 -07:00
model Revert "Update megatron GPT2Model" 2023-04-05 15:12:01 -07:00
onebit Update DeepSpeed copyright license to Apache 2.0 (#3111) 2023-03-30 17:14:38 -07:00
perf Update DeepSpeed copyright license to Apache 2.0 (#3111) 2023-03-30 17:14:38 -07:00
small_model_debugging ZeRO++ (#3784) 2023-06-23 14:30:49 -07:00
unit [ROCm] Enable TestCUDABackward::test_backward unit tests (#3849) 2023-07-07 00:56:19 +00:00
conftest.py Revert "Prevent hangs in CI during parallel run compilation (#2844)" (#3817) 2023-06-26 10:10:12 -07:00
pytest.ini Re-enable GPT-J unit tests and refactor inference tests (#3618) 2023-06-28 19:33:55 +00:00