Skip to content

Commit f9b9f72

Browse files
committed
ci: exclude splash_attention kernel tests from CI due to JAX/libtpu incompatibility
1 parent 81afe73 commit f9b9f72

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

.github/workflows/UnitTests.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ jobs:
5959
- name: PyTest
6060
run: | #--deselect=src/maxdiffusion/tests/input_pipeline_interface_test.py
6161
export LIBTPU_INIT_ARGS='--xla_tpu_scoped_vmem_limit_kib=65536'
62-
HF_HUB_CACHE=/mnt/disks/github-runner-disk/ HF_HOME=/mnt/disks/github-runner-disk/ TOKENIZERS_PARALLELISM=false python3 -m pytest --deselect=src/maxdiffusion/tests/ltx_transformer_step_test.py -x
62+
HF_HUB_CACHE=/mnt/disks/github-runner-disk/ HF_HOME=/mnt/disks/github-runner-disk/ TOKENIZERS_PARALLELISM=false python3 -m pytest --deselect=src/maxdiffusion/tests/ltx_transformer_step_test.py --ignore=src/maxdiffusion/kernels/splash_attention -x
6363
# add_pull_ready:
6464
# if: github.ref != 'refs/heads/main'
6565
# permissions:

0 commit comments

Comments
 (0)