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