Fix memory issues in tests (#5183)

Commit

3 years ago

Fix memory issues in tests (#5183) * fix memory issues * set _offload_gpu_id * set gpu offload id