SConscript 835 B

1234567891011121314151617181920212223242526272829
  1. from building import *
  2. import os
  3. cwd = GetCurrentDir()
  4. src = Glob('*.c') + Glob('*.cc')
  5. #.
  6. root = str(Dir('#'))
  7. packages = os.path.join(root, 'packages')
  8. file_list = os.listdir(packages)
  9. for f in file_list:
  10. if(f.split('-')[0] == 'TensorflowLiteMicro'):
  11. tflm_pkg = os.path.join(packages, f)
  12. break
  13. #./third_party/flatbuffer/include
  14. flatbuffer = os.path.join(tflm_pkg, "third_party/flatbuffers/include")
  15. #./third_party/gemmlowp
  16. gemmlowp = os.path.join(tflm_pkg, "third_party/gemmlowp")
  17. #./third_party/kissfft
  18. kissfft = os.path.join(tflm_pkg, "third_party/kissfft")
  19. #./third_party/ruy
  20. ruy = os.path.join(tflm_pkg, "third_party/ruy")
  21. CPPPATH = [tflm_pkg, flatbuffer, gemmlowp, kissfft, ruy]
  22. group = DefineGroup('lite/c', src, depend = ['PKG_USING_TENSORFLOWLITEMICRO'], CPPPATH = CPPPATH)
  23. Return('group')