Skip to content

test: add SDDMM/SPMM Ascend NPU test suite - #19

Open
xuejiakn wants to merge 1 commit into
BUPT-GAMMA:masterfrom
xuejiakn:test/sddmm-spmm
Open

test: add SDDMM/SPMM Ascend NPU test suite#19
xuejiakn wants to merge 1 commit into
BUPT-GAMMA:masterfrom
xuejiakn:test/sddmm-spmm

Conversation

@xuejiakn

Copy link
Copy Markdown

Summary

Add Ascend NPU test suite for SDDMM/SPMM operators.

Changes

  • tests/python/ascend/init.py
  • tests/python/ascend/test_sddmm_spmm.py: comprehensive SDDMM/SPMM tests

Restructure note

Merges the two duplicate test commits #14[5] and #14[6] (same subject, same file) plus #15[8] into a single clean commit. Depends on the SDDMM (#2) and SPMM (#3) PRs for the tests to pass.

- 51 test cases covering:
  - SDDMM: dot/add/mul/copy_lhs/copy_rhs × COO/CSR × FP32/FP16 × int32/int64
  - SPMM: copy_lhs + sum/max/min × int32/int64, update_all, backward
  - UDF reduce (degree bucketing) + backward
  - Stream sync: in_edges/in_degrees after update_all/gsddmm
  - Repeated update_all (multi-step stability)
  - segment_reduce (mean_nodes) forward + backward
  - End-to-end: GCN forward/backward, multi-layer GCN, training loop
- 4 skipped: sub/div CPU fallback (linker limitation)
- 3 xfailed: FP16 SDDMM (intermittent NPU stream race, passes in isolation)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant