Enable FP64 emulation for E2E AlbertForMaskedLM test on A770 (#1879)

Closes #1921 --------- Signed-off-by: Anatoly Myachev <[email protected]>
intel · Aug 23, 2024 · 87e5c06 · 87e5c06
1 parent a3f10d7
commit 87e5c06
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/.github/workflows/build-test-reusable.yml b/.github/workflows/build-test-reusable.yml
@@ -269,6 +269,15 @@ jobs:
           export WORKSPACE=$GITHUB_WORKSPACE
           cd pytorch
           pip install pyyaml pandas scipy numpy psutil pyre_extensions torchrec
+
+          source $GITHUB_WORKSPACE/scripts/capture-hw-details.sh
+          if [[ $GPU_DEVICE = *A770* ]]; then
+            # To enable FP64 type emulation.
+            # For details see: https://github.com/intel/intel-xpu-backend-for-triton/pull/1879
+            export OverrideDefaultFP64Settings=1
+            export IGC_EnableDPEmulation=1
+          fi
+
           # TODO: Find the fastest Hugging Face model
           $GITHUB_WORKSPACE/scripts/inductor_xpu_test.sh huggingface float32 inference accuracy xpu 0 static 1 0 AlbertForMaskedLM
           # The script above always returns 0, so we need an additional check to see if the accuracy test passed