Commit 4bb40baa by Howave Committed by Tianqi Chen

[DOCS] Fix C++ example:graph_runtime.cc:151: Check failed: data->ndim ==…

[DOCS] Fix C++ example:graph_runtime.cc:151: Check failed: data->ndim == data_out->ndim (2 vs. 1) (#1987)
parent db3dcba2
...@@ -96,8 +96,8 @@ int main() ...@@ -96,8 +96,8 @@ int main()
run(); run();
DLTensor* y; DLTensor* y;
int out_ndim = 1; int out_ndim = 2;
int64_t out_shape[1] = {1000, }; int64_t out_shape[2] = {1, 1000, };
TVMArrayAlloc(out_shape, out_ndim, dtype_code, dtype_bits, dtype_lanes, device_type, device_id, &y); TVMArrayAlloc(out_shape, out_ndim, dtype_code, dtype_bits, dtype_lanes, device_type, device_id, &y);
// get the function from the module(get output data) // get the function from the module(get output data)
......
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