Skip to content

Commit

Permalink
Merge pull request torch#391 from hughperkins/short-explanation-of-ke…
Browse files Browse the repository at this point in the history
…y-values-sort

description that sort orders the 'key' tensor numerically
  • Loading branch information
soumith committed Apr 15, 2016
2 parents adc1e54 + 3025080 commit bf8f88b
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions lib/THC/THCTensorSort.cu
Original file line number Diff line number Diff line change
Expand Up @@ -106,6 +106,9 @@ void THCudaTensor_fillSliceWithIndex(THCState* state,
THCudaCheck(cudaGetLastError());
}

// In alignment with default sort on a c++ map, this function
// will permute key and value tensors identically, and
// in such a way that the 'key' tensor is ordered numerically
THC_API void THCudaTensor_sortKeyValueInplace(THCState* state,
THCudaTensor* key,
THCudaTensor* value,
Expand Down

0 comments on commit bf8f88b

Please sign in to comment.