Unverified Commit 253d4cdf authored by Robert Kimball's avatar Robert Kimball Committed by GitHub

Pass HostTensor all the way down (#2454)

* pass HostTensor all the way down

* fix distributed build error
parent 0a3858a0
...@@ -182,22 +182,11 @@ bool runtime::interpreter::INTExecutable::call(const vector<shared_ptr<runtime:: ...@@ -182,22 +182,11 @@ bool runtime::interpreter::INTExecutable::call(const vector<shared_ptr<runtime::
return true; return true;
} }
void runtime::interpreter::INTExecutable::generate_calls( void runtime::interpreter::INTExecutable::generate_calls(const element::Type& type,
const element::Type& type, const NodeWrapper& op,
const NodeWrapper& op, const vector<shared_ptr<HostTensor>>& out,
const vector<shared_ptr<HostTensor>>& outputs, const vector<shared_ptr<HostTensor>>& in)
const vector<shared_ptr<HostTensor>>& inputs)
{ {
vector<void*> out;
vector<const void*> in;
for (auto t : outputs)
{
out.push_back(t->get_data_ptr());
}
for (auto t : inputs)
{
in.push_back(t->get_data_ptr());
}
stringstream ss; stringstream ss;
switch (type.get_type_enum()) switch (type.get_type_enum())
{ {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment