X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;ds=inline;f=build.py;h=264d1a9cff4e32cb51fc97ffda88cdd189c6d9df;hb=cdcaa8361d43a497d50c2f3703f9f5b4be9c2298;hp=ef0b46d24b3fd05bc3ca74bbb334e6ad7027115e;hpb=f542d0542b1e51ca7dd12bc6b96f6a299371ae8d;p=pysvrt.git
diff --git a/build.py b/build.py
index ef0b46d..264d1a9 100755
--- a/build.py
+++ b/build.py
@@ -19,23 +19,19 @@
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
-# along with selector. If not, see .
+# along with svrt. If not, see .
-import os
+from os import path
from torch.utils.ffi import create_extension
-abs_path = os.path.dirname(os.path.abspath(__file__))
+abs_path = path.dirname(path.abspath(__file__))
ffi = create_extension(
- '_ext.svrt',
+ 'svrt',
headers = [ 'svrt.h' ],
sources = [ 'svrt.c' ],
extra_objects = [ abs_path + '/libsvrt.so' ],
- libraries = [ ],
- library_dirs = [ ],
- define_macros = [ ],
with_cuda = False
)
-if __name__ == '__main__':
- ffi.build()
+ffi.build()