Skip to content

Commit 0126430

Browse files
committed
issue/1158 - add missing empty impl for ascend, kunlunxin, and moore
1 parent 4577ffb commit 0126430

4 files changed

Lines changed: 24 additions & 39 deletions

File tree

src/infiniop/ops/fused_ffn/operator.cc

Lines changed: 0 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -8,24 +8,9 @@
88
#if defined(ENABLE_NVIDIA_API) || defined(ENABLE_ILUVATAR_API) || defined(ENABLE_QY_API) || defined(ENABLE_HYGON_API) || defined(ENABLE_ALI_API)
99
#include "nvidia/fused_ffn_nvidia.cuh"
1010
#endif
11-
#ifdef ENABLE_ASCEND_API
12-
// TODO: Add Ascend implementation
13-
// #include "ascend/fused_ffn_aclnn.h"
14-
#endif
15-
#ifdef ENABLE_CAMBRICON_API
16-
// TODO: Add Cambricon implementation
17-
// #include "bang/fused_ffn_bang.h"
18-
#endif
1911
#ifdef ENABLE_METAX_API
2012
#include "metax/fused_ffn_metax.cuh"
2113
#endif
22-
#ifdef ENABLE_MOORE_API
23-
#include "moore/fused_ffn_moore.h"
24-
#endif
25-
#ifdef ENABLE_KUNLUN_API
26-
// TODO: Add Kunlun implementation
27-
// #include "kunlun/fused_ffn_kunlun.h"
28-
#endif
2914

3015
__INFINI_C infiniStatus_t infiniopCreateFusedFFNDescriptor(
3116
infiniopHandle_t handle,
@@ -64,9 +49,6 @@ __INFINI_C infiniStatus_t infiniopCreateFusedFFNDescriptor(
6449
#ifdef ENABLE_ALI_API
6550
CREATE(INFINI_DEVICE_ALI, nvidia);
6651
#endif
67-
#ifdef ENABLE_MOORE_API
68-
CREATE(INFINI_DEVICE_MOORE, moore);
69-
#endif
7052
#ifdef ENABLE_METAX_API
7153
CREATE(INFINI_DEVICE_METAX, metax);
7254
#endif
@@ -75,9 +57,6 @@ __INFINI_C infiniStatus_t infiniopCreateFusedFFNDescriptor(
7557
#endif
7658
#ifdef ENABLE_HYGON_API
7759
CREATE(INFINI_DEVICE_HYGON, nvidia);
78-
#endif
79-
#ifdef ENABLE_KUNLUN_API
80-
// CREATE(INFINI_DEVICE_KUNLUN, kunlun);
8160
#endif
8261
default:
8362
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
@@ -107,9 +86,6 @@ __INFINI_C infiniStatus_t infiniopGetFusedFFNWorkspaceSize(
10786
#ifdef ENABLE_ALI_API
10887
GET(INFINI_DEVICE_ALI, nvidia);
10988
#endif
110-
#ifdef ENABLE_MOORE_API
111-
GET(INFINI_DEVICE_MOORE, moore);
112-
#endif
11389
#ifdef ENABLE_METAX_API
11490
GET(INFINI_DEVICE_METAX, metax);
11591
#endif
@@ -118,9 +94,6 @@ __INFINI_C infiniStatus_t infiniopGetFusedFFNWorkspaceSize(
11894
#endif
11995
#ifdef ENABLE_HYGON_API
12096
GET(INFINI_DEVICE_HYGON, nvidia);
121-
#endif
122-
#ifdef ENABLE_KUNLUN_API
123-
// GET(INFINI_DEVICE_KUNLUN, kunlun);
12497
#endif
12598
default:
12699
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
@@ -162,9 +135,6 @@ __INFINI_C infiniStatus_t infiniopFusedFFN(
162135
#ifdef ENABLE_ALI_API
163136
CALCULATE(INFINI_DEVICE_ALI, nvidia);
164137
#endif
165-
#ifdef ENABLE_MOORE_API
166-
CALCULATE(INFINI_DEVICE_MOORE, moore);
167-
#endif
168138
#ifdef ENABLE_METAX_API
169139
CALCULATE(INFINI_DEVICE_METAX, metax);
170140
#endif
@@ -173,9 +143,6 @@ __INFINI_C infiniStatus_t infiniopFusedFFN(
173143
#endif
174144
#ifdef ENABLE_HYGON_API
175145
CALCULATE(INFINI_DEVICE_HYGON, nvidia);
176-
#endif
177-
#ifdef ENABLE_KUNLUN_API
178-
// CALCULATE(INFINI_DEVICE_KUNLUN, kunlun);
179146
#endif
180147
default:
181148
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
@@ -207,9 +174,6 @@ __INFINI_C infiniStatus_t infiniopDestroyFusedFFNDescriptor(
207174
#ifdef ENABLE_ALI_API
208175
DESTROY(INFINI_DEVICE_ALI, nvidia);
209176
#endif
210-
#ifdef ENABLE_MOORE_API
211-
DESTROY(INFINI_DEVICE_MOORE, moore);
212-
#endif
213177
#ifdef ENABLE_METAX_API
214178
DESTROY(INFINI_DEVICE_METAX, metax);
215179
#endif
@@ -218,9 +182,6 @@ __INFINI_C infiniStatus_t infiniopDestroyFusedFFNDescriptor(
218182
#endif
219183
#ifdef ENABLE_HYGON_API
220184
DESTROY(INFINI_DEVICE_HYGON, nvidia);
221-
#endif
222-
#ifdef ENABLE_KUNLUN_API
223-
// DESTROY(INFINI_DEVICE_KUNLUN, kunlun);
224185
#endif
225186
default:
226187
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;

src/infinirt/ascend/infinirt_ascend.cc

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -180,5 +180,13 @@ infiniStatus_t graphLuanch(infinirtGraphExec_t graph_exec, infinirtStream_t stre
180180
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
181181
}
182182

183+
infiniStatus_t getMemInfo(int device_id, size_t *free_bytes, size_t *total_bytes) {
184+
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
185+
}
186+
187+
infiniStatus_t getDeviceResourceSnapshot(int device_id, infinirtDeviceResourceSnapshot_t *snapshot) {
188+
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
189+
}
190+
183191
} // namespace infinirt::ascend
184192
#undef CHECK_ACLRT

src/infinirt/kunlun/infinirt_kunlun.cc

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -182,4 +182,12 @@ infiniStatus_t graphLuanch(infinirtGraphExec_t graph_exec, infinirtStream_t stre
182182
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
183183
}
184184

185+
infiniStatus_t getMemInfo(int device_id, size_t *free_bytes, size_t *total_bytes) {
186+
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
187+
}
188+
189+
infiniStatus_t getDeviceResourceSnapshot(int device_id, infinirtDeviceResourceSnapshot_t *snapshot) {
190+
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
191+
}
192+
185193
} // namespace infinirt::kunlun

src/infinirt/moore/infinirt_moore.cc

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -168,4 +168,12 @@ infiniStatus_t graphLuanch(infinirtGraphExec_t graph_exec, infinirtStream_t stre
168168
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
169169
}
170170

171+
infiniStatus_t getMemInfo(int device_id, size_t *free_bytes, size_t *total_bytes) {
172+
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
173+
}
174+
175+
infiniStatus_t getDeviceResourceSnapshot(int device_id, infinirtDeviceResourceSnapshot_t *snapshot) {
176+
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
177+
}
178+
171179
} // namespace infinirt::musa

0 commit comments

Comments
 (0)