DeepSpeed
ab1f32de - Update skip on torch in tests (#3136)

Commit

2 years ago

Update skip on torch in tests (#3136) * Replace old torch version checks with existing function * Clean up formatting

References

#3136 - Fixes code and tests skipping/asserting incorrectly on torch 2+.

Author

loadams

loadams

Parents

Files9

deepspeed/runtime
- comm
  - nccl.py
- fp16/onebit
  - adam.py
  - lamb.py
  - zoadam.py
tests/unit
- compression
  - test_compression.py
- model_parallelism
  - test_configurable_parallel_mp.py
  - test_configurable_parallel_pp.py
- profiling/flops_profiler
  - test_flops_profiler.py
- runtime/half_precision/onebit
  - test_onebit.py