search

sktime / pytorch-forecasting

Time series forecasting with PyTorch
https://pytorch-forecasting.readthedocs.io/
MIT License
3.99k stars 631 forks source link

[MNT] Test mps #1654

Closed XinyuWuu closed 1 month ago

XinyuWuu commented 2 months ago

Test MPS in Github runner.

fkiraly commented 2 months ago

:-( seems not to work (yet)

XinyuWuu commented 2 months ago

:-( seems not to work (yet)

Yeah, I could not find a way to disable MPS. More specificly make this line return False:

https://github.com/pytorch/pytorch/blob/main/torch/backends/mps/__init__.py#L25

Underlying probably here: https://github.com/pytorch/pytorch/blob/main/aten/src/ATen/mps/MPSDevice.mm#L77-L103

XinyuWuu commented 1 month ago

Fixed in https://github.com/sktime/pytorch-forecasting/pull/1648