summaryrefslogtreecommitdiff
path: root/Wrappers/Python/src
diff options
context:
space:
mode:
authorEdoardo Pasca <edo.paskino@gmail.com>2018-01-29 15:46:04 +0000
committerEdoardo Pasca <edo.paskino@gmail.com>2018-01-30 12:03:59 +0000
commit25c43ebd60c141bacf3ff9fd9c5e2f357bc5e4b6 (patch)
tree4ffc8f08d9dcdfbceb068ca4c2a1fc19527610cd /Wrappers/Python/src
parentfd7236f721017a6e8c082f01545e6b5aaf4c2cac (diff)
finds boost libraries during wrappers build (win)
finds and set the correct name for the boost libraries. Builds GPU wrapper if CUDA is found.
Diffstat (limited to 'Wrappers/Python/src')
-rw-r--r--Wrappers/Python/src/cpu_regularizers.pyx19
-rw-r--r--Wrappers/Python/src/fista_module.cpp4
2 files changed, 21 insertions, 2 deletions
diff --git a/Wrappers/Python/src/cpu_regularizers.pyx b/Wrappers/Python/src/cpu_regularizers.pyx
index e69de29..a8f8c8f 100644
--- a/Wrappers/Python/src/cpu_regularizers.pyx
+++ b/Wrappers/Python/src/cpu_regularizers.pyx
@@ -0,0 +1,19 @@
+# distutils: language=c++
+"""
+Copyright 2018 CCPi
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+ http://www.apache.org/licenses/LICENSE-2.0
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+
+Author: Edoardo Pasca
+"""
+
+import cython
+import numpy as np
+cimport numpy as np
diff --git a/Wrappers/Python/src/fista_module.cpp b/Wrappers/Python/src/fista_module.cpp
index 3876cad..cef3ecc 100644
--- a/Wrappers/Python/src/fista_module.cpp
+++ b/Wrappers/Python/src/fista_module.cpp
@@ -1028,13 +1028,13 @@ bp::list TGV_PD(np::ndarray input, double d_lambda, double d_alpha1, double d_al
return result;
}
-BOOST_PYTHON_MODULE(cpu_regularizers)
+BOOST_PYTHON_MODULE(cpu_regularizers_boost)
{
np::initialize();
//To specify that this module is a package
bp::object package = bp::scope();
- package.attr("__path__") = "cpu_regularizers";
+ package.attr("__path__") = "cpu_regularizers_boost";
np::dtype dt1 = np::dtype::get_builtin<uint8_t>();
np::dtype dt2 = np::dtype::get_builtin<uint16_t>();