[BUG] Pytorch Extension side Segmentation Fault

[DandinPower]

All SPMM strategies (including the built-in extension version) have the same problem. Assume I have matrix_1 (row1, col1) and matrix_2 (row2, col1); if col1 is smaller than row1, it will cause Segmentation Fault or memory Aborted.