commit fc23f65d4f6bc6b99e61ce577d84072cee1724b0
author Dominik Grewe <dominikg@google.com> Fri Oct 23 09:16:50 2015 +0100
committer Dominik Grewe <dominikg@google.com> Fri Oct 23 09:16:50 2015 +0100
tree 068515768796c99d30f5e300d51ea6acfb82ef7a
parent b3e094ec1de3c35f545a9429f96aa489dfcfe7b4

Check that dimension to be sorted is contiguous when using Thrust.

THCTensorSort.cu

diff --git a/THCTensorSort.cu b/THCTensorSort.cu
index 84b8c6d..1bc3826 100644
--- a/THCTensorSort.cu
+++ b/THCTensorSort.cu
@@ -377,6 +377,9 @@
   long sliceStride = THCudaTensor_stride(state, input, dim);
   long numSlices = totalElements / sliceSize;
 
+  THArgCheck(THCudaTensor_stride(state, input, dim) == 1, 1,
+             "The dimension to be sorted must be contiguous.");
+
   // Copy input to sorted, since we sort in place
   if (sorted != input) {
     THCudaTensor_copy(state, sorted, input);