Skip to content

Commit 62b9b2d

Browse files
Remove ROCm fused-attention backend variables
Removed ROCm specific environment variables for fused-attention.
1 parent cec342e commit 62b9b2d

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
@@ -102,9 +102,6 @@ jobs:
102102
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)
103103
DECOUPLE_GCLOUD: ${{ inputs.decoupled_mode && 'TRUE' || '' }}
104104
LOCAL_GCLOUD_PROJECT: ${{ inputs.decoupled_mode && 'ci-decoupled' || '' }}
105-
# ROCm: prefer CK fused-attention backend over AOTriton for stability.
106-
NVTE_FUSED_ATTN_CK: ${{ inputs.device_type == 'rocm' && '1' || '' }}
107-
NVTE_FUSED_ATTN_AOTRITON: ${{ inputs.device_type == 'rocm' && '0' || '' }}
108105
options: ${{ inputs.container_resource_option }}
109106
steps:
110107
- name: Checkout MaxText

0 commit comments

Comments
 (0)