Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
1c7d0802
MegEngine
项目概览
MegEngine 天元
/
MegEngine
接近 2 年 前同步成功
通知
414
Star
4708
Fork
583
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
MegEngine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
提交
1c7d0802
编写于
5月 26, 2021
作者:
M
Megvii Engine Team
提交者:
huangxinda
7月 19, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(cuda): remove cuda driver version check and runtime minor version
GitOrigin-RevId: 4463beccf15e6eec5b9a1f081523e127aeb94c96
上级
b87af9f7
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
4 addition
and
4 deletion
+4
-4
src/core/impl/utils/persistent_cache.cpp
src/core/impl/utils/persistent_cache.cpp
+4
-4
未找到文件。
src/core/impl/utils/persistent_cache.cpp
浏览文件 @
1c7d0802
...
...
@@ -41,14 +41,14 @@ std::string PersistentCache::make_category_from_comp_node(CompNode comp_node) {
switch
(
env
.
property
().
type
)
{
#if MGB_CUDA
case
CompNode
::
DeviceType
::
CUDA
:
{
int
drv
=
-
1
,
cuda_rt
=
-
1
;
MGB_CUDA_CHECK
(
cudaDriverGetVersion
(
&
drv
));
int
cuda_rt
=
-
1
;
MGB_CUDA_CHECK
(
cudaRuntimeGetVersion
(
&
cuda_rt
));
int
cuda_rt_major
=
cuda_rt
/
1000
;
auto
&&
prop
=
env
.
cuda_env
().
device_prop
;
// note: we do not contain library versions such as cudnn here. They
// are handled by opr impls in MegDNN
return
ssprintf
(
"plat=cuda;dev=%s;cap=%d.%d
,drv=%d
;runtime=%d"
,
prop
.
name
,
prop
.
major
,
prop
.
minor
,
drv
,
cuda_rt
);
return
ssprintf
(
"plat=cuda;dev=%s;cap=%d.%d;runtime=%d"
,
prop
.
name
,
prop
.
major
,
prop
.
minor
,
cuda_rt_major
);
break
;
}
#endif
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录