look into the memory leak

[Saibo-creator]

As mentioned in this PR #74 , there is a memory leak concern on GPU. check if this is solved and add tests

[Saibo-creator]

@nathanrchn