build.py

   1 #!/usr/bin/env python-for-pytorch
   2
   3 import os
   4 from torch.utils.ffi import create_extension
   5
   6 abs_path = os.path.dirname(os.path.abspath(__file__))
   7
   8 ffi = create_extension(
   9     '_ext.flatland',
  10     headers = [ 'flatland.h' ],
  11     sources = [ 'flatland.c' ],
  12     extra_objects = [ abs_path + '/flatland_generator.so' ],
  13     libraries = [ ],
  14     library_dirs = [ ],
  15     define_macros = [ ],
  16     with_cuda = False
  17 )
  18
  19 if __name__ == '__main__':
  20     ffi.build()