tvm icon indicating copy to clipboard operation
tvm copied to clipboard

[Unity][Transform] Handle dynamic shapes in CombineParallelMatmul

Open Lunderberg opened this issue 1 year ago • 0 comments
trafficstars

Prior to this commit, if the weight of a matmul a dynamic shape, and that matmul is being combined with the CombineParallelMatmul pass, it could cause a segfault when dim.as<IntImmNode>() returns a null pointer.

This commit adds explicit test cases for these dynamic shapes, and updates CombineParallelMatmul to handle the dynamic shapes.

Lunderberg avatar Feb 16 '24 17:02 Lunderberg