apache/mxnet

too large launch parameter: MapReduceKeepDim1

Closed this issue · 1 comments

error message

cuda/tensor_gpu-inl.cuh:58: too large launch parameter: MapReduceKeepDim1[70000], [256,1,1]

this will happen when use mx.sym.L2Normalization(), and one dim of the shape is bigger than 65536.
how to fix it? @sxjscience @piiswrong
some similar issue which not solved dmlc/mshadow#131

thanks.

this has been fixed from dmlc/mshadow#285