bugfix

jdtuck · Dec 20, 2023 · eac2b0b · eac2b0b
1 parent 67096b2
commit eac2b0b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -110,7 +110,7 @@ def run(self):
         sources=["src/crbfgs.pyx", "src/rbfgs.cpp"],
         include_dirs=[numpy.get_include()],
         language="c++",
-        extra_compile_args=["-std=c++11"],
+        extra_compile_args=["-std=c++11", "-lblas", "-llapack"],
     ),
     Extension(
         name="cimage",

diff --git a/src/crbfgs.pyx b/src/crbfgs.pyx
@@ -8,8 +8,8 @@ include "cyarma.pyx"
 
 from libcpp cimport bool
 
-def rlbfs(np.ndarray[double, ndim=1, mode="c"] q1, np.ndarray[double, ndim=1, mode="c"] q2, 
-          np.ndarray[double, ndim=1, mode="c"] time, maxiter=30, lam=0.0, penalty=0):
+def rlbfgs(np.ndarray[double, ndim=1, mode="c"] q1, np.ndarray[double, ndim=1, mode="c"] q2, 
+           np.ndarray[double, ndim=1, mode="c"] time, maxiter=30, lam=0.0, penalty=0):
     q1 = np.ascontiguousarray(q1)
     q2 = np.ascontiguousarray(q2)
     time = np.ascontiguousarray(time)