Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
机器未来
Paddle
提交
8c296dea
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看板
未验证
提交
8c296dea
编写于
5月 08, 2020
作者:
P
Pei Yang
提交者:
GitHub
5月 08, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix compile error(cpuid.h not found) on nvidia jetson platforms. test=develop (#24329)
上级
4a5de144
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
5 addition
and
0 deletion
+5
-0
cmake/cuda.cmake
cmake/cuda.cmake
+1
-0
paddle/fluid/platform/cpu_info.cc
paddle/fluid/platform/cpu_info.cc
+2
-0
paddle/fluid/platform/cpu_info.h
paddle/fluid/platform/cpu_info.h
+2
-0
未找到文件。
cmake/cuda.cmake
浏览文件 @
8c296dea
...
...
@@ -4,6 +4,7 @@ endif()
if
(
WITH_NV_JETSON
)
add_definitions
(
-DWITH_NV_JETSON
)
set
(
paddle_known_gpu_archs
"53 62 72"
)
set
(
paddle_known_gpu_archs7
"53"
)
set
(
paddle_known_gpu_archs8
"53 62"
)
...
...
paddle/fluid/platform/cpu_info.cc
浏览文件 @
8c296dea
...
...
@@ -139,6 +139,7 @@ bool MayIUse(const cpu_isa_t cpu_isa) {
if
(
cpu_isa
==
isa_any
)
{
return
true
;
}
else
{
#ifndef WITH_NV_JETSON
int
reg
[
4
];
cpuid
(
reg
,
0
);
int
nIds
=
reg
[
0
];
...
...
@@ -164,6 +165,7 @@ bool MayIUse(const cpu_isa_t cpu_isa) {
return
(
reg
[
1
]
&
avx512f_mask
)
!=
0
;
}
}
#endif
return
false
;
}
}
...
...
paddle/fluid/platform/cpu_info.h
浏览文件 @
8c296dea
...
...
@@ -40,12 +40,14 @@ limitations under the License. */
#ifdef _WIN32
#define cpuid(reg, x) __cpuidex(reg, x, 0)
#else
#ifndef WITH_NV_JETSON
#include <cpuid.h>
inline
void
cpuid
(
int
reg
[
4
],
int
x
)
{
__cpuid_count
(
x
,
0
,
reg
[
0
],
reg
[
1
],
reg
[
2
],
reg
[
3
]);
}
#endif
#endif
#endif
namespace
paddle
{
namespace
platform
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录