Before submitting a pull request (PR), please read the contributing guide.
Please fill out as much of this template as you can, but if you have any problems or questions, just leave a comment and we will help out :)
Description
What is this PR
[x] Bug fix
[ ] Addition of a new feature
[ ] Other
Why is this PR needed?
Some functions are not currently available on MPS, currently PyTorch will throw an error and request a flag be set in env variables (PYTORCH_MPS_FALLBACK).
What does this PR do?
Sets the PYTORCH_MPS_FALLBACK flag as the default behaviour.
How has this PR been tested?
Tested locally.
Is this a breaking change?
No
Does this PR require an update to the documentation?
No
Checklist:
[x] The code has been tested locally
[ ] Tests have been added to cover all new functionality (unit & integration)
[ ] The documentation has been updated to reflect any changes
Before submitting a pull request (PR), please read the contributing guide.
Please fill out as much of this template as you can, but if you have any problems or questions, just leave a comment and we will help out :)
Description
What is this PR
Why is this PR needed? Some functions are not currently available on MPS, currently PyTorch will throw an error and request a flag be set in env variables (PYTORCH_MPS_FALLBACK).
What does this PR do? Sets the
PYTORCH_MPS_FALLBACKflag as the default behaviour.How has this PR been tested?
Tested locally.
Is this a breaking change?
No
Does this PR require an update to the documentation?
No
Checklist: