From 99878e6cbf07b86b0e180c3eb5bee055c7444b73 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Wed, 14 Jun 2017 23:02:17 +0200 Subject: [PATCH] Renamed the generated python module. --- build.py | 2 +- test-svrt.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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) -- 2.39.5