Google Git
Sign in
android / platform / external / pytorch / f425c5216b
commitf425c5216b7fe35dd03e0161a3440ec968c63636[log] [tgz]
authorLuca Antiga <luca.antiga@orobix.com>Wed Jun 07 02:21:20 2017 +0200
committerSoumith Chintala <soumith@gmail.com>Tue Jul 04 14:55:19 2017 -0400
treebdae6248865e564fcb66ffe23c86afa3fa3ca163
parente5bac2dd2d69772938482c1431db1fc1efb64c6f [diff]
Have median reduce over all dims and return just the value when dim is not provided
2 files changed
tree: bdae6248865e564fcb66ffe23c86afa3fa3ca163
  1. cmake/
  2. generic/
  3. vector/
  4. CMakeLists.txt
  5. README.md
  6. TH.h
  7. THAllocator.c
  8. THAllocator.h
  9. THAtomic.c
  10. THAtomic.h
  11. THBlas.c
  12. THBlas.h
  13. THConfig.cmake.in
  14. THDiskFile.c
  15. THDiskFile.h
  16. THFile.c
  17. THFile.h
  18. THFilePrivate.h
  19. THGeneral.c
  20. THGeneral.h.in
  21. THGenerateAllTypes.h
  22. THGenerateByteType.h
  23. THGenerateCharType.h
  24. THGenerateDoubleType.h
  25. THGenerateFloatType.h
  26. THGenerateFloatTypes.h
  27. THGenerateHalfType.h
  28. THGenerateIntType.h
  29. THGenerateIntTypes.h
  30. THGenerateLongType.h
  31. THGenerateShortType.h
  32. THHalf.c
  33. THHalf.h
  34. THLapack.c
  35. THLapack.h
  36. THLogAdd.c
  37. THLogAdd.h
  38. THMath.h
  39. THMemoryFile.c
  40. THMemoryFile.h
  41. THRandom.c
  42. THRandom.h
  43. THSize.c
  44. THSize.h
  45. THStorage.c
  46. THStorage.h
  47. THTensor.c
  48. THTensor.h
  49. THTensorApply.h
  50. THTensorDimApply.h
  51. THTensorMacros.h
  52. THVector.c
  53. THVector.h
README.md

Environment variables control the disabling of certain explicit SIMD optimizations.

x64 options:
TH_NO_AVX2=1 # disable AVX2 codepaths
TH_NO_AVX=1  # disable AVX codepaths
TH_NO_SSE=1  # disable SSE codepaths

ppc64le options:
TH_NO_VSX=1  # disable VSX codepaths