Commit c0bb1b5
committed
Fix macOS MPS compatibility in test server startup
Add PYTORCH_ENABLE_MPS_FALLBACK=1 environment variable when starting
test servers to allow PyTorch operations to fall back to CPU when
MPS (Metal Performance Shaders) doesn't support them.
This fixes test failures on macOS with Apple Silicon where operations
like aten::linalg_cholesky_ex.L are not implemented for MPS device.1 parent 8b2e035 commit c0bb1b5
2 files changed
Lines changed: 4 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
94 | 94 | | |
95 | 95 | | |
96 | 96 | | |
| 97 | + | |
| 98 | + | |
97 | 99 | | |
98 | 100 | | |
99 | 101 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
42 | 42 | | |
43 | 43 | | |
44 | 44 | | |
| 45 | + | |
| 46 | + | |
45 | 47 | | |
46 | 48 | | |
47 | 49 | | |
| |||
0 commit comments