Moved numpy conversion inside CPU branch to avoid unnecessary conversions when using CUDA. Added memory pool cleanup for CUDA to prevent memory leaks.
Moved numpy conversion inside CPU branch to avoid unnecessary conversions when using CUDA. Added memory pool cleanup for CUDA to prevent memory leaks.