You are viewing a plain text version of this content. The canonical link for it is here.

Posted to dev@singa.apache.org by "wangwei (JIRA)" <ji...@apache.org> on 2016/12/21 10:14:58 UTC

[jira] [Created] (SINGA-287) Add memory size check for cudnn convolution

wangwei created SINGA-287:
-----------------------------

             Summary: Add memory size check for cudnn convolution
                 Key: SINGA-287
                 URL: https://issues.apache.org/jira/browse/SINGA-287
             Project: Singa
          Issue Type: Improvement
            Reporter: wangwei
            Assignee: wangwei


display a warning message if the cudnn conv workspace size is too large (>512MB); 
add size check when malloc memory for tensor. Singa tensor casts the memory size from size_t into int, which could be a negative value (rare case, happens when  the size is very large) and hence the malloc (by the Device) would fail.




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