DeepSpeed
2a0c0e3c - Remove compile wrapper to simplify access to model attributes (#5581)

Commit

1 year ago

Remove compile wrapper to simplify access to model attributes (#5581) Having a wrapper of a compiled module brings various restrictions about accessing attributes of the compiled model. This PR removes the wrapper of compiled module to simplify the access to the compiled model. --------- Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>

References

#5581 - Remove compile wrapper to simplify access to model attributes

Author

tohtana

Parents

8831b57b

Files9

deepspeed/runtime
- compiler.py
- config.py
- engine.py
- zero
  - partition_parameters.py
tests/unit
- common.py
- runtime/compile
  - test_compile_wrapper.py
  - test_compile_zero.py
  - test_load_config.py
  - util.py

DeepSpeed 2a0c0e3c - Remove compile wrapper to simplify access to model attributes (#5581)

DeepSpeed
2a0c0e3c - Remove compile wrapper to simplify access to model attributes (#5581)