From: Francois Fleuret Date: Wed, 14 Jun 2017 21:02:17 +0000 (+0200) Subject: Renamed the generated python module. X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pysvrt.git;a=commitdiff_plain;h=99878e6cbf07b86b0e180c3eb5bee055c7444b73 Renamed the generated python module. --- diff --git a/build.py b/build.py index ef0b46d..da18d39 100755 --- a/build.py +++ b/build.py @@ -27,7 +27,7 @@ from torch.utils.ffi import create_extension abs_path = os.path.dirname(os.path.abspath(__file__)) ffi = create_extension( - '_ext.svrt', + 'svrt', headers = [ 'svrt.h' ], sources = [ 'svrt.c' ], extra_objects = [ abs_path + '/libsvrt.so' ], diff --git a/test-svrt.py b/test-svrt.py index cd98f21..5c16069 100755 --- a/test-svrt.py +++ b/test-svrt.py @@ -34,7 +34,7 @@ from torch.nn import functional as fn from torchvision import datasets, transforms, utils -from _ext import svrt +import svrt labels = torch.LongTensor(12).zero_() labels.narrow(0, 0, labels.size(0)//2).fill_(1)