Commit e34021e1 authored by yimeisun123's avatar yimeisun123 Committed by Robert Kimball

Fix the error in nbench for input tensor data (#2057)

parent 45752b0f
......@@ -181,7 +181,7 @@ vector<runtime::PerformanceCounter> run_benchmark(shared_ptr<Function> f,
auto tensor = backend->create_tensor(param->get_element_type(), param->get_shape());
auto tensor_data =
make_shared<runtime::HostTensor>(param->get_element_type(), param->get_shape());
random_init(tensor);
random_init(tensor_data);
args.push_back(tensor);
arg_data.push_back(tensor_data);
args_cacheable.push_back(param->get_cacheable());
......
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