Skip to content

Commit e598f9d

Browse files
committed
Fix setup.py
1 parent dff68cf commit e598f9d

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

setup.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@
4848
enable_eigen = eigen_bundled_exists
4949

5050
if build_core:
51+
import skbuild
5152
from skbuild import setup
5253
else:
5354
from setuptools import setup
@@ -88,10 +89,10 @@ def ext_common_args(*args, libraries=[], **kwargs):
8889
*args, **kwargs,
8990
language="c++",
9091
libraries=libs,
91-
library_dirs=["_skbuild/cmake-install/lib"],
92+
library_dirs=[os.path.join(skbuild.constants.CMAKE_INSTALL_DIR, "lib")],
9293
include_dirs=[
9394
np.get_include(),
94-
"_skbuild/cmake-install/include",
95+
os.path.join(skbuild.constants.CMAKE_INSTALL_DIR, "include"),
9596
os.path.join(dirname, "primitiv"),
9697
],
9798
extra_compile_args=["-std=c++11"],
@@ -166,7 +167,7 @@ def ext_common_args(*args, libraries=[], **kwargs):
166167
ext_common_args(
167168
"primitiv.devices._opencl_device",
168169
libraries=[
169-
"clBLAS",
170+
"clblast",
170171
"OpenCL",
171172
],
172173
sources=["primitiv/devices/_opencl_device.pyx"],

0 commit comments

Comments
 (0)