sparseml
sparseml copied to clipboard
Draft: Supporting torch 1.12
Todos:
- Guard specific torch function calls in torch version checks:
- [ ] fuse_modules
- [ ] onnx.export
- [ ] Get all the unit tests passing (there are some that still fill due to onnx node name changes)
@bfineran @dbogunowicz assigned for review
I'm going to be breaking the parts of this PR that can easily be merged into multiple smaller PRs.
Closing this for now - will reopen when work resumes