From 2c7571c371a7dfd532f1ec237a9a470405bf4699 Mon Sep 17 00:00:00 2001
From: lucasdelimanogueira <lucasln96@hotmail.com>
Date: Wed, 22 May 2024 20:59:46 -0300
Subject: [PATCH] small fix sum cuda

---
 norch/csrc/tensor.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/norch/csrc/tensor.cpp b/norch/csrc/tensor.cpp
index 1fc95e1..1f374f8 100644
--- a/norch/csrc/tensor.cpp
+++ b/norch/csrc/tensor.cpp
@@ -211,6 +211,7 @@ extern "C" {
 
             float* result_data;
             cudaMalloc((void**)&result_data, size * sizeof(float));
+            cudaMemset(result_data, 0, size * sizeof(float));
             sum_tensor_cuda(tensor, result_data, axis);
             
             if (keepdim) {