trevor-m opened a new pull request #5018: Look for TupleType instead of TupleNode in LayoutRewriter URL: https://github.com/apache/incubator-tvm/pull/5018 When getting the input shapes from the args LayoutRewriter checked for tuples only by seeing if the node is an instance of TupleNOde. With that code, the following relay program would encounter an error during LayoutRewriter because the arg is a relay.Var but has a TupleTypeNode instead of a TensorTypeNode as it expected. ``` def @main(%input_0: Tensor[(1, 2, 6, 6), float32], %input_1: Tensor[(1, 3, 6, 6), float32]) -> Tensor[(1, 5, 6, 6), float32] { %0 = (%input_0, %input_1); %1 = fn (%x: (Tensor[(1, 2, 6, 6), float32], Tensor[(1, 3, 6, 6), float32])) -> Tensor[(1, 5, 6, 6), float32] { concatenate(%x, axis=1) /* ty=Tensor[(1, 5, 6, 6), float32] */ }; %1(%0) /* ty=Tensor[(1, 5, 6, 6), float32] */ } ``` @zhiics @anijain2305 Could you please review?
---------------------------------------------------------------- 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. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
