|
95 | 95 | Operating System :: MacOS
|
96 | 96 | """
|
97 | 97 |
|
| 98 | +IS_CONDA_BUILD = True if os.environ.get("CONDA_BUILD", None) == "1" else False |
| 99 | + |
98 | 100 | IS_WIN = False
|
99 | 101 | IS_MAC = False
|
100 | 102 | IS_LIN = False
|
|
205 | 207 | """
|
206 | 208 | _mkl_include = None
|
207 | 209 | _mkl_libpath = None
|
208 |
| -# TODO make it as a function in utils |
209 |
| -_conda_root = os.environ.get("CONDA_PREFIX", None) |
210 |
| -if _mkl_include is None and _mkl_libpath is None and _conda_root is not None: |
| 210 | + |
| 211 | +# try to find math library in environment |
| 212 | +if IS_CONDA_BUILD: |
| 213 | + _conda_root = os.environ.get("PREFIX", None) |
| 214 | +else: |
| 215 | + _conda_root = os.environ.get("CONDA_PREFIX", None) |
| 216 | +if _conda_root is not None: |
211 | 217 | _mkl_include_find = os.path.join(_conda_root, "include")
|
212 | 218 | _mkl_libpath_find = os.path.join(_conda_root, "lib")
|
213 |
| - _required_header = os.path.join(_mkl_include_find, "mkl_blas_sycl.hpp") |
| 219 | + _required_header = os.path.join(_mkl_include_find, "oneapi", "mkl.hpp") |
214 | 220 | _required_library = os.path.join(_mkl_libpath_find, "libmkl_sycl.so")
|
215 | 221 |
|
216 | 222 | if (os.path.exists(_required_header) and os.path.exists(_required_library)):
|
|
219 | 225 | _mkl_include = [_mkl_include_find]
|
220 | 226 | _mkl_libpath = [_mkl_libpath_find]
|
221 | 227 |
|
222 |
| - |
223 | 228 | _mkl_root = os.environ.get("MKLROOT", None)
|
224 |
| -if _mkl_include is None and _mkl_libpath is None and _mkl_root is not None: |
| 229 | +if ((_mkl_include is None or _mkl_libpath is None) and (_mkl_root is not None)): # if MKLROOT was specified then use it |
| 230 | + # TODO change paths and file names for new version |
| 231 | + # paths and file names are aligned to beta09 at this moment |
225 | 232 | _mkl_include_find = os.path.join(_mkl_root, "include")
|
226 | 233 | _mkl_libpath_find = os.path.join(_mkl_root, "lib", "intel64")
|
227 | 234 | _required_header = os.path.join(_mkl_include_find, "mkl_blas_sycl.hpp")
|
|
233 | 240 | _mkl_include = [_mkl_include_find]
|
234 | 241 | _mkl_libpath = [_mkl_libpath_find]
|
235 | 242 |
|
236 |
| -if _mkl_include is None and _mkl_libpath is None: |
237 |
| - raise EnvironmentError("Intel DPNP: Please install Intel OneAPI environment. MKLROOT is empty") |
| 243 | +if _mkl_include is None or _mkl_libpath is None: |
| 244 | + raise EnvironmentError("Intel DPNP: Unable to find math library") |
238 | 245 |
|
239 | 246 | _project_cmplr_macro += [("MKL_ILP64", "1")] # using 64bit integers in MKL interface (long)
|
240 | 247 | _mkl_libs = ["mkl_rt", "mkl_sycl", "mkl_intel_ilp64", "mkl_sequential",
|
|
248 | 255 | """
|
249 | 256 | Get the compiler environemnt
|
250 | 257 | """
|
251 |
| -_cmplr_root = os.environ.get('ONEAPI_ROOT', None) |
| 258 | +_cmplr_root = os.environ.get('DPCPPROOT', None) |
252 | 259 | if _cmplr_root is None:
|
253 |
| - raise EnvironmentError("Please install Intel OneAPI environment. ONEAPI_ROOT is empty") |
| 260 | + raise EnvironmentError("Please install Intel OneAPI environment. DPCPPROOT is empty") |
254 | 261 |
|
255 | 262 | if IS_LIN:
|
256 |
| - _cmplr_libpath = [os.path.join(_cmplr_root, 'compiler', 'latest', 'linux', 'lib')] |
257 |
| - _omp_libpath = [os.path.join(_cmplr_root, 'compiler', 'latest', 'linux', 'compiler', 'lib', 'intel64')] |
| 263 | + _cmplr_libpath = [os.path.join(_cmplr_root, 'linux', 'lib')] |
| 264 | + _omp_libpath = [os.path.join(_cmplr_root, 'linux', 'compiler', 'lib', 'intel64')] |
258 | 265 | _cmplr_rpath = _cmplr_libpath
|
259 | 266 | _omp_rpath = _omp_libpath
|
260 | 267 | elif IS_WIN:
|
261 |
| - _cmplr_libpath = [os.path.join(_cmplr_root, 'compiler', 'latest', 'windows', 'lib')] |
262 |
| - _omp_libpath = [os.path.join(_cmplr_root, 'compiler', 'latest', 'windows', 'compiler', 'lib', 'intel64_win')] |
| 268 | + _cmplr_libpath = [os.path.join(_cmplr_root, 'windows', 'lib')] |
| 269 | + _omp_libpath = [os.path.join(_cmplr_root, 'windows', 'compiler', 'lib', 'intel64_win')] |
263 | 270 |
|
264 | 271 |
|
265 | 272 | """
|
|
0 commit comments