Add PYTORCH_MPS_HIGH_WATERMARK_RATIO=0.0

bips-hb · Dec 20, 2023 · 814da93 · 814da93
1 parent 3e4639c
commit 814da93
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/R-CMD-check.yaml b/.github/workflows/R-CMD-check.yaml
@@ -38,6 +38,7 @@ jobs:
       TORCH_INSTALL: 1
       TORCH_COMMIT_SHA: "none"
       PYTORCH_ENABLE_MPS_FALLBACK: 1
+      PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.0
 
     steps:
       - uses: actions/checkout@v3