singa-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From "wangwei (JIRA)" <>
Subject [jira] [Created] (SINGA-287) Add memory size check for cudnn convolution
Date Wed, 21 Dec 2016 10:14:58 GMT
wangwei created SINGA-287:

             Summary: Add memory size check for cudnn convolution
                 Key: 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

View raw message