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 ;
0 commit comments