diff --git a/core/conversion/evaluators/aten.cpp b/core/conversion/evaluators/aten.cpp index 1eacdc8c2d..7202b0dae6 100644 --- a/core/conversion/evaluators/aten.cpp +++ b/core/conversion/evaluators/aten.cpp @@ -207,7 +207,7 @@ auto aten_registrations TRTORCH_UNUSED = RegisterNodeEvaluators() auto list = args.at(n->input(0)).IValue()->to>(); auto el = args.at(n->input(1)).IValue(); - list.push_back(std::move(el)); + list.push_back(std::move(*el)); return list; }, EvalOptions().validSchemas({ @@ -430,7 +430,7 @@ auto aten_registrations TRTORCH_UNUSED = RegisterNodeEvaluators() [](const torch::jit::Node* n, kwargs& args) -> c10::optional { auto el = args.at(n->input(0)).unwrapToDouble(); - return std::floor(el); + return static_cast(std::floor(el)); }, EvalOptions().validSchemas({ "aten::floor.float(float a) -> (int)", @@ -438,8 +438,8 @@ auto aten_registrations TRTORCH_UNUSED = RegisterNodeEvaluators() }).evaluator({ c10::Symbol::fromQualString("aten::warn"), [](const torch::jit::Node* n, kwargs& args) -> c10::optional { - auto warning = args.at(n->input(0)).IValue()->toString(); - LOG_WARNING(warning); + auto warning = args.at(n->input(0)).IValue(); + LOG_WARNING("Warning from TorchScript: " << *warning); return {}; }, EvalOptions() diff --git a/core/conversion/evaluators/prim.cpp b/core/conversion/evaluators/prim.cpp index f89dfa02d7..d40a33f4e7 100644 --- a/core/conversion/evaluators/prim.cpp +++ b/core/conversion/evaluators/prim.cpp @@ -242,8 +242,8 @@ auto prim_registrations = RegisterNodeEvaluators() }).evaluator({ c10::Symbol::fromQualString("prim::RaiseException"), [](const torch::jit::Node* n, kwargs& args) -> c10::optional { - auto exception = args.at(n->input(0)).IValue()->toString(); - TRTORCH_THROW_ERROR(exception); + auto exception = args.at(n->input(0)).IValue(); + TRTORCH_THROW_ERROR("Error from TorchScript: " << *exception); return {}; } });