sxjscience commented on a change in pull request #16979: [Bugfix] [Numpy] Add
`kAddTo` and kNullOp to Transpose
URL: https://github.com/apache/incubator-mxnet/pull/16979#discussion_r354652158
##########
File path: src/operator/tensor/pseudo2DTranspose_op-inl.cuh
##########
@@ -78,23 +85,34 @@ __global__ void transpose_pseudo2D(DType* out, DType* inp,
}
__syncthreads();
- // read from shared to registers
- transp_t tmp[TSR];
+ // read from shared to local registers
+ CType tmp[TSR];
#pragma unroll
for (index_t i = 0; i < TSR; i++) {
+ DType* tmp_dptr = reinterpret_cast<DType*>(&tmp[i]);
Review comment:
Have you find any problem? I can compile that and it also passed CI.
----------------------------------------------------------------
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:
us...@infra.apache.org
With regards,
Apache Git Services
