File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
1725INCLUDEFLAGS =-Iinclude/
1826NVCCLDFLAGS = -lcublas
19- LDFLAGS = -llapacke -lcblas
27+ LDFLAGS = $( LAPACKE_LIBS )
2028
2129LIBNAME =liblanczos.so
2230PYTHON_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 )
2634NVCCFLAGS=-ccbin =$(CXX ) -Xcompiler "$(CXXFLAGS ) " -std=c++14 -O3 $(INCLUDEFLAGS ) $(DOUBLE_PRECISION ) $(CUDA_ENABLED )
2735
2836PYTHON_LIBRARY_NAME =python/$(PYTHON_MODULE_NAME )$(shell $(PYTHON3 ) -config --extension-suffix)
Original file line number Diff line number Diff 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"
You can’t perform that action at this time.
0 commit comments