DeepSpeed
c2ba829d - update DistributedTest to work with torch.compile tests

Commit

1 year ago

update DistributedTest to work with torch.compile tests

References

#4878 - Enable torch.compile with ZeRO (Experimental)

Author

mrwyattii

mrwyattii

Parents

Files5

tests/unit
- common.py
- runtime/compile
  - test_compile_wrapper.py
  - test_compile_zero.py
  - test_load_config.py
  - util.py