From 298a5c02d0f5b00429ca1578a5364643ca74e37e Mon Sep 17 00:00:00 2001 From: Gulsum Gudukbay Akbulut Date: Wed, 11 Mar 2026 07:40:50 -0500 Subject: [PATCH] Remove ROCm fused-attention backend variables Removed ROCm specific environment variables for fused-attention. --- .github/workflows/run_tests_against_package.yml | 3 --- 1 file changed, 3 deletions(-) diff --git a/.github/workflows/run_tests_against_package.yml b/.github/workflows/run_tests_against_package.yml index 5aad97c100..ebbbfb2be9 100644 --- a/.github/workflows/run_tests_against_package.yml +++ b/.github/workflows/run_tests_against_package.yml @@ -86,9 +86,6 @@ jobs: ALLOW_MULTIPLE_LIBTPU_LOAD: ${{ inputs.device_type == 'cpu' && 'true' || '' }} # bypass /tmp/libtpu_lockfile check for cpu tests, which don't actually use accelerators (to allow concurrency) DECOUPLE_GCLOUD: ${{ inputs.decoupled_mode && 'TRUE' || '' }} LOCAL_GCLOUD_PROJECT: ${{ inputs.decoupled_mode && 'ci-decoupled' || '' }} - # ROCm: prefer CK fused-attention backend over AOTriton for stability. - NVTE_FUSED_ATTN_CK: ${{ inputs.device_type == 'rocm' && '1' || '' }} - NVTE_FUSED_ATTN_AOTRITON: ${{ inputs.device_type == 'rocm' && '0' || '' }} options: ${{ inputs.container_resource_option }} steps: - name: Checkout MaxText