AdvancedPhotonSource / tike

Repository for ptychography software
http://tike.readthedocs.io
Other
29 stars 15 forks source link

BUG: Don't initialize CUDA Streams at import time #292

Closed carterbox closed 10 months ago

carterbox commented 10 months ago

Purpose

Prevent GPU memory from being allocated at import time on the 0th GPU.

Approach

Streams take GPU memory and may be unused if only non-GPU functions are used, so wait until functions are called to do initialization.

Pre-Merge Checklists

Submitter

Reviewer