pytorch\_pfn\_extras.cuda.use\_torch\_mempool\_in\_cupy ======================================================= .. currentmodule:: pytorch_pfn_extras.cuda .. autofunction:: use_torch_mempool_in_cupy