[
https://issues.apache.org/jira/browse/SINGA-287?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
wangwei resolved SINGA-287.
---------------------------
Resolution: Fixed
> 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)
