Skip to content

Commit bb34665

Browse files
committed
Change makefile to allow MKL
1 parent 39d6b12 commit bb34665

2 files changed

Lines changed: 11 additions & 2 deletions

File tree

Makefile

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,16 +13,24 @@ PYBIND_INCLUDE=extern/pybind11/include
1313
#Uncomment to compile in double precision mode
1414
#DOUBLE_PRECISION=-DDOUBLE_PRECISION
1515

16+
#Lapack/Cblas related flags
17+
LAPACKE_FLAGS=
18+
LAPACKE_LIBS=-llapacke -lcblas
19+
20+
#Alternativeky you can use MKL
21+
#LAPACKE_FLAGS= -I$(MKLROOT)/include -L$(MKLROOT)/lib/intel64 -DUSE_MKL
22+
#LAPACKE_LIBS= -lmkl_rt -lpthread -ldl
23+
1624

1725
INCLUDEFLAGS=-Iinclude/
1826
NVCCLDFLAGS= -lcublas
19-
LDFLAGS= -llapacke -lcblas
27+
LDFLAGS= $(LAPACKE_LIBS)
2028

2129
LIBNAME=liblanczos.so
2230
PYTHON_MODULE_NAME=Lanczos
2331

2432

25-
CXXFLAGS=-fPIC -w -O3 -g -std=c++14 $(INCLUDEFLAGS) $(DOUBLE_PRECISION)
33+
CXXFLAGS=-fPIC -w -O3 -g -std=c++14 $(INCLUDEFLAGS) $(DOUBLE_PRECISION) $(LAPACKE_FLAGS)
2634
NVCCFLAGS=-ccbin=$(CXX) -Xcompiler "$(CXXFLAGS)" -std=c++14 -O3 $(INCLUDEFLAGS) $(DOUBLE_PRECISION) $(CUDA_ENABLED)
2735

2836
PYTHON_LIBRARY_NAME=python/$(PYTHON_MODULE_NAME)$(shell $(PYTHON3)-config --extension-suffix)

include/LanczosAlgorithm.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ References:
88
#include"LanczosAlgorithm.h"
99
#include<string.h>
1010
#include"utils/lapack_and_blas_defines.h"
11+
#include<cmath>
1112
#include<stdexcept>
1213
#ifdef CUDA_ENABLED
1314
#include"utils/debugTools.h"

0 commit comments

Comments
 (0)