pyjhzwh opened a new pull request #4984: Fix stride default value None in torch.nn.functional.avg_pool URL: https://github.com/apache/incubator-tvm/pull/4984 The default value of stride in nn.functional.avg_pool is set as None, https://github.com/apache/incubator-tvm/blob/5a0f39b5481a30a2eec49e27cbc17a722bd6ee6a/python/tvm/relay/frontend/pytorch.py#L461 It will cause `TVMError: Check failed: ObjectTypeChecker<TObjectRef>: :Check(ptr): Expect relay.Expr but get Array` since `inputs[2]` is empty. Fix it by detecting None type of stride in tvm frontend pytorch.
---------------------------------------------------------------- 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: us...@infra.apache.org With regards, Apache Git Services