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 <6035284+Borda@users.noreply.github.com>
Co-authored-by: Deependu <deependujha21@gmail.com>
Co-authored-by: Jirka B <j.borovec+github@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
