HyperZealot edited a comment on issue #12997: A better take forward kernel for CPU URL: https://github.com/apache/incubator-mxnet/pull/12997#issuecomment-434408459 @rongzha1 From your code M should be num_cols(61400), that is a pretty big number. Update: I tested my changes with a smaller M too, with: num_rows = 1000000 test_rows = 20000 num_cols = 8 I got: before: 1.8296743018731323 GB/s after: 10.363484717590893 GB/s
---------------------------------------------------------------- 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
