未验证 提交 1924aa14 编写于 作者: T Tao Luo 提交者: GitHub

Merge pull request #8568 from QiJune/fix_wrap_ctc_dir

fix warp ctc dir error
...@@ -28,10 +28,9 @@ int main(int argc, char** argv) { ...@@ -28,10 +28,9 @@ int main(int argc, char** argv) {
} }
#ifdef PADDLE_WITH_CUDA #ifdef PADDLE_WITH_CUDA
new_argv.push_back( new_argv.push_back(
strdup("--tryfromenv=fraction_of_gpu_memory_to_use,use_pinned_memory," strdup("--tryfromenv=fraction_of_gpu_memory_to_use,use_pinned_memory"));
"warpctc_dir"));
#else #else
new_argv.push_back(strdup("--tryfromenv=use_pinned_memory,warpctc_dir")); new_argv.push_back(strdup("--tryfromenv=use_pinned_memory"));
#endif #endif
int new_argc = static_cast<int>(new_argv.size()); int new_argc = static_cast<int>(new_argv.size());
char** new_argv_address = new_argv.data(); char** new_argv_address = new_argv.data();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册