From ab41a9ee8902dbf461b55ef9347071d7eb71fd76 Mon Sep 17 00:00:00 2001 From: Zhou Wei <52485244+zhouwei25@users.noreply.github.com> Date: Thu, 10 Jun 2021 20:58:37 +0800 Subject: [PATCH] fix unittest failure due to the path is too long (#33447) --- paddle/fluid/platform/enforce.h | 4 ++-- paddle/fluid/platform/enforce_test.cc | 3 +-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/paddle/fluid/platform/enforce.h b/paddle/fluid/platform/enforce.h index d3890de89a5..c63ea3fa857 100644 --- a/paddle/fluid/platform/enforce.h +++ b/paddle/fluid/platform/enforce.h @@ -775,13 +775,13 @@ inline std::string GetExternalErrorMsg(T status) { } } #else - char buf[100]; + char buf[512]; MEMORY_BASIC_INFORMATION mbi; HMODULE h_module = (::VirtualQuery(GetCurrentTraceBackString, &mbi, sizeof(mbi)) != 0) ? (HMODULE)mbi.AllocationBase : NULL; - GetModuleFileName(h_module, buf, 100); + GetModuleFileName(h_module, buf, 512); std::string strModule(buf); const size_t last_slash_idx = strModule.find_last_of("\\"); std::string compare_path = strModule.substr(strModule.length() - 7); diff --git a/paddle/fluid/platform/enforce_test.cc b/paddle/fluid/platform/enforce_test.cc index 842d4cc1392..95a852ad6e9 100644 --- a/paddle/fluid/platform/enforce_test.cc +++ b/paddle/fluid/platform/enforce_test.cc @@ -417,7 +417,7 @@ TEST(enforce, cuda_success) { "An unsupported value or parameter was passed to the function (a " "negative vector size, for example).To correct: ensure that all the " "parameters being passed have valid values")); - /* + #if !defined(__APPLE__) && defined(PADDLE_WITH_NCCL) EXPECT_TRUE(CheckCudaStatusSuccess(ncclSuccess)); EXPECT_TRUE(CheckCudaStatusFailure(ncclUnhandledCudaError, "NCCL error")); @@ -430,7 +430,6 @@ TEST(enforce, cuda_success) { "The call to NCCL is incorrect. This is " "usually reflecting a programming error")); #endif -*/ } #endif #endif -- GitLab