Can we use pin_memory and pin_memory_device of DataLoader with mps?

mps
1

I tried enabling memory pinning in a DataLoader via DataLoader(..., pin_memory=True, pin_memory_device="mps"), but the device is seemingly ignored and it defaults to CUDA.

2

Based on this code it seems mps is not a valid pin_memory_device argument.