DeepSpeed
3e5658be - add api to set a function to run torch.compile

Commit

2 years ago

add api to set a function to run torch.compile

References

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

Author

Masahiro Tanaka

Parents

Loading