wangkuiyi-patch-1
tensor-tang 7 years ago
parent e797dd1786
commit 9dd993950a

@ -61,6 +61,8 @@ void Main(bool use_gpu) {
for (size_t i = 0; i < std::min(5UL, num_elements); i++) {
LOG(INFO) << static_cast<float*>(outputs.front().data.data)[i];
}
// TODO(Superjomn): this is should be free automatically
free(outputs[0].data.data);
}
}
@ -101,6 +103,7 @@ void MainThreads(int num_threads) {
for (size_t i = 0; i < std::min(5UL, num_elements); i++) {
LOG(INFO) << static_cast<float*>(outputs.front().data.data)[i];
}
free(outputs[0].data.data);
}
});
}

Loading…
Cancel
Save