We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 74d456b commit 69aabeeCopy full SHA for 69aabee
core/conversion/evaluators/prim.cpp
@@ -111,7 +111,7 @@ auto prim_registrations =
111
} else if (input.isIValue()) {
112
if (input.IValue()->isTensor()) {
113
auto pyt_input = input.IValue()->toTensor();
114
- return static_cast<int>(c10::typeMetaToScalarType(pyt_input.dtype()));
+ return static_cast<int>(pyt_input.scalar_type());
115
} else {
116
TRTORCH_THROW_ERROR("Unsupported input type in prim::dtype operator");
117
return {};
0 commit comments