sunwen/Aurora
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add convertfp16tofloatCuda memory free

Browse Source
This commit is contained in:
kradchen
2023-12-19 13:10:12 +08:00
parent cd10ebb5e8
commit ea68e6c5af
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/Function1D.cu
View File

@@ -1631,5 +1631,6 @@ CudaMatrix Aurora::convertfp16tofloatCuda(short* aData, int aRows, int aColumns)
cudaMalloc((void**)&output,size*sizeof(float)); cudaMalloc((void**)&output,size*sizeof(float));
int blocksPerGrid = (size + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK; int blocksPerGrid = (size + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK;
convertValueKernel<<<blocksPerGrid, THREADS_PER_BLOCK>>>(input,output, size); convertValueKernel<<<blocksPerGrid, THREADS_PER_BLOCK>>>(input,output, size);
cudaFree(input);
return CudaMatrix::fromRawData(output, aRows, aColumns); return CudaMatrix::fromRawData(output, aRows, aColumns);
} }