mosaicml / composer

Supercharge Your Model Training
http://docs.mosaicml.com
Apache License 2.0
5.06k stars 407 forks source link

Skip HSDP + TP pytests that require torch 2.3 or above #3426

Closed KuuCi closed 5 days ago

KuuCi commented 5 days ago

What does this PR do?

Some pytests require torch >= 2.3.0 but were not properly skipping. This pr simply checks the torch version to determine if the test should run

Before submitting