On my M1 MacBook Pro it works perfectly on the CPU, but I just get noise if I try and use the hardware accelerated MPS device.
CPU:
MPS:
I assume this is really a PyTorch/MPS issue rather than with your code, but I wonder if you can shed any light on what the problem might be?
There are no errors indicating usage of PyTorch features unimplemented for Apple hardware, and unsurprisingly using PYTORCH_ENABLE_MPS_FALLBACK=1 has no effect.
Hi - nice work btw!
On my M1 MacBook Pro it works perfectly on the CPU, but I just get noise if I try and use the hardware accelerated MPS device.
CPU:
MPS:
I assume this is really a PyTorch/MPS issue rather than with your code, but I wonder if you can shed any light on what the problem might be?
There are no errors indicating usage of PyTorch features unimplemented for Apple hardware, and unsurprisingly using
PYTORCH_ENABLE_MPS_FALLBACK=1
has no effect.