Skip to content

Commit d2e7276

Browse files
committed
fix warp ctc dir error
1 parent efc094f commit d2e7276

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

paddle/testing/paddle_gtest_main.cc

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,9 @@ int main(int argc, char** argv) {
2828
}
2929
#ifdef PADDLE_WITH_CUDA
3030
new_argv.push_back(
31-
strdup("--tryfromenv=fraction_of_gpu_memory_to_use,use_pinned_memory,"
32-
"warpctc_dir"));
31+
strdup("--tryfromenv=fraction_of_gpu_memory_to_use,use_pinned_memory"));
3332
#else
34-
new_argv.push_back(strdup("--tryfromenv=use_pinned_memory,warpctc_dir"));
33+
new_argv.push_back(strdup("--tryfromenv=use_pinned_memory"));
3534
#endif
3635
int new_argc = static_cast<int>(new_argv.size());
3736
char** new_argv_address = new_argv.data();

0 commit comments

Comments
 (0)