issue/291/style: 根据实际情况将 cuda 改为 nvidia

Signed-off-by: YdrMaster <ydrml@hotmail.com>

src/infiniop/ops/sub/cpu/sub_cpu.h

@@ -3,7 +3,7 @@
 
 #include "../../../elementwise/cpu/elementwise_cpu.h"
 
-ELEMENTWISE_DESCRIPTOR(sub, cpu, cpu)
+ELEMENTWISE_DESCRIPTOR(sub, cpu)
 
 namespace op::sub::cpu {
 typedef struct SubOp {

src/infiniop/ops/sub/metax/sub_metax.h

@@ -3,6 +3,6 @@
 
 #include "../../../elementwise/metax/elementwise_metax_api.h"
 
-ELEMENTWISE_DESCRIPTOR(sub, metax, metax)
+ELEMENTWISE_DESCRIPTOR(sub, metax)
 
 #endif // __SUB_METAX_API_H__

src/infiniop/ops/sub/nvidia/sub_nvidia.cu

-#include "../../../elementwise/cuda/elementwise_cuda.cuh"
+#include "../../../elementwise/nvidia/elementwise_nvidia.cuh"
 
 #include "../cuda/kernel.cuh"
 #include "sub_nvidia.cuh"
@@ -13,7 +13,7 @@ infiniStatus_t Descriptor::create(
     infiniopTensorDescriptor_t out_desc,
     std::vector<infiniopTensorDescriptor_t> input_desc_vec) {
 
-    auto handle = reinterpret_cast<device::cuda::Handle *>(handle_);
+    auto handle = reinterpret_cast<device::nvidia::Handle *>(handle_);
     auto dtype = out_desc->dtype();
 
     const auto &a_desc = input_desc_vec.at(0);

src/infiniop/ops/sub/nvidia/sub_nvidia.cuh

 #ifndef __SUB_CUDA_API_H__
 #define __SUB_CUDA_API_H__
 
-#include "../../../elementwise/cuda/elementwise_cuda_api.cuh"
+#include "../../../elementwise/nvidia/elementwise_nvidia_api.cuh"
 
-ELEMENTWISE_DESCRIPTOR(sub, nvidia, cuda)
+ELEMENTWISE_DESCRIPTOR(sub, nvidia)
 
 #endif // __SUB_CUDA_API_H__

src/infiniop/ops/swiglu/cpu/swiglu_cpu.h

@@ -3,7 +3,7 @@
 
 #include "../../../elementwise/cpu/elementwise_cpu.h"
 
-ELEMENTWISE_DESCRIPTOR(swiglu, cpu, cpu)
+ELEMENTWISE_DESCRIPTOR(swiglu, cpu)
 
 namespace op::swiglu::cpu {
 typedef struct SwiGLUOp {

src/infiniop/ops/swiglu/metax/swiglu_metax.h

@@ -3,6 +3,6 @@
 
 #include "../../../elementwise/metax/elementwise_metax_api.h"
 
-ELEMENTWISE_DESCRIPTOR(swiglu, metax, metax)
+ELEMENTWISE_DESCRIPTOR(swiglu, metax)
 
 #endif // __SWIGLU_METAX_API_H__

src/infiniop/ops/swiglu/nvidia/swiglu_nvidia.cu

 #include "swiglu_nvidia.cuh"
 
-#include "../../../elementwise/cuda/elementwise_cuda.cuh"
+#include "../../../elementwise/nvidia/elementwise_nvidia.cuh"
 
 #include "../cuda/kernel.cuh"
 
@@ -14,7 +14,7 @@ infiniStatus_t Descriptor::create(
     infiniopTensorDescriptor_t out_desc,
     std::vector<infiniopTensorDescriptor_t> input_desc_vec) {
 
-    auto handle = reinterpret_cast<device::cuda::Handle *>(handle_);
+    auto handle = reinterpret_cast<device::nvidia::Handle *>(handle_);
     auto dtype = out_desc->dtype();
 
     const auto &up_desc = input_desc_vec.at(0);

src/infiniop/ops/swiglu/nvidia/swiglu_nvidia.cuh

 #ifndef __SWIGLU_CUDA_API_H__
 #define __SWIGLU_CUDA_API_H__
 
-#include "../../../elementwise/cuda/elementwise_cuda_api.cuh"
+#include "../../../elementwise/nvidia/elementwise_nvidia_api.cuh"
 
-ELEMENTWISE_DESCRIPTOR(swiglu, nvidia, cuda)
+ELEMENTWISE_DESCRIPTOR(swiglu, nvidia)
 
 #endif // __SWIGLU_CUDA_API_H__

xmake/cuda.lua

@@ -46,7 +46,7 @@ target("infiniop-cuda")
     add_cuflags("-Xcompiler=-Wno-error=deprecated-declarations")
 
     set_languages("cxx17")
-    add_files("../src/infiniop/devices/cuda/*.cu", "../src/infiniop/ops/*/cuda/*.cu", "../src/infiniop/ops/*/nvidia/*.cu", "../build/ninetoothed/*.c")
+    add_files("../src/infiniop/devices/nvidia/*.cu", "../src/infiniop/ops/*/cuda/*.cu", "../src/infiniop/ops/*/nvidia/*.cu", "../build/ninetoothed/*.c")
 target_end()
 
 target("infinirt-cuda")