Google Git
Sign in
android / platform / external / pytorch / 0df2f1cbd6
commit0df2f1cbd62ab2a7d507bc68d8d43509ca268a0e[log] [tgz]
authorFrancisco Massa <fvsmassa@gmail.com>Tue Sep 05 20:53:58 2017 +0100
committerSoumith Chintala <soumith@gmail.com>Sun Sep 10 13:50:50 2017 -0400
tree9d7d20860821efedeebc1a328fdc05825d855e04
parent8176a558277aeec831f2f1a846cb4856a58fb941 [diff]
Optimize pow for different exponents and add tests
1 file changed
tree: 9d7d20860821efedeebc1a328fdc05825d855e04
  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