-
Notifications
You must be signed in to change notification settings - Fork 6.8k
mxnet::cpp::NDArray::WaitAll() take about 160ms on gtx1080ti #13245
Comments
@mxnet-label-bot add [C++, Performance, Pending Requester Info] Hi @aaron900813, can you help provide some info about your environment? Environment info (Required)
Build info (Required if built from source)Compiler (gcc/clang/mingw/visual studio): MXNet commit hash: Build config: |
hi @marcoabreu ----------Python Info---------- git rev-parse HEAD CMakeLists.txt configuration nvidia-smi +-----------------------------------------------------------------------------+ update: cmake: is there anything wrong? |
Yes, building with CMake and building with Makefile are different. The former will be sometimes slow and stuck on Linux. |
@mxnet-label-bot remove [Pending Requester Info] |
|
here are codes:
tValRestart;
net271_executor->Forward(false);
std::cout << "Forward use " << tValDuration << " ms" << std::endl;
tValRestart;
auto targetx = net271_executor->outputs[0].Copy(global_cpu_ctx);
auto targety = net271_executor->outputs[1].Copy(global_cpu_ctx);
auto targetw = net271_executor->outputs[2].Copy(global_cpu_ctx);
auto targeth = net271_executor->outputs[3].Copy(global_cpu_ctx);
auto softmax_score = net271_executor->outputs[4].Copy(global_cpu_ctx);
auto penalty_t = net271_executor->outputs[5].Copy(global_cpu_ctx);
std::cout << "copy cpu use " << tValDuration << " ms" << std::endl;
tValRestart;
mxnet::cpp::NDArray::WaitAll();
std::cout << "waitall use " << tValDuration << " ms" << std::endl;
The text was updated successfully, but these errors were encountered: