Added PYTORCH_MPS_HIGH_WATERMARK_RATION env variable

brainglobe · May 9, 2024 · ada5f77 · ada5f77
1 parent 933e5dd
commit ada5f77
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/test_and_deploy.yml b/.github/workflows/test_and_deploy.yml
@@ -39,6 +39,7 @@ jobs:
     runs-on: ${{ matrix.os }}
     env:
       PYTORCH_ENABLE_MPS_FALLBACK: "1"
+      PYTORCH_MPS_HIGH_WATERMARK_RATIO: "0.0"
     strategy:
       matrix:
         # Run all supported Python versions on linux