Bug when indexing 2D tensors using an MPS device

[joaquimgomez]

Bug when indexing 2D tensors using an MPS device

[ayush0x00]

Hey @joaquimgomez I think the issue is fixed on torch==2.3.0.dev20240212 It's working fine on my machine.

[malfet]

@ayush0x00 are you on Intel or Apple Silicon Mac? It works my M2, but it's very likely broken on Intel machines

[ayush0x00]

@ayush0x00 are you on Intel or Apple Silicon Mac? It works my M2, but it's very likely broken on Intel machines

I am using it on Apple silicon M1. Not sure about intel machines then. Any way of verifying it on intel?

[ayush0x00]

@joaquimgomez could you verify if it works when using device=torch.device("cpu")?

[joaquimgomez]

I tested it on a Linux machine with an Intel Xeon E5645 and it works.

In my M2 Max, if device=torch.device("cpu") it works @ayush0x00 . It also works if x is in MPS but mask is in the CPU. I did the tests with 2.3.0 and 2.4.0a0+git94b328e and it is still not working well when both are in MPS.

[ayush0x00]

@joaquimgomez I guess Apple has discontinued its support for Intel GPUs. Have a look at this thread. So unless you are working on Apple silicon, I think device(mps) won't work.

[joaquimgomez]

No, sorry for the misunderstanding. I'm working on an Apple Silicon, a MacBook Pro 14'' 2023 with an M2 Max.

[ayush0x00]

@joaquimgomez maybe try updating torch. It works fine on my mac, which runs M1 silicon.

[joaquimgomez]

Tried to update torch without compiling it by myself and it works now. Thanks.