Error in the The operator 'aten::_linalg_solve_ex.result' in the aug_transforms function in the Data Augmentation

[Akhil-Raj]

https://github.com/fastai/fastbook/blob/823b69e00aa1e1c1a45fe88bd346f11e8f89c1ff/02_production.ipynb#LL929C5-L931C58

In the lines highlighted above(See the 'Code' tab of .ipynb), I was getting this error :

NotImplementedError: The operator 'aten::_linalg_solve_ex.result' is not currently implemented for the MPS device. If you want this op to be added in priority during the prototype phase of this feature, please comment on pytorch/pytorch#77764. As a temporary fix, you can set the environment variable PYTORCH_ENABLE_MPS_FALLBACK=1 to use the CPU as a fallback for this op. WARNING: this will be slower than running natively on MPS.

A temporary solution is also mentioned in the issue : https://forums.fast.ai/t/lesson-2-troubleshoot-macbook-m1-issue/105584

Is there any permanent solution to it?

[gamedevCloudy]

The issue seems to be in:
Warp() function in fastai.vision.augment

Thus the workaround: other augments works by creating custom_augments that does not include Warp.

Including Warp() in custom augment gives same error.