DeepSpeed
ca419b4d - disable memory_efficient_linear when torch.compile is enabled

Commit

2 years ago

disable memory_efficient_linear when torch.compile is enabled

References

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

Author

Masahiro Tanaka

Parents

Loading