Merge pull request #332 from InfiniTensor/issue/277

Issue/277: 为 ReLU 的九齿实现接入天数设备

README.md

@@ -155,7 +155,7 @@ pip install -e .
 2. 在 `InfiniCore` 文件夹下运行以下命令 AOT 编译库中的九齿算子：
 
 ```shell
-PYTHONPATH=src/ python scripts/build_ntops.py
+PYTHONPATH=${PYTHONPATH}:src python scripts/build_ntops.py
 ```
 
 注：如果对九齿相关文件有修改，需要重新构建 InfiniCore 时，也需要同时运行以上命令进行重新生成。

src/infiniop/ops/relu/operator.cc

@@ -5,7 +5,7 @@
 #ifdef ENABLE_CPU_API
 #include "cpu/relu_cpu.h"
 #endif
-#ifdef ENABLE_NVIDIA_API
+#if defined(ENABLE_NVIDIA_API) || defined(ENABLE_ILUVATAR_API)
 #ifdef ENABLE_NINETOOTHED
 #include "nvidia/relu_nvidia.cuh"
 #endif
@@ -40,6 +40,11 @@ __C infiniStatus_t infiniopCreateReluDescriptor(
         CREATE(INFINI_DEVICE_NVIDIA, nvidia);
 #endif
 #endif
+#ifdef ENABLE_ILUVATAR_API
+#ifdef ENABLE_NINETOOTHED
+        CREATE(INFINI_DEVICE_ILUVATAR, nvidia);
+#endif
+#endif
 #ifdef ENABLE_METAX_API
 #ifdef ENABLE_NINETOOTHED
         CREATE(INFINI_DEVICE_METAX, metax);
@@ -69,6 +74,11 @@ __C infiniStatus_t infiniopGetReluWorkspaceSize(infiniopReluDescriptor_t desc, s
         GET(INFINI_DEVICE_NVIDIA, nvidia)
 #endif
 #endif
+#ifdef ENABLE_ILUVATAR_API
+#ifdef ENABLE_NINETOOTHED
+        GET(INFINI_DEVICE_ILUVATAR, nvidia)
+#endif
+#endif
 #ifdef ENABLE_METAX_API
 #ifdef ENABLE_NINETOOTHED
         GET(INFINI_DEVICE_METAX, metax)
@@ -105,6 +115,11 @@ __C infiniStatus_t infiniopRelu(
         CALCULATE(INFINI_DEVICE_NVIDIA, nvidia);
 #endif
 #endif
+#ifdef ENABLE_ILUVATAR_API
+#ifdef ENABLE_NINETOOTHED
+        CALCULATE(INFINI_DEVICE_ILUVATAR, nvidia);
+#endif
+#endif
 #ifdef ENABLE_METAX_API
 #ifdef ENABLE_NINETOOTHED
         CALCULATE(INFINI_DEVICE_METAX, metax);
@@ -136,6 +151,11 @@ infiniopDestroyReluDescriptor(infiniopReluDescriptor_t desc) {
         DELETE(INFINI_DEVICE_NVIDIA, nvidia);
 #endif
 #endif
+#ifdef ENABLE_ILUVATAR_API
+#ifdef ENABLE_NINETOOTHED
+        DELETE(INFINI_DEVICE_ILUVATAR, nvidia);
+#endif
+#endif
 #ifdef ENABLE_METAX_API
 #ifdef ENABLE_NINETOOTHED
         DELETE(INFINI_DEVICE_METAX, metax);

xmake/iluvatar.lua

@@ -49,6 +49,10 @@ target("infiniop-iluvatar")
 
     -- set_languages("cxx17") 天数似乎不能用这个配置
     add_files("../src/infiniop/devices/nvidia/*.cu", "../src/infiniop/ops/*/nvidia/*.cu")
+
+    if has_config("ninetoothed") then
+        add_files("../build/ninetoothed/*.c", {cxflags = {"-Wno-return-type"}})
+    end
 target_end()
 
 target("infinirt-iluvatar")