diff options
Diffstat (limited to 'Wrappers/Python/setup.py')
| -rw-r--r-- | Wrappers/Python/setup.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/Wrappers/Python/setup.py b/Wrappers/Python/setup.py index 951146a..00c93fc 100644 --- a/Wrappers/Python/setup.py +++ b/Wrappers/Python/setup.py @@ -60,7 +60,7 @@ setup( cmdclass = {'build_ext': build_ext}, ext_modules = [Extension("ccpi.filters.gpu_regularizers", sources=[ - os.path.join("." , "src", "fista_module_gpu.pyx" ), + os.path.join("." , "src", "gpu_regularizers.pyx" ), ], include_dirs=extra_include_dirs, library_dirs=extra_library_dirs, @@ -79,6 +79,7 @@ setup( cmdclass = {'build_ext': build_ext}, ext_modules = [Extension("ccpi.filters.cpu_regularizers", sources=[os.path.join("." , "src", "fista_module.cpp" ), + os.path.join("." , "src", "cpu_regularizers.pyx" ) # os.path.join("@CMAKE_SOURCE_DIR@" , "main_func" , "regularizers_CPU", "FGP_TV_core.c"), # os.path.join("@CMAKE_SOURCE_DIR@" , "main_func" , "regularizers_CPU", "SplitBregman_TV_core.c"), # os.path.join("@CMAKE_SOURCE_DIR@" , "main_func" , "regularizers_CPU", "LLT_model_core.c"), |