Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
103d7f33
MegEngine
项目概览
MegEngine 天元
/
MegEngine
9 个月 前同步成功
通知
392
Star
4702
Fork
582
代码
文件
提交
分支
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看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
103d7f33
编写于
4月 30, 2021
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
refactor(dnn/rocm): update hip license header
GitOrigin-RevId: 79d684755d31a982446a2c136ce800d70afae91f
上级
5aa52d38
变更
51
隐藏空白更改
内联
并排
Showing
51 changed file
with
322 addition
and
231 deletion
+322
-231
dnn/src/rocm/add_update/add_update.cpp.hip
dnn/src/rocm/add_update/add_update.cpp.hip
+7
-4
dnn/src/rocm/add_update/add_update.h.hip
dnn/src/rocm/add_update/add_update.h.hip
+7
-4
dnn/src/rocm/argmxx/argmxx.cpp.hip
dnn/src/rocm/argmxx/argmxx.cpp.hip
+7
-4
dnn/src/rocm/argsort/opr_impl.cpp
dnn/src/rocm/argsort/opr_impl.cpp
+2
-2
dnn/src/rocm/checksum/kern.cpp.hip
dnn/src/rocm/checksum/kern.cpp.hip
+7
-4
dnn/src/rocm/checksum/kern.h.hip
dnn/src/rocm/checksum/kern.h.hip
+7
-4
dnn/src/rocm/convolution/chanwise/bwd_data.cpp.hip
dnn/src/rocm/convolution/chanwise/bwd_data.cpp.hip
+7
-4
dnn/src/rocm/convolution/chanwise/bwd_filter.cpp.hip
dnn/src/rocm/convolution/chanwise/bwd_filter.cpp.hip
+7
-4
dnn/src/rocm/convolution/chanwise/fwd.cpp.hip
dnn/src/rocm/convolution/chanwise/fwd.cpp.hip
+7
-4
dnn/src/rocm/convolution/chanwise/kern.h.hip
dnn/src/rocm/convolution/chanwise/kern.h.hip
+7
-4
dnn/src/rocm/convolution/chanwise/kern_helper.h.hip
dnn/src/rocm/convolution/chanwise/kern_helper.h.hip
+7
-4
dnn/src/rocm/convolution/forward/inplace_matmul_impl.cpp.hip
dnn/src/rocm/convolution/forward/inplace_matmul_impl.cpp.hip
+7
-4
dnn/src/rocm/convolution/forward/inplace_matmul_impl.h.hip
dnn/src/rocm/convolution/forward/inplace_matmul_impl.h.hip
+7
-4
dnn/src/rocm/convolution/im2col.cpp.hip
dnn/src/rocm/convolution/im2col.cpp.hip
+7
-4
dnn/src/rocm/convolution/im2col.h.hip
dnn/src/rocm/convolution/im2col.h.hip
+7
-4
dnn/src/rocm/elemwise/kern_wrapper.h.hip
dnn/src/rocm/elemwise/kern_wrapper.h.hip
+6
-5
dnn/src/rocm/elemwise/special_kerns.h.hip
dnn/src/rocm/elemwise/special_kerns.h.hip
+1
-1
dnn/src/rocm/elemwise_helper.h.hip
dnn/src/rocm/elemwise_helper.h.hip
+6
-4
dnn/src/rocm/error_info.h.hip
dnn/src/rocm/error_info.h.hip
+7
-4
dnn/src/rocm/eye/eye.cpp.hip
dnn/src/rocm/eye/eye.cpp.hip
+7
-4
dnn/src/rocm/eye/eye.h.hip
dnn/src/rocm/eye/eye.h.hip
+7
-4
dnn/src/rocm/indexing_multi_axis_vec/kern.h.hip
dnn/src/rocm/indexing_multi_axis_vec/kern.h.hip
+7
-4
dnn/src/rocm/indexing_multi_axis_vec/kern_apply_opr_fwd.cpp.hip
...c/rocm/indexing_multi_axis_vec/kern_apply_opr_fwd.cpp.hip
+7
-4
dnn/src/rocm/indexing_multi_axis_vec/kern_apply_opr_impl.hipinl
...c/rocm/indexing_multi_axis_vec/kern_apply_opr_impl.hipinl
+7
-4
dnn/src/rocm/indexing_multi_axis_vec/kern_apply_opr_incr.cpp.hip
.../rocm/indexing_multi_axis_vec/kern_apply_opr_incr.cpp.hip
+7
-4
dnn/src/rocm/indexing_multi_axis_vec/kern_apply_opr_set.cpp.hip
...c/rocm/indexing_multi_axis_vec/kern_apply_opr_set.cpp.hip
+7
-4
dnn/src/rocm/indexing_multi_axis_vec/kern_gen_offset_base.cpp.hip
...rocm/indexing_multi_axis_vec/kern_gen_offset_base.cpp.hip
+7
-4
dnn/src/rocm/indexing_one_hot/indexing_one_hot.cpp.hip
dnn/src/rocm/indexing_one_hot/indexing_one_hot.cpp.hip
+7
-4
dnn/src/rocm/indexing_one_hot/indexing_one_hot.h.hip
dnn/src/rocm/indexing_one_hot/indexing_one_hot.h.hip
+7
-4
dnn/src/rocm/int_fastdiv.h.hip
dnn/src/rocm/int_fastdiv.h.hip
+6
-5
dnn/src/rocm/linspace/linspace.cpp.hip
dnn/src/rocm/linspace/linspace.cpp.hip
+7
-4
dnn/src/rocm/linspace/linspace.h.hip
dnn/src/rocm/linspace/linspace.h.hip
+7
-4
dnn/src/rocm/powc/powc.cpp.hip
dnn/src/rocm/powc/powc.cpp.hip
+7
-4
dnn/src/rocm/powc/powc.h.hip
dnn/src/rocm/powc/powc.h.hip
+7
-4
dnn/src/rocm/reduce/reduce.cpp.hip
dnn/src/rocm/reduce/reduce.cpp.hip
+7
-4
dnn/src/rocm/reduce_helper.h.hip
dnn/src/rocm/reduce_helper.h.hip
+7
-5
dnn/src/rocm/reduce_helper.hipinl
dnn/src/rocm/reduce_helper.hipinl
+7
-5
dnn/src/rocm/reduce_helper/column.hipinl
dnn/src/rocm/reduce_helper/column.hipinl
+6
-5
dnn/src/rocm/reduce_helper/largeBC.hipinl
dnn/src/rocm/reduce_helper/largeBC.hipinl
+6
-5
dnn/src/rocm/relayout/relayout.cpp.hip
dnn/src/rocm/relayout/relayout.cpp.hip
+7
-4
dnn/src/rocm/relayout/relayout.h.hip
dnn/src/rocm/relayout/relayout.h.hip
+7
-4
dnn/src/rocm/relayout/relayout_contiguous.cpp.hip
dnn/src/rocm/relayout/relayout_contiguous.cpp.hip
+7
-4
dnn/src/rocm/relayout/relayout_contiguous.h.hip
dnn/src/rocm/relayout/relayout_contiguous.h.hip
+7
-4
dnn/src/rocm/sleep/kern.cpp.hip
dnn/src/rocm/sleep/kern.cpp.hip
+7
-4
dnn/src/rocm/sleep/kern.h.hip
dnn/src/rocm/sleep/kern.h.hip
+7
-4
dnn/src/rocm/topk/opr_impl.cpp
dnn/src/rocm/topk/opr_impl.cpp
+2
-10
dnn/src/rocm/topk/topk_radix.h.hip
dnn/src/rocm/topk/topk_radix.h.hip
+0
-23
dnn/src/rocm/type_cvt/type_cvt.cpp.hip
dnn/src/rocm/type_cvt/type_cvt.cpp.hip
+7
-4
dnn/src/rocm/type_cvt/type_cvt.h.hip
dnn/src/rocm/type_cvt/type_cvt.h.hip
+7
-4
dnn/src/rocm/utils.h.hip
dnn/src/rocm/utils.h.hip
+7
-4
dnn/test/rocm/topk.cpp
dnn/test/rocm/topk.cpp
+0
-5
未找到文件。
dnn/src/rocm/add_update/add_update.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/add_update/add_update.cpp.hip
* \file
dnn/
src/rocm/add_update/add_update.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
#include "./add_update.h.hip"
#include "./add_update.h.hip"
...
...
dnn/src/rocm/add_update/add_update.h.hip
浏览文件 @
103d7f33
/**
/**
*
*
* \file src/rocm/add_update/add_update.h.hip
* \file
dnn/
src/rocm/add_update/add_update.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/argmxx/argmxx.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/argmxx/argmxx.cpp.hip
* \file
dnn/
src/rocm/argmxx/argmxx.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
#include "hip_header.h"
#include "hip_header.h"
...
...
dnn/src/rocm/argsort/opr_impl.cpp
浏览文件 @
103d7f33
...
@@ -37,7 +37,7 @@ void ArgsortForwardImpl::exec(_megdnn_tensor_in src, _megdnn_tensor_out dst,
...
@@ -37,7 +37,7 @@ void ArgsortForwardImpl::exec(_megdnn_tensor_in src, _megdnn_tensor_out dst,
ARGSORT_FOREACH_CTYPE
(
cb
);
ARGSORT_FOREACH_CTYPE
(
cb
);
#undef cb
#undef cb
default:
default:
megdnn_throw
(
ssprintf
(
"unsupported argsort dtype on
cuda
: %s"
,
megdnn_throw
(
ssprintf
(
"unsupported argsort dtype on
rocm
: %s"
,
src
.
layout
.
dtype
.
name
()));
src
.
layout
.
dtype
.
name
()));
}
}
}
}
...
@@ -71,7 +71,7 @@ void ArgsortBackwardImpl::exec(_megdnn_tensor_in diff,
...
@@ -71,7 +71,7 @@ void ArgsortBackwardImpl::exec(_megdnn_tensor_in diff,
ARGSORT_FOREACH_CTYPE
(
cb
);
ARGSORT_FOREACH_CTYPE
(
cb
);
#undef cb
#undef cb
default:
default:
megdnn_throw
(
ssprintf
(
"unsupported argsort dtype on
cuda
: %s"
,
megdnn_throw
(
ssprintf
(
"unsupported argsort dtype on
rocm
: %s"
,
diff
.
layout
.
dtype
.
name
()));
diff
.
layout
.
dtype
.
name
()));
}
}
}
}
...
...
dnn/src/rocm/checksum/kern.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/checksum/kern.cpp.hip
* \file
dnn/
src/rocm/checksum/kern.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
#include "hip_header.h"
#include "hip_header.h"
...
...
dnn/src/rocm/checksum/kern.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/checksum/kern.h.hip
* \file
dnn/
src/rocm/checksum/kern.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/convolution/chanwise/bwd_data.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/convolution/chanwise/bwd_data.cpp.hip
* \file
dnn/
src/rocm/convolution/chanwise/bwd_data.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hip_header.h"
#include "hip_header.h"
...
...
dnn/src/rocm/convolution/chanwise/bwd_filter.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/convolution/chanwise/bwd_filter.cpp.hip
* \file
dnn/
src/rocm/convolution/chanwise/bwd_filter.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hip_header.h"
#include "hip_header.h"
...
...
dnn/src/rocm/convolution/chanwise/fwd.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/convolution/chanwise/fwd.cpp.hip
* \file
dnn/
src/rocm/convolution/chanwise/fwd.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hip_header.h"
#include "hip_header.h"
...
...
dnn/src/rocm/convolution/chanwise/kern.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/convolution/chanwise/kern.h.hip
* \file
dnn/
src/rocm/convolution/chanwise/kern.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/convolution/chanwise/kern_helper.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/convolution/chanwise/kern_helper.h.hip
* \file
dnn/
src/rocm/convolution/chanwise/kern_helper.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/convolution/forward/inplace_matmul_impl.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/convolution/forward/inplace_matmul_impl.cpp.hip
* \file
dnn/
src/rocm/convolution/forward/inplace_matmul_impl.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "./inplace_matmul_impl.h.hip"
#include "./inplace_matmul_impl.h.hip"
#include "src/rocm/utils.h.hip"
#include "src/rocm/utils.h.hip"
...
...
dnn/src/rocm/convolution/forward/inplace_matmul_impl.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/convolution/forward/inplace_matmul_impl.h.hip
* \file
dnn/
src/rocm/convolution/forward/inplace_matmul_impl.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/convolution/im2col.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/convolution/im2col.cpp.hip
* \file
dnn/
src/rocm/convolution/im2col.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "./im2col.h.hip"
#include "./im2col.h.hip"
#include "megdnn/dtype.h"
#include "megdnn/dtype.h"
...
...
dnn/src/rocm/convolution/im2col.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/convolution/im2col.h.hip
* \file
dnn/
src/rocm/convolution/im2col.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/elemwise/kern_wrapper.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/elemwise/kern_wrapper.h.hip
* \file
dnn/
src/rocm/elemwise/kern_wrapper.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
*
\brief helper for implementing elemwise oprs
*
Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/elemwise/special_kerns.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/elemwise/special_kerns.h.hip
* \file
dnn/
src/rocm/elemwise/special_kerns.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* This file is part of MegDNN, a deep neural network run-time library
* developed by Megvii.
* developed by Megvii.
...
...
dnn/src/rocm/elemwise_helper.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/elemwise_helper.h.hip
* \file
dnn/
src/rocm/elemwise_helper.h.hip
*
*
*
This file is part of MegBrain, a deep learning framework developed by Megvii.
*
MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
*
*
\brief helper utilities for implementing element-wise kernels
*
Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/error_info.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/error_info.h.hip
* \file
dnn/
src/rocm/error_info.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/eye/eye.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/eye/eye.cpp.hip
* \file
dnn/
src/rocm/eye/eye.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
#include "hip_header.h"
#include "hip_header.h"
...
...
dnn/src/rocm/eye/eye.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/eye/eye.h.hip
* \file
dnn/
src/rocm/eye/eye.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
#include <stdint.h>
#include <stdint.h>
...
...
dnn/src/rocm/indexing_multi_axis_vec/kern.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/indexing_multi_axis_vec/kern.h.hip
* \file
dnn/
src/rocm/indexing_multi_axis_vec/kern.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/indexing_multi_axis_vec/kern_apply_opr_fwd.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/indexing_multi_axis_vec/kern_apply_opr_fwd.cpp.hip
* \file
dnn/
src/rocm/indexing_multi_axis_vec/kern_apply_opr_fwd.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
...
...
dnn/src/rocm/indexing_multi_axis_vec/kern_apply_opr_impl.hipinl
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/indexing_multi_axis_vec/kern_apply_opr_impl.hipinl
* \file
dnn/
src/rocm/indexing_multi_axis_vec/kern_apply_opr_impl.hipinl
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#ifndef KERN_APPLY_OPR_OPR
#ifndef KERN_APPLY_OPR_OPR
...
...
dnn/src/rocm/indexing_multi_axis_vec/kern_apply_opr_incr.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/indexing_multi_axis_vec/kern_apply_opr_incr.cpp.hip
* \file
dnn/
src/rocm/indexing_multi_axis_vec/kern_apply_opr_incr.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
...
...
dnn/src/rocm/indexing_multi_axis_vec/kern_apply_opr_set.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/indexing_multi_axis_vec/kern_apply_opr_set.cpp.hip
* \file
dnn/
src/rocm/indexing_multi_axis_vec/kern_apply_opr_set.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
...
...
dnn/src/rocm/indexing_multi_axis_vec/kern_gen_offset_base.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/indexing_multi_axis_vec/kern_gen_offset_base.hip.cpp
* \file
dnn/
src/rocm/indexing_multi_axis_vec/kern_gen_offset_base.hip.cpp
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
...
...
dnn/src/rocm/indexing_one_hot/indexing_one_hot.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/indexing_one_hot/indexing_one_hot.cpp.hip
* \file
dnn/
src/rocm/indexing_one_hot/indexing_one_hot.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2016 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
#include "./indexing_one_hot.h.hip"
#include "./indexing_one_hot.h.hip"
...
...
dnn/src/rocm/indexing_one_hot/indexing_one_hot.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/indexing_one_hot/indexing_one_hot.h.hip
* \file
dnn/
src/rocm/indexing_one_hot/indexing_one_hot.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/int_fastdiv.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/int_fastdiv.h.hip
* \file
dnn/
src/rocm/int_fastdiv.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
*
\brief fast integer division for constant divisor
*
Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/linspace/linspace.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/linspace/linspace.cpp.hip
* \file
dnn/
src/rocm/linspace/linspace.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
#include "./linspace.h.hip"
#include "./linspace.h.hip"
...
...
dnn/src/rocm/linspace/linspace.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/linspace/linspace.h.hip
* \file
dnn/
src/rocm/linspace/linspace.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hip_header.h"
#include "hip_header.h"
...
...
dnn/src/rocm/powc/powc.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/powc/powc.cpp.hip
* \file
dnn/
src/rocm/powc/powc.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
#include "src/rocm/powc/powc.h.hip"
#include "src/rocm/powc/powc.h.hip"
...
...
dnn/src/rocm/powc/powc.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/powc/powc.h.hip
* \file
dnn/
src/rocm/powc/powc.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hip_header.h"
#include "hip_header.h"
...
...
dnn/src/rocm/reduce/reduce.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/reduce/reduce.cpp.hip
* \file
dnn/
src/rocm/reduce/reduce.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
#include "hip_header.h"
#include "hip_header.h"
...
...
dnn/src/rocm/reduce_helper.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/reduce_helper.h.hip
* \file
dnn/
src/rocm/reduce_helper.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \brief helper for implementing reduce operators
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/reduce_helper.hipinl
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/reduce_helper.hipinl
* \file
dnn/
src/rocm/reduce_helper.hipinl
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
*
\brief helper for implementing reduce operators
*
Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/reduce_helper/column.hipinl
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/reduce_helper/column.hipinl
* \file
dnn/
src/rocm/reduce_helper/column.hipinl
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
*
\brief reduce kernel for (A, B) -> (A, 1) with small B
*
Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/reduce_helper/largeBC.hipinl
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/reduce_helper/largeBC.hipinl
* \file
dnn/
src/rocm/reduce_helper/largeBC.hipinl
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
*
\brief reduce kernel for large (B * C) value
*
Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/relayout/relayout.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/relayout/relayout.cpp.hip
* \file
dnn/
src/rocm/relayout/relayout.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
#include "./relayout.h.hip"
#include "./relayout.h.hip"
...
...
dnn/src/rocm/relayout/relayout.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/relayout/relayout.h.hip
* \file
dnn/
src/rocm/relayout/relayout.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/relayout/relayout_contiguous.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/relayout/kern_contiguous.cpp.hip
* \file
dnn/
src/rocm/relayout/kern_contiguous.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
...
...
dnn/src/rocm/relayout/relayout_contiguous.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/relayout/relayout_contiguous.h.hip
* \file
dnn/
src/rocm/relayout/relayout_contiguous.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/sleep/kern.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/sleep/kern.cpp.hip
* \file
dnn/
src/rocm/sleep/kern.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
#include "hip_header.h"
#include "hip_header.h"
...
...
dnn/src/rocm/sleep/kern.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/sleep/kern.h.hip
* \file
dnn/
src/rocm/sleep/kern.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/src/rocm/topk/opr_impl.cpp
浏览文件 @
103d7f33
...
@@ -81,18 +81,10 @@ void TopKImpl::do_exec(int k, _megdnn_tensor_in data, _megdnn_tensor_out values,
...
@@ -81,18 +81,10 @@ void TopKImpl::do_exec(int k, _megdnn_tensor_in data, _megdnn_tensor_out values,
values
.
ptr
<
int32_t
>
(),
indices
,
values
.
ptr
<
int32_t
>
(),
indices
,
workspace
.
raw_ptr
);
workspace
.
raw_ptr
);
return
;
return
;
// #if !MEGDNN_DISABLE_FLOAT16
// case DTypeEnum::Float16:
// dispatch_with_ctype<dt_float16>(k, data.layout[0], data.layout[1],
// data.layout.stride[0], data.ptr<dt_float16>(),
// values.ptr<dt_float16>(), indices,
// workspace.raw_ptr);
// return;
// #endif
default:
default:
megdnn_throw
(
megdnn_throw
(
ssprintf
(
"only float32, int32 a
nd float16
supported for "
ssprintf
(
"only float32, int32 a
re
supported for "
"
cuda
topk, got: %s"
,
"
rocm
topk, got: %s"
,
data
.
layout
.
dtype
.
name
()));
data
.
layout
.
dtype
.
name
()));
}
}
}
}
...
...
dnn/src/rocm/topk/topk_radix.h.hip
浏览文件 @
103d7f33
...
@@ -59,29 +59,6 @@ struct RadixConverter<int32_t> {
...
@@ -59,29 +59,6 @@ struct RadixConverter<int32_t> {
}
}
};
};
// #if !MEGDNN_DISABLE_FLOAT16
// template <>
// struct RadixConverter<dt_float16> {
// union FIunion {
// FIunion() {}
// dt_float16 fv;
// uint16_t iv;
// };
// static __forceinline__ __device__ __host__ uint16_t to_radix(dt_float16 val) {
// FIunion fi;
// fi.fv = val;
// return fi.iv ^ (((!(fi.iv >> 15u)) - 1u) | 0x8000u);
// }
// static __forceinline__ __device__ __host__ dt_float16 from_radix(uint16_t val) {
// FIunion fi;
// // do not write as to_radix() to work around a compiler bug in cuda-9.0
// uint16_t m = 0x8000u;
// fi.iv = val ^ (m | (m - !(val >> 15u)));
// return fi.fv;
// }
// };
// #endif
}
// namespace internal
}
// namespace internal
/*!
/*!
...
...
dnn/src/rocm/type_cvt/type_cvt.cpp.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/type_cvt/type_cvt.cpp.hip
* \file
dnn/
src/rocm/type_cvt/type_cvt.cpp.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hcc_detail/hcc_defs_prologue.h"
#include "hcc_detail/hcc_defs_prologue.h"
#include "./type_cvt.h.hip"
#include "./type_cvt.h.hip"
...
...
dnn/src/rocm/type_cvt/type_cvt.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/type_cvt/type_cvt.h.hip
* \file
dnn/
src/rocm/type_cvt/type_cvt.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#include "hip_header.h"
#include "hip_header.h"
...
...
dnn/src/rocm/utils.h.hip
浏览文件 @
103d7f33
/**
/**
* \file src/rocm/utils.h.hip
* \file
dnn/
src/rocm/utils.h.hip
*
*
* This file is part of MegDNN, a deep neural network run-time library
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
*
*
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
*/
#pragma once
#pragma once
...
...
dnn/test/rocm/topk.cpp
浏览文件 @
103d7f33
...
@@ -29,10 +29,5 @@ TEST_F(ROCM, TOP_K) {
...
@@ -29,10 +29,5 @@ TEST_F(ROCM, TOP_K) {
TEST_F
(
ROCM
,
TOP_K_I32
)
{
TEST_F
(
ROCM
,
TOP_K_I32
)
{
run_topk_test
<
dtype
::
Int32
>
(
handle_rocm
());
run_topk_test
<
dtype
::
Int32
>
(
handle_rocm
());
}
}
// #if !MEGDNN_DISABLE_FLOAT16
// TEST_F(ROCM, TOP_K_F16) {
// run_topk_test<dtype::Float16>(handle_rocm());
// }
// #endif
// vim: syntax=cpp.doxygen
// vim: syntax=cpp.doxygen
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录