Skip to content

Commit 69aabee

Browse files
committed
chore: Use scalar_type in prim::dtype
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 74d456b commit 69aabee

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

core/conversion/evaluators/prim.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ auto prim_registrations =
111111
} else if (input.isIValue()) {
112112
if (input.IValue()->isTensor()) {
113113
auto pyt_input = input.IValue()->toTensor();
114-
return static_cast<int>(c10::typeMetaToScalarType(pyt_input.dtype()));
114+
return static_cast<int>(pyt_input.scalar_type());
115115
} else {
116116
TRTORCH_THROW_ERROR("Unsupported input type in prim::dtype operator");
117117
return {};

0 commit comments

Comments
 (0)