Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
机器未来
Paddle
提交
c5eeecca
P
Paddle
项目概览
机器未来
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
c5eeecca
编写于
5月 06, 2019
作者:
Z
Zeng Jinle
提交者:
GitHub
5月 06, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix tensor_py.h (#17195)
* fix tensor_py,test=develop * change class name,test=develop
上级
ee2028a1
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
18 addition
and
17 deletion
+18
-17
paddle/fluid/pybind/tensor_py.h
paddle/fluid/pybind/tensor_py.h
+18
-17
未找到文件。
paddle/fluid/pybind/tensor_py.h
浏览文件 @
c5eeecca
...
...
@@ -431,14 +431,14 @@ inline void PyCUDAPinnedTensorSetFromArray(
namespace
details
{
template
<
typename
T
>
constexpr
bool
IsValidDTypeToPyArray
()
{
return
false
;
}
#define DECLARE_VALID_DTYPE_TO_PY_ARRAY(type)
\
template <>
\
constexpr bool IsValidDTypeToPyArray<type>() {
\
return true;
\
struct
ValidDTypeToPyArrayChecker
{
static
constexpr
bool
kValue
=
false
;
}
;
#define DECLARE_VALID_DTYPE_TO_PY_ARRAY(type) \
template <> \
struct ValidDTypeToPyArrayChecker<type> {
\
static constexpr bool kValue = true;
\
}
DECLARE_VALID_DTYPE_TO_PY_ARRAY
(
platform
::
float16
);
...
...
@@ -452,15 +452,16 @@ DECLARE_VALID_DTYPE_TO_PY_ARRAY(int64_t);
inline
std
::
string
TensorDTypeToPyDTypeStr
(
framework
::
proto
::
VarType
::
Type
type
)
{
#define TENSOR_DTYPE_TO_PY_DTYPE(T, proto_type) \
if (type == proto_type) { \
if (std::is_same<T, platform::float16>::value) { \
return "e"; \
} else { \
PADDLE_ENFORCE(IsValidDTypeToPyArray<T>, \
"This type of tensor cannot be expose to Python"); \
return py::format_descriptor<T>::format(); \
} \
#define TENSOR_DTYPE_TO_PY_DTYPE(T, proto_type) \
if (type == proto_type) { \
if (std::is_same<T, platform::float16>::value) { \
return "e"; \
} else { \
constexpr auto kIsValidDType = ValidDTypeToPyArrayChecker<T>::kValue; \
PADDLE_ENFORCE(kIsValidDType, \
"This type of tensor cannot be expose to Python"); \
return py::format_descriptor<T>::format(); \
} \
}
_ForEachDataType_
(
TENSOR_DTYPE_TO_PY_DTYPE
);
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录