zxybazh commented on PR #14127: URL: https://github.com/apache/tvm/pull/14127#issuecomment-1447663997
Thanks @MasterJH5574 and @sunggg, I think Ruihang's suggestion makes sense and we can preserve the data type via IntImm and FloatImm. I also added an extra argument of whether to return native python type, because in some cases, e.g. `full` operator, we may want to get the python native value. This should also work with most of the current tests. For Sung's concern of preserving float type, please see the first added unit test where fp16 can be preserved. I think `fp64` can be preserved in the same way. Please take another look at your convenience, thanks! -- 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]
