From ea68e6c5afe5b09d9e01935ca1279749d82d052f Mon Sep 17 00:00:00 2001
From: kradchen <chenjuijun@realcan.cn>
Date: Tue, 19 Dec 2023 13:10:12 +0800
Subject: [PATCH] Add convertfp16tofloatCuda memory free

---
 src/Function1D.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/Function1D.cu b/src/Function1D.cu
index 5186021..1991706 100644
--- a/src/Function1D.cu
+++ b/src/Function1D.cu
@@ -1631,5 +1631,6 @@ CudaMatrix Aurora::convertfp16tofloatCuda(short* aData, int aRows, int aColumns)
     cudaMalloc((void**)&output,size*sizeof(float));
     int blocksPerGrid = (size + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK;
     convertValueKernel<<<blocksPerGrid, THREADS_PER_BLOCK>>>(input,output, size);
+    cudaFree(input);
     return CudaMatrix::fromRawData(output, aRows, aColumns);
 }