diff --git a/setup.py b/setup.py index e774a6c648c32f26226d9eb01b1ba96a8e2758a1..b85a04d4e952bc74559f18f81b8262741be6f8c9 100644 --- a/setup.py +++ b/setup.py @@ -1,9 +1,10 @@ import sys from glob import glob -from os.path import dirname, join +from os.path import join from pkg_resources import VersionConflict, require from setuptools import setup + from torch.utils.cpp_extension import BuildExtension, CUDAExtension try: @@ -15,7 +16,6 @@ except VersionConflict: if __name__ == "__main__": - object_cache = dirname(__file__) module_name = 'pyronn_torch_cpp' cuda_sources = glob(join('generated_files', '*.cu'))