Skip to content

Commit 07c5a70

Browse files
committed
issue/360: modified kunlun_common.h
1 parent 3a0c81a commit 07c5a70

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

src/infiniop/devices/kunlun/kunlun_common.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
#define __KUNLUN_COMMON_H__
33

44
#include "../../../utils.h"
5+
#include <xpu/refactor/context/xpu_act_type.h>
56
#include <xpu/runtime.h>
67
#include <xpu/runtime_ex.h>
78
#include <xpu/xdnn.h>

src/infiniop/ops/conv/kunlun/conv_kunlun.cc

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
#include "../../../../utils.h"
33
#include "../../../devices/kunlun/kunlun_common.h"
44
#include "../../../devices/kunlun/kunlun_handle.h"
5-
#include <xpu/refactor/context/xpu_act_type.h>
5+
66
namespace op::conv::kunlun {
77

88
struct Descriptor::Opaque {
@@ -96,7 +96,7 @@ infiniStatus_t conv_kernel(
9696
stride, pad,
9797
dilation, 1, nullptr,
9898
nullptr, nullptr, true, bias_F32,
99-
nullptr, baidu::xpu::api::Activation_t::LINEAR,
99+
nullptr, xdnn::Activation_t::LINEAR,
100100
nullptr)));
101101
return INFINI_STATUS_SUCCESS;
102102
}));
@@ -117,7 +117,7 @@ infiniStatus_t conv_kernel(
117117
stride, pad,
118118
dilation, 1, nullptr,
119119
nullptr, nullptr, true, bias_F32,
120-
nullptr, baidu::xpu::api::Activation_t::LINEAR, nullptr,
120+
nullptr, xdnn::Activation_t::LINEAR, nullptr,
121121
nullptr, -1)));
122122
return INFINI_STATUS_SUCCESS;
123123
}));
@@ -137,7 +137,7 @@ infiniStatus_t conv_kernel(
137137
stride, pad,
138138
dilation, 1, nullptr,
139139
nullptr, nullptr, true, bias_F32,
140-
nullptr, baidu::xpu::api::Activation_t::LINEAR,
140+
nullptr, xdnn::Activation_t::LINEAR,
141141
nullptr)));
142142
return INFINI_STATUS_SUCCESS;
143143
}));

0 commit comments

Comments
 (0)