wangwei created SINGA-175:
-----------------------------

             Summary: Add memory manange API and implement a subclass using 
CNMeM
                 Key: SINGA-175
                 URL: https://issues.apache.org/jira/browse/SINGA-175
             Project: Singa
          Issue Type: New Feature
            Reporter: wangwei


We have associated each Tensor associated with a reference counter for garbage 
collection. Tensor is then can be used without worrying about memory leak.
To avoid frequently malloc/free gpu or cpu memory due to Tensor creation and 
deletion, we need a memory pool. This ticket is going to create a memory 
manager that manages a memory pool for each device.
Particularly, we will implement a specific memory manager using the 
https://github.com/NVIDIA/cnmem for CudaGPU devices.



--
This message was sent by Atlassian JIRA
(v6.3.4#6332)

Reply via email to