Google Git
Sign in
android / platform / external / pytorch / bc9c0c1598
commitbc9c0c159894ead7da579d2e9497bde41c992c16[log] [tgz]
authorOwen Yamauchi <oyamauchi@fb.com>Tue Sep 22 09:12:41 2015 -0700
committerOwen Yamauchi <oyamauchi@fb.com>Tue Sep 22 09:13:59 2015 -0700
treef6c7823a45847e64862e3308b15cf0ef2389a73d
parentd42a3fb3c7bda939f5b0edefa68f336a326b989e [diff]
Rewrite indexSelect kernel for correctness

The generic indexSelect kernel (for non-contiguous tensors) was broken;
totally rewrite it for correctness.

Closes #240.
3 files changed
tree: f6c7823a45847e64862e3308b15cf0ef2389a73d
  1. CMakeLists.txt
  2. THC.h
  3. THCAllocator.c
  4. THCAllocator.h
  5. THCApply.cu
  6. THCApply.cuh
  7. THCBlas.cu
  8. THCBlas.h
  9. THCDeviceTensor-inl.cuh
  10. THCDeviceTensor.cuh
  11. THCDeviceTensorUtils-inl.cuh
  12. THCDeviceTensorUtils.cuh
  13. THCDeviceUtils.cuh
  14. THCGeneral.c
  15. THCGeneral.h.in
  16. THCReduce.cuh
  17. THCReduceAll.cuh
  18. THCReduceApplyUtils.cu
  19. THCReduceApplyUtils.cuh
  20. THCStorage.c
  21. THCStorage.cu
  22. THCStorage.h
  23. THCStorageCopy.c
  24. THCStorageCopy.cu
  25. THCStorageCopy.h
  26. THCTensor.c
  27. THCTensor.cu
  28. THCTensor.h
  29. THCTensorConv.cu
  30. THCTensorConv.h
  31. THCTensorCopy.c
  32. THCTensorCopy.cu
  33. THCTensorCopy.h
  34. THCTensorIndex.cu
  35. THCTensorMasked.cu
  36. THCTensorMath.cu
  37. THCTensorMath.h
  38. THCTensorMath2.cu
  39. THCTensorMathBlas.cu
  40. THCTensorMathCompare.cu
  41. THCTensorMathCompareT.cu
  42. THCTensorMathMagma.cu
  43. THCTensorMathPairwise.cu
  44. THCTensorMathPointwise.cu
  45. THCTensorMathScan.cu
  46. THCTensorMathTransformReduce.cu
  47. THCTensorRandom.cu
  48. THCTensorRandom.h
  49. THCTensorScatterGather.cu
  50. THCTensorSort.cu
  51. THCTensorSort.h