NotImplementedError: The operator 'aten::_linalg_solve_ex.result' is not currently implemented for the MPS device. Available in nightly for CPU. #151189
Description
🐛 Describe the bug
The HiDream-i1 model (currently among the most advanced open-source AI image generators) was released on Hugging Face this week (April 8, 2025). While the full model fails to run on an RTX 5090, I’ve successfully managed to load all components on my M1, with some minor code modifications to their open source project which I will contribute if I am able to have a fully functional mac implementation.
Initially, I ran into issues due to an unimplemented method—this has since been addressed in the latest nightly builds. However, when attempting to generate an image, I now encounter the following error:
NotImplementedError: The operator 'aten::_linalg_solve_ex.result' is not currently implemented for the MPS device. Available in nightly for CPU.
I would really like to get this working with MPS, as it will allow this model as well as others to run efficiently on mac hardware.
Appreciate you efforts.
Regards,
Peter
Versions
PyTorch version: 2.6.0
Is debug build: False
CUDA used to build PyTorch: None
ROCM used to build PyTorch: N/A
OS: macOS 15.3.2 (arm64)
GCC version: Could not collect
Clang version: 17.0.0 (clang-1700.0.13.3)
CMake version: version 3.29.5
Libc version: N/A
Python version: 3.10.17 | packaged by conda-forge | (main, Apr 10 2025, 22:23:34) [Clang 18.1.8 ] (64-bit runtime)
Python platform: macOS-15.3.2-arm64-arm-64bit
Is CUDA available: False
CUDA runtime version: No CUDA
CUDA_MODULE_LOADING set to: N/A
GPU models and configuration: No CUDA
Nvidia driver version: No CUDA
cuDNN version: No CUDA
HIP runtime version: N/A
MIOpen runtime version: N/A
Is XNNPACK available: True
CPU:
Apple M1 Ultra
Versions of relevant libraries:
[pip3] numpy==2.2.4
[pip3] torch==2.6.0
[pip3] torchaudio==2.6.0
[pip3] torchvision==0.21.0
[conda] numpy 2.2.4 pypi_0 pypi
[conda] torch 2.6.0 pypi_0 pypi
[conda] torchaudio 2.6.0 pypi_0 pypi
[conda] torchvision 0.21.0 pypi_0 pypi
cc @jianyuh @nikitaved @mruberry @walterddr @xwang233 @lezcano @kulinseth @albanD @malfet @DenisVieriu97 @jhavukainen