Skip to content
Snippets Groups Projects
Commit 451ac9e9 authored by Olivier BICHLER's avatar Olivier BICHLER Committed by Maxence Naud
Browse files

Fix bug #231

parent 2dcc7e76
No related branches found
No related tags found
1 merge request!325[Upd] Patch v0.5.1
......@@ -266,7 +266,12 @@ void Aidge::GraphView::logOutputs(const std::string& dirName) const {
AIDGE_THROW_OR_ABORT(std::runtime_error,
"Could not create graph view log file: {}", inputPath);
}
fmt::print(fp.get(), "{}\n", nodePtr->getOperator()->getRawOutput(outIdx)->toString().c_str());
auto oTensor = std::static_pointer_cast<OperatorTensor>(nodePtr->getOperator())->getOutput(outIdx);
std::shared_ptr<Tensor> fallback;
const Tensor& localTensor = oTensor->refFrom(fallback, "cpu");
fmt::print(fp.get(), "{}\n", localTensor.toString().c_str());
}
}
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment