Skip to content

Commit a427e76

Browse files
committed
skip use_mkldnn if do not use it
1 parent 6c1cf60 commit a427e76

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

paddle/testing/paddle_gtest_main.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,8 @@ int main(int argc, char** argv) {
3030
new_argv.push_back(
3131
strdup("--tryfromenv=fraction_of_gpu_memory_to_use,use_pinned_memory"));
3232
#else
33-
new_argv.push_back(strdup("--tryfromenv=use_pinned_memory"));
33+
new_argv.push_back(strdup("--tryfromenv=use_pinned_memory,use_mkldnn"));
34+
new_argv.push_back(strdup("--undefok=use_mkldnn"));
3435
#endif
3536
int new_argc = static_cast<int>(new_argv.size());
3637
char** new_argv_address = new_argv.data();

0 commit comments

Comments
 (0)