szha commented on a change in pull request #12804: CudnnFind() usage 
improvements
URL: https://github.com/apache/incubator-mxnet/pull/12804#discussion_r227973746
 
 

 ##########
 File path: perl-package/AI-MXNetCAPI/mxnet.i
 ##########
 @@ -349,7 +349,7 @@ int MXGetGPUCount(int* out);
  * \param total_mem pointer to the integer holding total GPU memory
  * \return 0 when success, -1 when failure happens
  */
-int MXGetGPUMemoryInformation(int dev, int *out, int *out);
+int MXGetGPUMemoryInformation(int dev, uint64_t *out, uint64_t *out);
 
 Review comment:
   I think @DickJC123 you meant 2GB? One alternative is to provide the new 
interface as MXGetGPUMemoryInformation64 and update the old interface in major 
version

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
[email protected]


With regards,
Apache Git Services

Reply via email to