Google Git
Sign in
android / platform / external / pytorch / 09a7788a2f / . / aten / src / TH
tree: b3c1249e23a7a2aad5ccb82399c0a8194fa203e7 [path history] [tgz]
  1. generic/
  2. vector/
  3. CMakeLists.txt
  4. README.md
  5. TH.h
  6. THAllocator.cpp
  7. THAllocator.h
  8. THBlas.cpp
  9. THBlas.h
  10. THBlasUtils.h
  11. THDiskFile.cpp
  12. THDiskFile.h
  13. THFile.cpp
  14. THFile.h
  15. THFilePrivate.h
  16. THGeneral.cpp
  17. THGeneral.h.in
  18. THGenerateAllTypes.h
  19. THGenerateBFloat16Type.h
  20. THGenerateBoolType.h
  21. THGenerateByteType.h
  22. THGenerateCharType.h
  23. THGenerateDoubleType.h
  24. THGenerateFloatType.h
  25. THGenerateFloatTypes.h
  26. THGenerateHalfType.h
  27. THGenerateIntType.h
  28. THGenerateIntTypes.h
  29. THGenerateLongType.h
  30. THGenerateQInt32Type.h
  31. THGenerateQInt8Type.h
  32. THGenerateQTypes.h
  33. THGenerateQUInt8Type.h
  34. THGenerateShortType.h
  35. THGenerator.hpp
  36. THHalf.h
  37. THLapack.cpp
  38. THLapack.h
  39. THMemoryFile.cpp
  40. THMemoryFile.h
  41. THStorage.h
  42. THStorageFunctions.cpp
  43. THStorageFunctions.h
  44. THStorageFunctions.hpp
  45. THTensor.cpp
  46. THTensor.h
  47. THTensor.hpp
  48. THTensorApply.h
  49. THTensorDimApply.h
  50. THTensorEvenMoreMath.cpp
  51. THTensorFill.cpp
  52. THTensorLapack.cpp
  53. THTensorMath.cpp
  54. THTensorMoreMath.cpp
  55. THTensorRandom.cpp
  56. THVector.cpp
  57. THVector.h
aten/src/TH/README.md

Environment variables control the disabling of certain explicit SIMD optimizations.

x64 options:
ATEN_CPU_CAPABILITY=avx2    # Force AVX2 codepaths to be used
ATEN_CPU_CAPABILITY=avx     # Force AVX codepaths to be used
ATEN_CPU_CAPABILITY=default # Use oldest supported vector instruction set

ppc64le options:
TH_NO_VSX=1  # disable VSX codepaths