diff --git a/.github/workflows/_e2e_nightly_single_node.yaml b/.github/workflows/_e2e_nightly_single_node.yaml index 92345042628..5cd846ce454 100644 --- a/.github/workflows/_e2e_nightly_single_node.yaml +++ b/.github/workflows/_e2e_nightly_single_node.yaml @@ -143,10 +143,6 @@ jobs: run: | # ignore test_dispatch_ffn_combine until the test is fixed pytest -sv ${{ inputs.tests }} \ - --ignore=tests/e2e/nightly/ops/test_dispatch_ffn_combine.py \ - --ignore=tests/e2e/nightly/ops/test_fused_moe.py \ - --ignore=tests/e2e/nightly/ops/test_rotary_embedding.py \ - --ignore=tests/e2e/nightly/ops/test_matmul_allreduce_add_rmsnorm.py - + --ignore=tests/e2e/nightly/single_node/ops/singlecard_ops/test_fused_moe.py diff --git a/.github/workflows/nightly_test_a2.yaml b/.github/workflows/nightly_test_a2.yaml index 6e9bcc0d311..01a76b650a6 100644 --- a/.github/workflows/nightly_test_a2.yaml +++ b/.github/workflows/nightly_test_a2.yaml @@ -61,6 +61,9 @@ jobs: - name: test_custom_op os: linux-aarch64-a2-1 tests: tests/e2e/nightly/single_node/ops/singlecard_ops + - name: test_custom_op_multi_card + os: linux-aarch64-a2-4 + tests: tests/e2e/nightly/single_node/ops/multicard_ops_a2/ uses: ./.github/workflows/_e2e_nightly_single_node.yaml with: vllm: v0.13.0 diff --git a/.github/workflows/nightly_test_a3.yaml b/.github/workflows/nightly_test_a3.yaml index 8c4c5b4da85..74c762f15b0 100644 --- a/.github/workflows/nightly_test_a3.yaml +++ b/.github/workflows/nightly_test_a3.yaml @@ -158,9 +158,9 @@ jobs: fail-fast: false matrix: test_config: - - name: custom-op-dispatch_gmm_combine_decode + - name: custom-multi-ops os: linux-aarch64-a3-16 - tests: tests/e2e/nightly/single_node/ops/multicard_ops/test_dispatch_gmm_combine_decode.py + tests: tests/e2e/nightly/single_node/ops/multicard_ops_a3/ uses: ./.github/workflows/_e2e_nightly_single_node.yaml with: runner: ${{ matrix.test_config.os }} diff --git a/tests/e2e/nightly/single_node/ops/singlecard_ops/test_matmul_allreduce_add_rmsnorm.py b/tests/e2e/nightly/single_node/ops/multicard_ops_a2/test_matmul_allreduce_add_rmsnorm.py similarity index 100% rename from tests/e2e/nightly/single_node/ops/singlecard_ops/test_matmul_allreduce_add_rmsnorm.py rename to tests/e2e/nightly/single_node/ops/multicard_ops_a2/test_matmul_allreduce_add_rmsnorm.py diff --git a/tests/e2e/nightly/single_node/ops/multicard_ops/__init__.py b/tests/e2e/nightly/single_node/ops/multicard_ops_a3/__init__.py similarity index 100% rename from tests/e2e/nightly/single_node/ops/multicard_ops/__init__.py rename to tests/e2e/nightly/single_node/ops/multicard_ops_a3/__init__.py diff --git a/tests/e2e/nightly/single_node/ops/singlecard_ops/test_dispatch_ffn_combine.py b/tests/e2e/nightly/single_node/ops/multicard_ops_a3/test_dispatch_ffn_combine.py similarity index 100% rename from tests/e2e/nightly/single_node/ops/singlecard_ops/test_dispatch_ffn_combine.py rename to tests/e2e/nightly/single_node/ops/multicard_ops_a3/test_dispatch_ffn_combine.py diff --git a/tests/e2e/nightly/single_node/ops/multicard_ops/test_dispatch_gmm_combine_decode.py b/tests/e2e/nightly/single_node/ops/multicard_ops_a3/test_dispatch_gmm_combine_decode.py similarity index 100% rename from tests/e2e/nightly/single_node/ops/multicard_ops/test_dispatch_gmm_combine_decode.py rename to tests/e2e/nightly/single_node/ops/multicard_ops_a3/test_dispatch_gmm_combine_decode.py