fix: add MPS device support#3763

Closed

Aanushka001 wants to merge 2 commits into

pytorch:masterfrom

Aanushka001:fix-mps-device

Commits on May 31, 2026

Update auto_dataloader pin_memory logic and docstring to include MPS device support
Aanushka001
committed
Add empty_mps_cache function and MPS cache clearing handler
Aanushka001
committed