MasterJH5574 commented on PR #14127: URL: https://github.com/apache/tvm/pull/14127#issuecomment-1447545072
Thanks for catching this and sorry for the late reply. Here the place where we turn scalars into Python native POD value is here https://github.com/apache/tvm/blob/ec868ebc58175431250c5f56af10749b77347f51/python/tvm/relax/transform/legalize_ops/common.py#L60 Considering this issue, I suppose changing the logic here to “create and return IntImm/FloatImm/...” instead of directly returning Python POD value will work. Would be great if you can help try and confirm this. -- 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]
