Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
Add missing device id for pytorch 2.8 (#21105)
* add missing device id for pytorch 2.8 * skip device id for older pytorch versions * add testing * fix mypy without touching submodule * fix failing tests * Apply suggestions from code review --------- Co-authored-by: Jirka Borovec <[email protected]> Co-authored-by: Deependu <[email protected]> Co-authored-by: Jirka B <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
N
Nicki Skafte Detlefsen committed
6fc44c9254e9ab2c4e49643f34f1c18d05c36e91
Parent: 3d56296
Committed by GitHub <[email protected]>
on 9/1/2025, 1:16:17 PM