jonatan1626 commented on issue #17462: Updated PartialSortSmallK for LT support URL: https://github.com/apache/incubator-mxnet/pull/17462#issuecomment-579961169 My gpu memory usage results from running topk operation with Large Tensor and without Large Tensor. | With Large Tensor | Without Large Tensor -- | -- | -- | Avg (Mb) | Max (Mb) | Avg (Mb) | Max (Mb) Both | 1001.66667 | 1410 | 945.333333 | 1264 Indicies | 984 | 1410 | 828.333333 | 1250 Value | 998.428571 | 1371 | 902.666667 | 1264
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services