Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
wmsofts
Paddle
提交
002f325d
P
Paddle
项目概览
wmsofts
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
002f325d
编写于
8月 16, 2019
作者:
Z
Zeng Jinle
提交者:
GitHub
8月 16, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add PADDLE_ENFORCE_CUDA_SUCCESS, test=develop (#19211)
上级
07a4d8f8
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
90 addition
and
0 deletion
+90
-0
paddle/fluid/platform/enforce.h
paddle/fluid/platform/enforce.h
+47
-0
paddle/fluid/platform/enforce_test.cc
paddle/fluid/platform/enforce_test.cc
+43
-0
未找到文件。
paddle/fluid/platform/enforce.h
浏览文件 @
002f325d
...
...
@@ -236,6 +236,31 @@ inline void throw_on_error(ncclResult_t stat, const std::string& msg) {
#endif // __APPLE__ and windows
#endif // PADDLE_WITH_CUDA
#ifdef PADDLE_WITH_CUDA
namespace
details
{
template
<
typename
T
>
struct
CudaStatusType
{};
#define DEFINE_CUDA_STATUS_TYPE(type, success_value) \
template <> \
struct CudaStatusType<type> { \
using Type = type; \
static constexpr Type kSuccess = success_value; \
}
DEFINE_CUDA_STATUS_TYPE
(
cudaError_t
,
cudaSuccess
);
DEFINE_CUDA_STATUS_TYPE
(
curandStatus_t
,
CURAND_STATUS_SUCCESS
);
DEFINE_CUDA_STATUS_TYPE
(
cudnnStatus_t
,
CUDNN_STATUS_SUCCESS
);
DEFINE_CUDA_STATUS_TYPE
(
cublasStatus_t
,
CUBLAS_STATUS_SUCCESS
);
#if !defined(__APPLE__) && !defined(_WIN32)
DEFINE_CUDA_STATUS_TYPE
(
ncclResult_t
,
ncclSuccess
);
#endif
}
// namespace details
#endif
#define PADDLE_THROW(...) \
do { \
throw ::paddle::platform::EnforceNotMet( \
...
...
@@ -256,6 +281,28 @@ inline void throw_on_error(ncclResult_t stat, const std::string& msg) {
} \
} while (0)
#ifdef PADDLE_WITH_CUDA
#define PADDLE_ENFORCE_CUDA_SUCCESS(COND, ...) \
do { \
auto __cond__ = (COND); \
using __CUDA_STATUS_TYPE__ = decltype(__cond__); \
constexpr auto __success_type__ = \
::paddle::platform::details::CudaStatusType< \
__CUDA_STATUS_TYPE__>::kSuccess; \
if (UNLIKELY(__cond__ != __success_type__)) { \
try { \
::paddle::platform::throw_on_error( \
__cond__, ::paddle::string::Sprintf(__VA_ARGS__)); \
} catch (...) { \
throw ::paddle::platform::EnforceNotMet(std::current_exception(), \
__FILE__, __LINE__); \
} \
} \
} while (0)
#undef DEFINE_CUDA_STATUS_TYPE
#endif
#define PADDLE_THROW_EOF() \
do { \
throw ::paddle::platform::EOFException("There is no next data.", __FILE__, \
...
...
paddle/fluid/platform/enforce_test.cc
浏览文件 @
002f325d
...
...
@@ -253,3 +253,46 @@ TEST(EOF_EXCEPTION, THROW_EOF) {
}
EXPECT_TRUE
(
caught_eof
);
}
#ifdef PADDLE_WITH_CUDA
template
<
typename
T
>
bool
CheckCudaStatusSuccess
(
T
value
,
const
std
::
string
&
msg
=
"success"
)
{
PADDLE_ENFORCE_CUDA_SUCCESS
(
value
,
msg
);
return
true
;
}
template
<
typename
T
>
bool
CheckCudaStatusFailure
(
T
value
,
const
std
::
string
&
msg
=
"self-defined cuda status failed"
)
{
try
{
PADDLE_ENFORCE_CUDA_SUCCESS
(
value
,
msg
);
return
false
;
}
catch
(
paddle
::
platform
::
EnforceNotMet
&
error
)
{
std
::
string
ex_msg
=
error
.
what
();
return
ex_msg
.
find
(
msg
)
!=
std
::
string
::
npos
;
}
}
TEST
(
enforce
,
cuda_success
)
{
EXPECT_TRUE
(
CheckCudaStatusSuccess
(
cudaSuccess
));
EXPECT_TRUE
(
CheckCudaStatusFailure
(
cudaErrorInvalidValue
));
EXPECT_TRUE
(
CheckCudaStatusFailure
(
cudaErrorMemoryAllocation
));
EXPECT_TRUE
(
CheckCudaStatusSuccess
(
CURAND_STATUS_SUCCESS
));
EXPECT_TRUE
(
CheckCudaStatusFailure
(
CURAND_STATUS_VERSION_MISMATCH
));
EXPECT_TRUE
(
CheckCudaStatusFailure
(
CURAND_STATUS_NOT_INITIALIZED
));
EXPECT_TRUE
(
CheckCudaStatusSuccess
(
CUDNN_STATUS_SUCCESS
));
EXPECT_TRUE
(
CheckCudaStatusFailure
(
CUDNN_STATUS_NOT_INITIALIZED
));
EXPECT_TRUE
(
CheckCudaStatusFailure
(
CUDNN_STATUS_ALLOC_FAILED
));
EXPECT_TRUE
(
CheckCudaStatusSuccess
(
CUBLAS_STATUS_SUCCESS
));
EXPECT_TRUE
(
CheckCudaStatusFailure
(
CUBLAS_STATUS_NOT_INITIALIZED
));
EXPECT_TRUE
(
CheckCudaStatusFailure
(
CUBLAS_STATUS_INVALID_VALUE
));
#if !defined(__APPLE__) && !defined(_WIN32)
EXPECT_TRUE
(
CheckCudaStatusSuccess
(
ncclSuccess
));
EXPECT_TRUE
(
CheckCudaStatusFailure
(
ncclUnhandledCudaError
));
EXPECT_TRUE
(
CheckCudaStatusFailure
(
ncclSystemError
));
#endif
}
#endif
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录