From e2a0262c4e0658282f2815f1b0d92b5556a9f5ca Mon Sep 17 00:00:00 2001
From: lucasdelimanogueira <lucasln96@hotmail.com>
Date: Thu, 23 May 2024 01:29:23 -0300
Subject: [PATCH] small fix max cuda axis was setting init to 0 instead of -inf

---
 norch/csrc/cuda.cu | 1 -
 1 file changed, 1 deletion(-)

diff --git a/norch/csrc/cuda.cu b/norch/csrc/cuda.cu
index 9150d60..fdaaba5 100644
--- a/norch/csrc/cuda.cu
+++ b/norch/csrc/cuda.cu
@@ -328,7 +328,6 @@ __host__ void max_tensor_cuda(Tensor* tensor, float* result_data, int axis) {
         // Initialize result_data to 0
         float neg_inf = -FLT_MAX;
         cudaMemset(result_data, *reinterpret_cast<int*>(&neg_inf), result_size * sizeof(float));
-        cudaMemset(result_data, 0, result_size * sizeof(float));
 
         int num_threads = result_size;
         int num_blocks = (num_threads + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK;