DeepSpeed
7b850d3d - Re-enable skipped unit tests (#3939)

Commit

1 year ago

Re-enable skipped unit tests (#3939) * fix skipped tests due to bad version check util function * revert local changes in EP size * remove added dummy test file * refactor torch version check function * Update util.py * fix for moe test memory leak --------- Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>

References

#3939 - Re-enable skipped unit tests

Author

mrwyattii

Parents

1bc3b784

Files13

tests/unit
- checkpoint
  - test_moe_checkpoint.py
  - test_zero_optimizer.py
- compression
  - test_compression.py
- model_parallelism
  - test_configurable_parallel_mp.py
  - test_configurable_parallel_pp.py
- moe
  - test_moe.py
  - test_moe_tp.py
- profiling/flops_profiler
  - test_flops_profiler.py
- runtime
  - half_precision
    - onebit
      - test_onebit.py
    - test_fp16.py
  - test_ds_initialize.py
  - zero
    - test_zero.py
- util.py

DeepSpeed 7b850d3d - Re-enable skipped unit tests (#3939)

DeepSpeed
7b850d3d - Re-enable skipped unit tests (#3939)