look into the memory leak #87
Description
As mentioned in this PR #74 , there is a memory leak concern on GPU. check if this is solved and add tests
{{ message }}
As mentioned in this PR #74 , there is a memory leak concern on GPU. check if this is solved and add tests