Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
机器未来
Paddle
提交
f39183ea
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看板
未验证
提交
f39183ea
编写于
7月 06, 2022
作者:
X
xiaoxiaohehe001
提交者:
GitHub
7月 06, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
【Paddle-Inference】 fix nvcc_lazy (#44114)
* nvcc_lazy * nvcc_lazy * conv_fusion
上级
b603dd55
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
10 addition
and
8 deletion
+10
-8
cmake/experiments/cuda_module_loading_lazy.cmake
cmake/experiments/cuda_module_loading_lazy.cmake
+5
-5
tools/nvcc_lazy
tools/nvcc_lazy
+5
-3
未找到文件。
cmake/experiments/cuda_module_loading_lazy.cmake
浏览文件 @
f39183ea
...
...
@@ -13,8 +13,8 @@
# limitations under the License.
# this file contains experimental build options for lazy cuda module loading
# cuda moduel lazy loading is supported by CUDA 11.
6
+
# this experiment option makes Paddle supports lazy loading before CUDA 11.
6
.
# cuda moduel lazy loading is supported by CUDA 11.
7
+
# this experiment option makes Paddle supports lazy loading before CUDA 11.
7
.
option
(
EXP_CUDA_MODULE_LOADING_LAZY
"enable lazy cuda module loading"
OFF
)
if
(
${
EXP_CUDA_MODULE_LOADING_LAZY
}
)
...
...
@@ -28,13 +28,13 @@ if(${EXP_CUDA_MODULE_LOADING_LAZY})
message
(
"EXP_CUDA_MODULE_LOADING_LAZY only works with CUDA"
)
return
()
endif
()
if
(
${
CUDA_VERSION
}
VERSION_GREATER_EQUAL
"11.
6
"
)
message
(
"cuda 11.
6
+ already support lazy module loading"
)
if
(
${
CUDA_VERSION
}
VERSION_GREATER_EQUAL
"11.
7
"
)
message
(
"cuda 11.
7
+ already support lazy module loading"
)
return
()
endif
()
message
(
"for cuda before 11.
6
, libcudart.so must be used for the lazy module loading trick to work, instead of libcudart_static.a"
"for cuda before 11.
7
, libcudart.so must be used for the lazy module loading trick to work, instead of libcudart_static.a"
)
set
(
CUDA_USE_STATIC_CUDA_RUNTIME
OFF
...
...
tools/nvcc_lazy
浏览文件 @
f39183ea
#!/usr/bin/env bash
unset
GREP_OPTIONS
set
-e
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
#
...
...
@@ -17,11 +19,11 @@
## CUDA_MODULE_LOADING=EAGER,DEFAULT,LAZY
# check nvcc version, if nvcc >= 11.
6
, just run nvcc itself
CUDA_VERSION
=
$(
nvcc
--version
|
grep
-oP
'(?<=
cuda_
)\d*\.\d*'
)
# check nvcc version, if nvcc >= 11.
7
, just run nvcc itself
CUDA_VERSION
=
$(
nvcc
--version
|
grep
-oP
'(?<=
V
)\d*\.\d*'
)
CUDA_VERSION_MAJOR
=
${
CUDA_VERSION
%.*
}
CUDA_VERSION_MINOR
=
${
CUDA_VERSION
#*.
}
if
((
CUDA_VERSION_MAJOR
>
11
||
(
CUDA_VERSION_MAJOR
==
11
&&
CUDA_VERSION_MINOR
>=
6
)
))
;
then
if
((
CUDA_VERSION_MAJOR
>
11
||
(
CUDA_VERSION_MAJOR
==
11
&&
CUDA_VERSION_MINOR
>=
7
)
))
;
then
nvcc
"
$@
"
exit
fi
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录