File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -63,6 +63,7 @@ python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR/pytest_test_linear_mxfp8_att
6363python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR /pytest_test_fused_mla_q_uproj.xml $TE_PATH /tests/pytorch/attention/test_fused_mla_q_uproj.py || test_fail " test_fused_mla_q_uproj.py"
6464python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR /pytest_test_kv_cache.xml $TE_PATH /tests/pytorch/attention/test_kv_cache.py || test_fail " test_kv_cache.py"
6565python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR /pytest_test_cu_seqlens_cache.xml $TE_PATH /tests/pytorch/attention/test_cu_seqlens_cache.py || test_fail " test_cu_seqlens_cache.py"
66+ python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR /pytest_test_dpa_mla_qkv_head_dim_pad.xml $TE_PATH /tests/pytorch/attention/test_dpa_mla_qkv_head_dim_pad.py || test_fail " test_dpa_mla_qkv_head_dim_pad.py"
6667python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR /pytest_test_hf_integration.xml $TE_PATH /tests/pytorch/test_hf_integration.py || test_fail " test_hf_integration.py"
6768export NVTE_TEST_CHECKPOINT_ARTIFACT_PATH=$TE_PATH /artifacts/tests/pytorch/test_checkpoint
6869if [ ! -d " $NVTE_TEST_CHECKPOINT_ARTIFACT_PATH " ]; then
You can’t perform that action at this time.
0 commit comments