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)
