diff options
author | fsuzanomassa <fvsmassa@gmail.com> | 2016-05-19 18:44:24 +0300 |
---|---|---|
committer | fsuzanomassa <fvsmassa@gmail.com> | 2016-05-19 18:44:24 +0300 |
commit | d191b31801a1350d5d732257b204218cbed6087a (patch) | |
tree | cc87cfcbe1d488bc59bc799a78546f04fc96c17e | |
parent | ca79216059423d47ed2d04b24334f8ed41744df5 (diff) |
Fix baddbmm for non-contiguous results tensor
-rw-r--r-- | lib/THC/THCTensorMathBlas.cu | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/lib/THC/THCTensorMathBlas.cu b/lib/THC/THCTensorMathBlas.cu index 230fe95..0e373d8 100644 --- a/lib/THC/THCTensorMathBlas.cu +++ b/lib/THC/THCTensorMathBlas.cu @@ -321,8 +321,9 @@ void THCudaTensor_baddbmm(THCState *state, THCudaTensor *result, float beta, THC { transpose_result = false; - result_ = THCudaTensor_newWithSize3d(state, result->size[0], result->size[2], result->size[1]); - THCudaTensor_copy(state, result_, result); + THCudaTensor *transp_r_ = THCudaTensor_newTranspose(state, result, 1, 2); + result_ = THCudaTensor_newClone(state, transp_r_); + THCudaTensor_free(state, transp_r_); THCudaTensor_transpose(state, result_, NULL, 1, 2); ldc = result_->stride[2]; |