diff --git a/.ci b/.ci index c6bf3697..ab56f544 160000 --- a/.ci +++ b/.ci @@ -1 +1 @@ -Subproject commit c6bf369739f16e759f46fd466f586c51c59f26c3 +Subproject commit ab56f5444799953e29bf495216b6201ef0448213 diff --git a/.github/workflows/ci_test.yml b/.github/workflows/ci_test.yml index 73672578..b22278e4 100644 --- a/.github/workflows/ci_test.yml +++ b/.github/workflows/ci_test.yml @@ -27,10 +27,10 @@ on: jobs: ci: - uses: InfiniTensor/ci/.github/workflows/infiniops-ci.yml@c6bf369739f16e759f46fd466f586c51c59f26c3 + uses: InfiniTensor/ci/.github/workflows/infiniops-ci.yml@ab56f5444799953e29bf495216b6201ef0448213 with: config_path: .github/ci_config.yml - ci_ref: c6bf369739f16e759f46fd466f586c51c59f26c3 + ci_ref: ab56f5444799953e29bf495216b6201ef0448213 max_parallel: 10 platform: ${{ github.event_name == 'workflow_dispatch' && (inputs.platform == 'all' && 'nvidia,iluvatar,metax,moore,cambricon,ascend' || inputs.platform) || 'nvidia,iluvatar,metax,moore,cambricon,ascend' }} secrets: inherit diff --git a/.github/workflows/ci_v2_shadow.yml b/.github/workflows/ci_v2_shadow.yml index 0653419a..b2c7bec3 100644 --- a/.github/workflows/ci_v2_shadow.yml +++ b/.github/workflows/ci_v2_shadow.yml @@ -28,10 +28,10 @@ on: jobs: ci-v2-shadow: - uses: InfiniTensor/ci/.github/workflows/infiniops-ci-v2-shadow.yml@c6bf369739f16e759f46fd466f586c51c59f26c3 + uses: InfiniTensor/ci/.github/workflows/infiniops-ci-v2-shadow.yml@ab56f5444799953e29bf495216b6201ef0448213 with: config_path: .github/ci_config.yml - ci_ref: c6bf369739f16e759f46fd466f586c51c59f26c3 + ci_ref: ab56f5444799953e29bf495216b6201ef0448213 max_parallel: 10 platform: ${{ github.event_name == 'workflow_dispatch' && (inputs.platform == 'all' && 'nvidia,iluvatar,metax,moore,cambricon,ascend' || inputs.platform == 'active' && 'nvidia,iluvatar,metax,moore,cambricon,ascend' || inputs.platform) || 'nvidia,iluvatar,metax,moore,cambricon,ascend' }} secrets: inherit diff --git a/tests/test_torch_ops.py b/tests/test_torch_ops.py index bc2a3337..2f28b24f 100644 --- a/tests/test_torch_ops.py +++ b/tests/test_torch_ops.py @@ -208,7 +208,11 @@ def _list_default(aten_type): _VENDOR_CRASH_OPS = frozenset( { ("npu", "mish"), + ("npu", "mse_loss"), + ("npu", "nonzero"), ("npu", "nuclear_norm"), + ("npu", "smooth_l1_loss"), + ("npu", "soft_margin_loss"), ("npu", "_linalg_svd"), ("npu", "svd"), }