pytorch\_pfn\_extras.cuda
=========================

.. automodule:: pytorch_pfn_extras.cuda

   
   
   

   
   
   .. rubric:: Functions

   .. autosummary::
      :toctree: .

      
      pytorch_pfn_extras.cuda.stream
      pytorch_pfn_extras.cuda.use_default_mempool_in_cupy
      pytorch_pfn_extras.cuda.use_torch_mempool_in_cupy