Oups.
[pysvrt.git] / build.py
1 #!/usr/bin/env python
2
3 #  svrt is the ``Synthetic Visual Reasoning Test'', an image
4 #  generator for evaluating classification performance of machine
5 #  learning systems, humans and primates.
6 #
7 #  Copyright (c) 2017 Idiap Research Institute, http://www.idiap.ch/
8 #  Written by Francois Fleuret <francois.fleuret@idiap.ch>
9 #
10 #  This file is part of svrt.
11 #
12 #  svrt is free software: you can redistribute it and/or modify it
13 #  under the terms of the GNU General Public License version 3 as
14 #  published by the Free Software Foundation.
15 #
16 #  svrt is distributed in the hope that it will be useful, but
17 #  WITHOUT ANY WARRANTY; without even the implied warranty of
18 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
19 #  General Public License for more details.
20 #
21 #  You should have received a copy of the GNU General Public License
22 #  along with selector.  If not, see <http://www.gnu.org/licenses/>.
23
24 import os
25 from torch.utils.ffi import create_extension
26
27 abs_path = os.path.dirname(os.path.abspath(__file__))
28
29 ffi = create_extension(
30     'svrt',
31     headers = [ 'svrt.h' ],
32     sources = [ 'svrt.c' ],
33     extra_objects = [ abs_path + '/libsvrt.so' ],
34     libraries = [ ],
35     library_dirs = [ ],
36     define_macros = [ ],
37     with_cuda = False
38 )
39
40 if __name__ == '__main__':
41     ffi.build()