upgrade torch support to 2.2

neuralmagic / sparseml

Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models

Apache License 2.0

2.08k stars 148 forks source link

Closed rahul-tuli closed 7 months ago

rahul-tuli commented 7 months ago

This PR upgrades torch support to 2.2.xx

Test plan, ran make test all tests pass

=============================== 494 passed, 4 skipped, 5 warnings in 14.85s ===============================