| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| |
| #include "c_lapack_api.h" |
| |
| #if (MXNET_USE_LAPACK && (MSHADOW_USE_MKL || MXNET_USE_LAPACKE_INTERFACE)) |
| #elif MXNET_USE_LAPACK |
| #else |
| // use pragma message instead of warning |
| #pragma message( \ |
| "Warning: lapack usage not enabled, linalg-operators will not be available." \ |
| " Ensure that lapack library is installed and build with USE_LAPACK=1 to get lapack" \ |
| " functionalities.") |
| |
| // Define compilable stubs. |
| #define MXNET_LAPACK_CWRAPPER1(func, dtype) \ |
| int MXNET_LAPACK_##func(int matrix_layout, char uplo, int n, dtype* a, int lda) { \ |
| LOG(FATAL) << "MXNet build without lapack. Function " << #func << " is not available."; \ |
| return 1; \ |
| } |
| |
| #define MXNET_LAPACK_CWRAPPER2(func, dtype) \ |
| int MXNET_LAPACK_##func( \ |
| int matrix_layout, int m, int n, dtype* a, int lda, dtype* tau, dtype* work, int lwork) { \ |
| LOG(FATAL) << "MXNet build without lapack. Function " << #func << " is not available."; \ |
| return 1; \ |
| } |
| |
| #define MXNET_LAPACK_CWRAPPER3(func, dtype) \ |
| int MXNET_LAPACK_##func(int matrix_layout, \ |
| char uplo, \ |
| int n, \ |
| dtype* a, \ |
| int lda, \ |
| dtype* w, \ |
| dtype* work, \ |
| int lwork, \ |
| int* iwork, \ |
| int liwork) { \ |
| LOG(FATAL) << "MXNet build without lapack. Function " << #func << " is not available."; \ |
| return 1; \ |
| } |
| |
| #define MXNET_LAPACK_CWRAPPER4(func, dtype) \ |
| int MXNET_LAPACK_##func(int matrix_layout, int m, int n, dtype* a, int lda, int* ipiv) { \ |
| LOG(FATAL) << "MXNet build without lapack. Function " << #func << " is not available."; \ |
| return 1; \ |
| } |
| |
| #define MXNET_LAPACK_CWRAPPER5(func, dtype) \ |
| int MXNET_LAPACK_##func( \ |
| int matrix_layout, int n, dtype* a, int lda, int* ipiv, dtype* work, int lwork) { \ |
| LOG(FATAL) << "MXNet build without lapack. Function " << #func << " is not available."; \ |
| return 1; \ |
| } |
| |
| #define MXNET_LAPACK_CWRAPPER6(func, dtype) \ |
| int MXNET_LAPACK_##func(int matrix_layout, \ |
| int m, \ |
| int n, \ |
| dtype* ut, \ |
| int ldut, \ |
| dtype* s, \ |
| dtype* v, \ |
| int ldv, \ |
| dtype* work, \ |
| int lwork) { \ |
| LOG(FATAL) << "MXNet build without lapack. Function " << #func << " is not available."; \ |
| return 1; \ |
| } |
| |
| #define MXNET_LAPACK_CWRAPPER7(func, dtype) \ |
| int MXNET_LAPACK_##func( \ |
| int matrix_order, int n, int nrhs, dtype* a, int lda, int* ipiv, dtype* b, int ldb) { \ |
| LOG(FATAL) << "MXNet build without lapack. Function " << #func << " is not available."; \ |
| return 1; \ |
| } |
| |
| #define MXNET_LAPACK_CWRAPPER8(func, dtype) \ |
| int MXNET_LAPACK_##func(int matrix_layout, \ |
| char jobvl, \ |
| char jobvr, \ |
| int n, \ |
| dtype* a, \ |
| int lda, \ |
| dtype* wr, \ |
| dtype* wi, \ |
| dtype* vl, \ |
| int ldvl, \ |
| dtype* vr, \ |
| int ldvr, \ |
| dtype* work, \ |
| int lwork) { \ |
| LOG(FATAL) << "MXNet build without lapack. Function " << #func << " is not available."; \ |
| return 1; \ |
| } |
| |
| #define MXNET_LAPACK_CWRAPPER9(func, dtype) \ |
| int MXNET_LAPACK_##func(int matrix_layout, \ |
| int m, \ |
| int n, \ |
| dtype* a, \ |
| int lda, \ |
| dtype* s, \ |
| dtype* u, \ |
| int ldu, \ |
| dtype* vt, \ |
| int ldvt, \ |
| dtype* work, \ |
| int lwork, \ |
| int* iwork) { \ |
| LOG(FATAL) << "MXNet build without lapack. Function " << #func << " is not available."; \ |
| return 1; \ |
| } |
| |
| #define MXNET_LAPACK_CWRAPPER10(func, dtype) \ |
| int MXNET_LAPACK_##func( \ |
| int matrix_layout, int m, int n, dtype* a, int lda, dtype* tau, dtype* work, int lwork) { \ |
| LOG(FATAL) << "MXNet build without lapack. Function " << #func << " is not available."; \ |
| return 1; \ |
| } |
| |
| #define MXNET_LAPACK_CWRAPPER11(func, dtype) \ |
| int MXNET_LAPACK_##func(int matrix_layout, \ |
| int m, \ |
| int n, \ |
| int nrhs, \ |
| dtype* a, \ |
| int lda, \ |
| dtype* b, \ |
| int ldb, \ |
| dtype* s, \ |
| dtype rcond, \ |
| int* rank, \ |
| dtype* work, \ |
| int lwork, \ |
| int* iwork) { \ |
| LOG(FATAL) << "MXNet build without lapack. Function " << #func << " is not available."; \ |
| return 1; \ |
| } |
| |
| #define MXNET_LAPACK_CWRAPPER12(func, dtype) \ |
| int MXNET_LAPACK_##func(int matrix_layout, \ |
| int m, \ |
| int n, \ |
| int k, \ |
| dtype* a, \ |
| int lda, \ |
| dtype* tau, \ |
| dtype* work, \ |
| int lwork) { \ |
| LOG(FATAL) << "MXNet build without lapack. Function " << #func << " is not available."; \ |
| return 1; \ |
| } |
| |
| #define MXNET_LAPACK_UNAVAILABLE(func) \ |
| int mxnet_lapack_##func(...) { \ |
| LOG(FATAL) << "MXNet build without lapack. Function " << #func << " is not available."; \ |
| return 1; \ |
| } |
| MXNET_LAPACK_CWRAPPER1(spotrf, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER1(dpotrf, double) // NOLINT |
| MXNET_LAPACK_CWRAPPER1(spotri, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER1(dpotri, double) // NOLINT |
| |
| MXNET_LAPACK_UNAVAILABLE(sposv) // NOLINT |
| MXNET_LAPACK_UNAVAILABLE(dposv) // NOLINT |
| |
| MXNET_LAPACK_CWRAPPER2(sgelqf, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER2(dgelqf, double) // NOLINT |
| MXNET_LAPACK_CWRAPPER2(sorglq, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER2(dorglq, double) // NOLINT |
| |
| MXNET_LAPACK_CWRAPPER3(ssyevd, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER3(dsyevd, double) // NOLINT |
| |
| MXNET_LAPACK_CWRAPPER4(sgetrf, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER4(dgetrf, double) // NOLINT |
| |
| MXNET_LAPACK_CWRAPPER5(sgetri, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER5(dgetri, double) // NOLINT |
| |
| MXNET_LAPACK_CWRAPPER6(sgesvd, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER6(dgesvd, double) // NOLINT |
| |
| MXNET_LAPACK_CWRAPPER7(sgesv, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER7(dgesv, double) // NOLINT |
| |
| MXNET_LAPACK_CWRAPPER8(sgeev, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER8(dgeev, double) // NOLINT |
| |
| MXNET_LAPACK_CWRAPPER9(sgesdd, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER9(dgesdd, double) // NOLINT |
| |
| MXNET_LAPACK_CWRAPPER10(sgeqrf, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER10(dgeqrf, double) // NOLINT |
| |
| MXNET_LAPACK_CWRAPPER11(sgelsd, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER11(dgelsd, double) // NOLINT |
| |
| MXNET_LAPACK_CWRAPPER12(sorgqr, float) // NOLINT |
| MXNET_LAPACK_CWRAPPER12(dorgqr, double) // NOLINT |
| |
| #endif // MSHADOW_USE_MKL == 0 |
