Skip to content

Commit 77372cf

Browse files
Remove ROCm fused-attention backend variables
Removed ROCm specific environment variables for fused-attention.
1 parent 86ebc19 commit 77372cf

1 file changed

Lines changed: 0 additions & 3 deletions

File tree

.github/workflows/run_tests_against_package.yml

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -105,9 +105,6 @@ jobs:
105105
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)
106106
DECOUPLE_GCLOUD: ${{ inputs.decoupled_mode && 'TRUE' || '' }}
107107
LOCAL_GCLOUD_PROJECT: ${{ inputs.decoupled_mode && 'ci-decoupled' || '' }}
108-
# ROCm: prefer CK fused-attention backend over AOTriton for stability.
109-
NVTE_FUSED_ATTN_CK: ${{ inputs.device_type == 'rocm' && '1' || '' }}
110-
NVTE_FUSED_ATTN_AOTRITON: ${{ inputs.device_type == 'rocm' && '0' || '' }}
111108
options: ${{ inputs.container_resource_option }}
112109
steps:
113110
- name: Checkout MaxText

0 commit comments

Comments
 (0)