tvm
tvm copied to clipboard
[Unity][Transform] Handle dynamic shapes in CombineParallelMatmul
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.