From 4523f8aa9346251630e98151ec4ebf0570430c6c Mon Sep 17 00:00:00 2001 From: Benson Ma Date: Tue, 9 Sep 2025 12:09:38 -0700 Subject: [PATCH] [fbgemm_gpu] Enable CUDA 13.0 for TBE in Nova - Enable CUDA 13.0 for TBE in Nova --- .github/workflows/build_wheels_genai_linux_aarch64.yml | 3 +-- .github/workflows/build_wheels_linux_x86.yml | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/.github/workflows/build_wheels_genai_linux_aarch64.yml b/.github/workflows/build_wheels_genai_linux_aarch64.yml index 186cabf32b..021007f36e 100644 --- a/.github/workflows/build_wheels_genai_linux_aarch64.yml +++ b/.github/workflows/build_wheels_genai_linux_aarch64.yml @@ -49,12 +49,11 @@ jobs: env: MAT: ${{ needs.generate-matrix.outputs.matrix }} # Nova Coordinate Filters: - # cuda/13.0-aarch64: Not supported in FBGEMM CI yet run: | set -ex pwd ls - MATRIX_BLOB="$( python .github/scripts/filter_nova_matrix.py --filter 'gpu_arch_version:13.0-aarch64' )" + MATRIX_BLOB="$( python .github/scripts/filter_nova_matrix.py )" echo "${MATRIX_BLOB}" echo "matrix=${MATRIX_BLOB}" >> "${GITHUB_OUTPUT}" diff --git a/.github/workflows/build_wheels_linux_x86.yml b/.github/workflows/build_wheels_linux_x86.yml index 9cdcbad82e..4a0667fa08 100644 --- a/.github/workflows/build_wheels_linux_x86.yml +++ b/.github/workflows/build_wheels_linux_x86.yml @@ -50,14 +50,13 @@ jobs: env: MAT: ${{ needs.generate-matrix.outputs.matrix }} # Nova Coordinate Filters: - # cuda/13.0: Not supported in FBGEMM CI yet # rocm/3.13t: causes segfaults at runtime # python/3.14: markupsafe is not available yet for python 3.14, so codegen breaks run: | set -ex pwd ls - MATRIX_BLOB="$( python .github/scripts/filter_nova_matrix.py --filter gpu_arch_version:13.0 --filter 'gpu_arch_type:rocm;python_version:3.13t' --filter 'python_version:3.14' )" + MATRIX_BLOB="$( python .github/scripts/filter_nova_matrix.py --filter 'gpu_arch_type:rocm;python_version:3.13t' --filter 'python_version:3.14' )" echo "${MATRIX_BLOB}" echo "matrix=${MATRIX_BLOB}" >> "${GITHUB_OUTPUT}"