From ada5f77f8d018d3282d910ce5e11b1547f26d18f Mon Sep 17 00:00:00 2001 From: Igor Tatarnikov Date: Thu, 9 May 2024 14:31:14 +0100 Subject: [PATCH] Added PYTORCH_MPS_HIGH_WATERMARK_RATION env variable --- .github/workflows/test_and_deploy.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/test_and_deploy.yml b/.github/workflows/test_and_deploy.yml index 0c41b4ae..a1f87e3f 100644 --- 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