diff options
author | jokeren <robinho364@gmail.com> | 2017-01-31 17:58:42 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2017-02-23 14:01:13 +0300 |
commit | 8701a03bebae65d37e7585d9577cb6faeccf0a32 (patch) | |
tree | 9669fef265e636fa08a8841415f33121a06d1f4e | |
parent | e4c263784baf5e5f7d62c4be2156d7c3f900fa26 (diff) |
THTensorApply2 copy optimization
-rw-r--r-- | lib/TH/generic/THTensorCopy.c | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/lib/TH/generic/THTensorCopy.c b/lib/TH/generic/THTensorCopy.c index 5dfdcf1..cc92c33 100644 --- a/lib/TH/generic/THTensorCopy.c +++ b/lib/TH/generic/THTensorCopy.c @@ -15,6 +15,8 @@ void THTensor_(copy)(THTensor *tensor, THTensor *src) for (i=0; i<sz; ++i) rp[i] = sp[i]; #endif + } else if (THTensor_(isContiguous)(tensor) || THTensor_(isContiguous)(src)) { + TH_TENSOR_APPLY2_CONTIGUOUS(real, tensor, real, src, *tensor_data = *src_data;) } else { TH_TENSOR_APPLY2(real, tensor, real, src, *tensor_data = *src_data;) } |