DeepSpeed
7b4eb61b - Replace old torch version checks with existing function

Commit

2 years ago

Replace old torch version checks with existing function

References

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

Author

loadams

loadams

Parents

Loading