apeforest commented on issue #13418: [MXNET-1185] Support large array in several operators (part 1) URL: https://github.com/apache/incubator-mxnet/pull/13418#issuecomment-442937625 @anirudh2290 This only changes the index type from int32_t to int64_t, not the data type of each element. So the performance impact is different from @wkcn test. I also did a test in broadcast_div operator in MXNet and found no obvious runtime change (int64_t version actually is slightly faster) `import mxnet as mx` `import numpy as np` `from mxnet.test_utils import check_speed` `a = mx.sym.ones(shape=(100000, 10000))` `b = mx.sym.ones(shape=(100000, 1)) * 2` `c = mx.sym.broadcast_div(a, b)` `print(check_speed(c))` Master branch: 15.357867193222045 seconds PR branch: 12.930877685546875 seconds
---------------------------------------------------------------- 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: us...@infra.apache.org With regards, Apache Git Services
