[ 
https://issues.apache.org/jira/browse/SINGA-287?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15768810#comment-15768810
 ] 

ASF subversion and git services commented on SINGA-287:
-------------------------------------------------------

Commit 458b0f6a8cb7a05081e1908465c552ba10d2b23f in incubator-singa's branch 
refs/heads/master from [~wangwei.cs]
[ https://git-wip-us.apache.org/repos/asf?p=incubator-singa.git;h=458b0f6 ]

SINGA-287 - Add memory size check for cudnn convolution

add warning if the cudnn conv workspace size is too large (>512MB);
add size check when malloc memory for tensor to make sure the size is >=0


> 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)

Reply via email to