This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 604b263dd5 [BugFix] [Relay][Pytorch] Fix missing `.dtype` (#16167)
604b263dd5 is described below
commit 604b263dd5834f8bd96dc1876f4bad74100f692b
Author: Masahiro Hiramori <[email protected]>
AuthorDate: Tue Nov 28 00:09:37 2023 -0800
[BugFix] [Relay][Pytorch] Fix missing `.dtype` (#16167)
Fix missing `.dtype`
---
python/tvm/relay/frontend/pytorch.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/python/tvm/relay/frontend/pytorch.py
b/python/tvm/relay/frontend/pytorch.py
index 9374a24912..30711de0a7 100644
--- a/python/tvm/relay/frontend/pytorch.py
+++ b/python/tvm/relay/frontend/pytorch.py
@@ -847,7 +847,7 @@ class PyTorchOpConverter:
dtype = _convert_dtype_value(inputs[2])
else:
# if dtype is None, use the dtype of the input tensor
- dtype = self.infer_type(inputs[0])
+ dtype = self.infer_type(inputs[0]).dtype
return self.full_impl(data, 0, dtype)
def full(self, inputs, input_types):
@@ -898,7 +898,7 @@ class PyTorchOpConverter:
dtype = _convert_dtype_value(inputs[3])
else:
# if dtype is None, use the dtype of the input tensor
- dtype = self.infer_type(inputs[0])
+ dtype = self.infer_type(inputs[0]).dtype
return self.full_impl(data, fill_value, dtype)
