Lunderberg opened a new pull request, #15883: URL: https://github.com/apache/tvm/pull/15883
Prior to this commit, the `relax.transform.FuseTIR` transform required that the shapes arguments passed into a `PrimFunc` be structurally equivalent to the shapes of the parameters, and that any replacement of symbolic `tir.Var` be with a symbolic `tir.Var` in the fused function. This commit updates the `SymbolicMatcher` to instead extract a `Map<tir::Var, PrimExpr>`. As a result, a Relax tensor with statically-known shape can be passed into a TIR PrimFunc with dynamic shape. The resulting fused TIR function is in terms of the statically-known shape, and no longer contains the symbolic variable. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
