diff options
author | Soumith Chintala <soumith@gmail.com> | 2016-05-19 18:56:50 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2016-05-19 18:56:50 +0300 |
commit | 369bc800a09fbddb7107e250a464c14a72180023 (patch) | |
tree | cc87cfcbe1d488bc59bc799a78546f04fc96c17e | |
parent | a5c84ee87ec65063871f0c87993111ad6b98e90e (diff) | |
parent | d191b31801a1350d5d732257b204218cbed6087a (diff) |
Merge pull request #412 from fmassa/baddbmm_fix
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]; |