piiswrong commented on a change in pull request #8566: optimize broadcast
URL: https://github.com/apache/incubator-mxnet/pull/8566#discussion_r150332277
##
File path: src/operator/mxnet_op.h
##
@@ -345,6 +394,13 @@ __global__ void mxnet_generic_kernel(int N, Args... args)
piiswrong commented on a change in pull request #8566: optimize broadcast
URL: https://github.com/apache/incubator-mxnet/pull/8566#discussion_r149823235
##
File path: src/operator/mxnet_op.h
##
@@ -345,6 +394,13 @@ __global__ void mxnet_generic_kernel(int N, Args... args)
piiswrong commented on a change in pull request #8566: optimize broadcast
URL: https://github.com/apache/incubator-mxnet/pull/8566#discussion_r149274245
##
File path: src/operator/mxnet_op.h
##
@@ -334,6 +365,24 @@ struct Kernel {
}
#endif
}
+
+ template
piiswrong commented on a change in pull request #8566: optimize broadcast
URL: https://github.com/apache/incubator-mxnet/pull/8566#discussion_r149274186
##
File path: src/operator/tensor/elemwise_binary_broadcast_op.h
##
@@ -133,13 +133,35 @@ inline int
piiswrong commented on a change in pull request #8566: optimize broadcast
URL: https://github.com/apache/incubator-mxnet/pull/8566#discussion_r149249010
##
File path: src/operator/mxnet_op.h
##
@@ -334,6 +365,24 @@ struct Kernel {
}
#endif
}
+
+ template