Run CI for bf16 with custom SDPA (#5554)

Summary: Pull Request resolved: #5554 These configs should work now. bypass-github-export-checks Reviewed By: mergennachin Differential Revision: D63158947 fbshipit-source-id: 92878bb799b311a857db6e12cb6103835514e998
pytorch · Oct 16, 2024 · cccc2c2 · cccc2c2
1 parent 8673567
commit cccc2c2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/.github/workflows/pull.yml b/.github/workflows/pull.yml
@@ -98,6 +98,12 @@ jobs:
           - dtype: bf16
             build-tool: buck2
             mode: portable
+          - dtype: bf16
+            build-tool: cmake
+            mode: custom
+          - dtype: bf16
+            build-tool: buck2
+            mode: custom
       fail-fast: false
     with:
       runner: linux.2xlarge

diff --git a/.github/workflows/trunk.yml b/.github/workflows/trunk.yml
@@ -227,6 +227,8 @@ jobs:
         include:
           - dtype: bf16
             mode: portable
+          - dtype: bf16
+            mode: custom
       fail-fast: false
     with:
       runner: macos-m1-stable