Migrate add layer to tensors

2025-12-22 14:24:22 +00:00 · 2025-11-22 00:12:20 +01:00
parent aeb1739c46
commit 4c8b2ef537
7 changed files with 56 additions and 101 deletions
--- a/include/backend/cuda.cuh
+++ b/include/backend/cuda.cuh
@@ -86,6 +86,12 @@ class CUDA : public Backend {
        CUDANet::Tensor& input_b,
        CUDANet::Tensor& output
    ) override;
+
+    CUDANet::Tensor& add(
+        CUDANet::Tensor& input_a,
+        CUDANet::Tensor& input_b,
+        CUDANet::Tensor& output
+    ) override;
 };

 }  // namespace CUDANet::Backend