pytorch-lightning
Set `_DeviceDtypeModuleMixin` _device from torch's default device function
#21164
Merged
Merged
Set `_DeviceDtypeModuleMixin` _device from torch's default device function #21164
Borda
merged
12 commits into
Lightning-AI:master
from
GdoongMathew:fix/trainer_init_module_device
fix: set `_DeviceDtypeModuleMixin` _device from torch's default devic…
e4c2ac54
Merge branch 'master' into fix/trainer_init_module_device
8cefd43a
Borda
approved these changes
on 2025-09-04
chlog
e249722f
add torch 2.2.2 or below compatibility
990e6611
bhimrazy
commented
on 2025-09-05
Update src/lightning/fabric/utilities/device_dtype_mixin.py
835c3864
add import for `_TORCH_GREATER_EQUAL_2_3
6af308f4
Merge branch 'master' into fix/trainer_init_module_device
dd12c596
Merge branch 'master' into fix/trainer_init_module_device
1433e9b0
fix: restore torch default dtype once test_submodules_context_device_…
c3235076
Merge remote-tracking branch 'origin/fix/trainer_init_module_device' …
9f612dad
Merge branch 'master' into fix/trainer_init_module_device
0e729d61
Merge branch 'master' into fix/trainer_init_module_device
1eec61b2
bhimrazy
approved these changes
on 2025-09-08
Borda
approved these changes
on 2025-09-08
Reviewers
Assignees
No one assigned
Labels
fabric
pl
Milestone
No milestone
Login to write a write a comment.
Login via GitHub