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<<>>(input,output, size); + cudaFree(input); return CudaMatrix::fromRawData(output, aRows, aColumns); }