Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Crayon鑫
Paddle
提交
3fcfcd51
P
Paddle
项目概览
Crayon鑫
/
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看板
未验证
提交
3fcfcd51
编写于
6月 02, 2022
作者:
S
sneaxiy
提交者:
GitHub
6月 02, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix bug of CUDAGraph kernel parameter comparation (#43163)
* fix cuda graph sizeof * fix tuple type
上级
5ccc49e7
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
4 addition
and
5 deletion
+4
-5
paddle/fluid/platform/device/gpu/cuda/cuda_graph.h
paddle/fluid/platform/device/gpu/cuda/cuda_graph.h
+4
-5
未找到文件。
paddle/fluid/platform/device/gpu/cuda/cuda_graph.h
浏览文件 @
3fcfcd51
...
...
@@ -60,17 +60,18 @@ template <typename Return, typename... FuncArgs,
Return
(
*
kernel_fn
)(
FuncArgs
...)>
struct
IsSameKernelHelper
<
Return
(
*
)(
FuncArgs
...),
kernel_fn
>
{
private:
using
FuncArgsTuple
=
decltype
(
std
::
make_tuple
(
std
::
declval
<
FuncArgs
>
()...));
template
<
typename
TupleT
,
size_t
IDX
,
bool
IsEnd
/*=false*/
>
struct
Impl
{
static
bool
Compare
(
const
CUDAKernelParams
&
params
,
const
TupleT
&
args
)
{
using
CompareT
=
typename
std
::
tuple_element
<
IDX
,
TupleT
>::
type
;
using
CompareT
=
typename
std
::
tuple_element
<
IDX
,
FuncArgsTuple
>::
type
;
if
(
!
IsBitwiseEqual
<
CompareT
>
(
params
.
As
<
CompareT
>
(
IDX
),
std
::
get
<
IDX
>
(
args
)))
{
return
false
;
}
constexpr
auto
NewIsEnd
=
(
IDX
+
1
==
sizeof
(
std
::
tuple_size
<
TupleT
>::
value
));
constexpr
auto
NewIsEnd
=
(
IDX
+
1
==
std
::
tuple_size
<
TupleT
>::
value
);
return
Impl
<
TupleT
,
IDX
+
1
,
NewIsEnd
>::
Compare
(
params
,
args
);
}
};
...
...
@@ -83,8 +84,6 @@ struct IsSameKernelHelper<Return (*)(FuncArgs...), kernel_fn> {
};
public:
using
FuncArgsTuple
=
decltype
(
std
::
make_tuple
(
std
::
declval
<
FuncArgs
>
()...));
template
<
typename
...
Args
>
static
bool
Compare
(
const
CUDAKernelParams
&
params
,
Args
...
args
)
{
constexpr
auto
kNumArgs
=
sizeof
...(
FuncArgs
);
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录