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 d6645d3 commit 3e27790Copy full SHA for 3e27790
tests/util/run_graph_engine.cpp
@@ -4,6 +4,7 @@
4
#include "core/ir/ir.h"
5
#include "core/runtime/runtime.h"
6
#include "core/util/prelude.h"
7
+#include "core/util/trt_util.h"
8
#include "cuda_runtime_api.h"
9
#include "torch/csrc/jit/ir/ir.h"
10
#include "torch/csrc/jit/ir/irparser.h"
@@ -19,7 +20,7 @@ namespace util {
19
20
std::vector<core::ir::Input> toInputs(std::vector<at::Tensor> ten) {
21
std::vector<core::ir::Input> a;
22
for (auto i : ten) {
- a.push_back(core::ir::Input(core::util::toVec(i.sizes())));
23
+ a.push_back(core::ir::Input(core::util::toVec(i.sizes()), core::util::ScalarTypeToTRTDataType(i.scalar_type())));
24
}
25
return std::move(a);
26
0 commit comments