Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
jerrrrry
infinicore
Commits
59e93ab4
Commit
59e93ab4
authored
Aug 08, 2025
by
zhangyue
Browse files
Merge branch 'main' of
https://github.com/InfiniTensor/InfiniCore
into p800-sub
parents
35388a54
60ca4508
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
29 additions
and
19 deletions
+29
-19
src/infiniop/ops/add/kunlun/add_kunlun.xpu
src/infiniop/ops/add/kunlun/add_kunlun.xpu
+5
-19
src/infiniop/ops/add/kunlun/kernel.h
src/infiniop/ops/add/kunlun/kernel.h
+24
-0
No files found.
src/infiniop/ops/add/kunlun/add_kunlun.xpu
View file @
59e93ab4
#include "../../../elementwise/kunlun/elementwise_kunlun.h"
#include "add_kunlun.h"
#include "kernel.h"
namespace op::elementwise::kunlun {
typedef struct AddOp {
public:
static constexpr int num_inputs = 2;
template <typename T>
inline __device__ T operator()(const T *inputs) const {
T a = inputs[0];
T b = inputs[1];
return a + b;
}
// bfloat16 特化版本(使用 float 计算精度)
inline __device__ bfloat16_t operator()(const bfloat16_t *inputs) const {
float a_f = __bfloat162float(inputs[0]);
float b_f = __bfloat162float(inputs[1]);
return __float2bfloat16(a_f + b_f);
}
} AddOp;
using AddOp = op::add::kunlun::AddOp;
INSTANTIATE_ELEMENTWISE_KERNEL(AddOp::num_inputs, AddOp, float);
INSTANTIATE_ELEMENTWISE_KERNEL(AddOp::num_inputs, AddOp, half);
...
...
@@ -67,11 +53,11 @@ infiniStatus_t Descriptor::calculate(
switch (_dtype) {
case INFINI_DTYPE_F16:
return _device_info->calculate<8,
op::elementwise::kunlun::
AddOp, half>(_info, workspace, output, inputs, stream);
return _device_info->calculate<8, AddOp, half>(_info, workspace, output, inputs, stream);
case INFINI_DTYPE_BF16:
return _device_info->calculate<8,
op::elementwise::kunlun::
AddOp, bfloat16_t>(_info, workspace, output, inputs, stream);
return _device_info->calculate<8, AddOp, bfloat16_t>(_info, workspace, output, inputs, stream);
case INFINI_DTYPE_F32:
return _device_info->calculate<8,
op::elementwise::kunlun::
AddOp, float>(_info, workspace, output, inputs, stream);
return _device_info->calculate<8, AddOp, float>(_info, workspace, output, inputs, stream);
default:
return INFINI_STATUS_BAD_TENSOR_DTYPE;
}
...
...
src/infiniop/ops/add/kunlun/kernel.h
0 → 100644
View file @
59e93ab4
#ifndef __ADD_KUNLUN_KERNEL_H__
#define __ADD_KUNLUN_KERNEL_H__
namespace
op
::
add
::
kunlun
{
typedef
struct
AddOp
{
public:
static
constexpr
int
num_inputs
=
2
;
template
<
typename
T
>
inline
__device__
T
operator
()(
const
T
*
inputs
)
const
{
T
a
=
inputs
[
0
];
T
b
=
inputs
[
1
];
return
a
+
b
;
}
// bfloat16 特化版本(使用 float 计算精度)
inline
__device__
bfloat16_t
operator
()(
const
bfloat16_t
*
inputs
)
const
{
float
a_f
=
__bfloat162float
(
inputs
[
0
]);
float
b_f
=
__bfloat162float
(
inputs
[
1
]);
return
__float2bfloat16
(
a_f
+
b_f
);
}
}
AddOp
;
}
// namespace op::add::kunlun
#endif // __ADD_KUNLUN_KERNEL_H__
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment