drivanov commented on issue #16218: Improving performance of argmax operator URL: https://github.com/apache/incubator-mxnet/pull/16218#issuecomment-534654101 @access2rohit: Regarding the crash you see in your benchmark... Following line from your output: ``` mxnet.base.MXNetError: [22:48:58] src/operator/tensor/./broadcast_reduce_op.h:689: Global reduction not supported yet ``` shows that there is some problem with the definition of `axis` parameter. I am not sure, perhaps, it's is not defined in your test at all or it's default value is not yet supported. This is an error message from following to lines of C++ code: ``` const ReduceAxisParam& param = nnvm::get<ReduceAxisParam>(attrs.parsed); if (!param.axis) LOG(FATAL) << "Global reduction not supported yet"; ``` When everything fine, `param.axis` should NOT be `null_ptr`, otherwise `param.axis.values()` would fail.
---------------------------------------------------------------- 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: [email protected] With regards, Apache Git Services
