diff --git a/paddle/fluid/platform/enforce.h b/paddle/fluid/platform/enforce.h index d3890de89a5d140bfa09d04909f703f4ca771a05..c63ea3fa8573b8a7fd739931869c8f53259d8a77 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 842d4cc139281aab48131759f63003b3fe3890c2..95a852ad6e92a3ec2f8ecc08f5378ed91301f3c3 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