Skip to content

Commit 2df5d3f

Browse files
amontoisonamontoison
committed
Fix the warning related to getri_batch
1 parent 0bca405 commit 2df5d3f

File tree

3 files changed

+77
-0
lines changed

3 files changed

+77
-0
lines changed

‎deps/generate_interfaces.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ function generate_headers(library::String, filename::String, output::String)
4444
occursin("gemm_bias", header) && continue # BLAS routine
4545
occursin("heevx", header) && continue # LAPACK routine
4646
occursin("hegvx", header) && continue # LAPACK routine
47+
occursin("getri_batch", header) && occursin("ldainv", header) && continue # LAPACK routine
4748
occursin("(matrix_handle_t handle", header) && continue # SPARSE routine
4849
occursin("update_diagonal_values", header) && continue # SPARSE routine
4950
occursin("get_matmat_data", header) && continue # SPARSE routine

‎deps/src/onemkl.cpp

Lines changed: 44 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3336,6 +3336,30 @@ extern "C" int onemklZungqr_batch(syclQueue_t device_queue, int64_t m, int64_t n
33363336
return 0;
33373337
}
33383338

3339+
extern "C" int onemklSgetri_batch(syclQueue_t device_queue, int64_t n, float *a, int64_t lda, int64_t stride_a, int64_t *ipiv, int64_t stride_ipiv, int64_t batch_size, float *scratchpad, int64_t scratchpad_size) {
3340+
auto status = oneapi::mkl::lapack::getri_batch(device_queue->val, n, a, lda, stride_a, ipiv, stride_ipiv, batch_size, scratchpad, scratchpad_size, {});
3341+
__FORCE_MKL_FLUSH__(status);
3342+
return 0;
3343+
}
3344+
3345+
extern "C" int onemklDgetri_batch(syclQueue_t device_queue, int64_t n, double *a, int64_t lda, int64_t stride_a, int64_t *ipiv, int64_t stride_ipiv, int64_t batch_size, double *scratchpad, int64_t scratchpad_size) {
3346+
auto status = oneapi::mkl::lapack::getri_batch(device_queue->val, n, a, lda, stride_a, ipiv, stride_ipiv, batch_size, scratchpad, scratchpad_size, {});
3347+
__FORCE_MKL_FLUSH__(status);
3348+
return 0;
3349+
}
3350+
3351+
extern "C" int onemklCgetri_batch(syclQueue_t device_queue, int64_t n, float _Complex *a, int64_t lda, int64_t stride_a, int64_t *ipiv, int64_t stride_ipiv, int64_t batch_size, float _Complex *scratchpad, int64_t scratchpad_size) {
3352+
auto status = oneapi::mkl::lapack::getri_batch(device_queue->val, n, reinterpret_cast<std::complex<float>*>(a), lda, stride_a, ipiv, stride_ipiv, batch_size, reinterpret_cast<std::complex<float>*>(scratchpad), scratchpad_size, {});
3353+
__FORCE_MKL_FLUSH__(status);
3354+
return 0;
3355+
}
3356+
3357+
extern "C" int onemklZgetri_batch(syclQueue_t device_queue, int64_t n, double _Complex *a, int64_t lda, int64_t stride_a, int64_t *ipiv, int64_t stride_ipiv, int64_t batch_size, double _Complex *scratchpad, int64_t scratchpad_size) {
3358+
auto status = oneapi::mkl::lapack::getri_batch(device_queue->val, n, reinterpret_cast<std::complex<double>*>(a), lda, stride_a, ipiv, stride_ipiv, batch_size, reinterpret_cast<std::complex<double>*>(scratchpad), scratchpad_size, {});
3359+
__FORCE_MKL_FLUSH__(status);
3360+
return 0;
3361+
}
3362+
33393363
extern "C" int onemklSgels_batch(syclQueue_t device_queue, onemklTranspose trans, int64_t m, int64_t n, int64_t nrhs, float *a, int64_t lda, int64_t stridea, float *b, int64_t ldb, int64_t strideb, int64_t batchsize, float *scratchpad, int64_t scratchpad_size) {
33403364
auto status = oneapi::mkl::lapack::gels_batch(device_queue->val, convert(trans), m, n, nrhs, a, lda, stridea, b, ldb, strideb, batchsize, scratchpad, scratchpad_size, {});
33413365
__FORCE_MKL_FLUSH__(status);
@@ -3440,6 +3464,26 @@ extern "C" int64_t onemklZungqr_batch_scratchpad_size(syclQueue_t device_queue,
34403464
return scratchpad_size;
34413465
}
34423466

3467+
extern "C" int64_t onemklSgetri_batch_scratchpad_size(syclQueue_t device_queue, int64_t n, int64_t lda, int64_t stride_a, int64_t stride_ipiv, int64_t batch_size) {
3468+
int64_t scratchpad_size = oneapi::mkl::lapack::getri_batch_scratchpad_size<float>(device_queue->val, n, lda, stride_a, stride_ipiv, batch_size);
3469+
return scratchpad_size;
3470+
}
3471+
3472+
extern "C" int64_t onemklDgetri_batch_scratchpad_size(syclQueue_t device_queue, int64_t n, int64_t lda, int64_t stride_a, int64_t stride_ipiv, int64_t batch_size) {
3473+
int64_t scratchpad_size = oneapi::mkl::lapack::getri_batch_scratchpad_size<double>(device_queue->val, n, lda, stride_a, stride_ipiv, batch_size);
3474+
return scratchpad_size;
3475+
}
3476+
3477+
extern "C" int64_t onemklCgetri_batch_scratchpad_size(syclQueue_t device_queue, int64_t n, int64_t lda, int64_t stride_a, int64_t stride_ipiv, int64_t batch_size) {
3478+
int64_t scratchpad_size = oneapi::mkl::lapack::getri_batch_scratchpad_size<std::complex<float>>(device_queue->val, n, lda, stride_a, stride_ipiv, batch_size);
3479+
return scratchpad_size;
3480+
}
3481+
3482+
extern "C" int64_t onemklZgetri_batch_scratchpad_size(syclQueue_t device_queue, int64_t n, int64_t lda, int64_t stride_a, int64_t stride_ipiv, int64_t batch_size) {
3483+
int64_t scratchpad_size = oneapi::mkl::lapack::getri_batch_scratchpad_size<std::complex<double>>(device_queue->val, n, lda, stride_a, stride_ipiv, batch_size);
3484+
return scratchpad_size;
3485+
}
3486+
34433487
extern "C" int64_t onemklSgels_batch_scratchpad_size(syclQueue_t device_queue, onemklTranspose trans, int64_t m, int64_t n, int64_t nrhs, int64_t lda, int64_t stride_a, int64_t ldb, int64_t stride_b, int64_t batch_size) {
34443488
int64_t scratchpad_size = oneapi::mkl::lapack::gels_batch_scratchpad_size<float>(device_queue->val, convert(trans), m, n, nrhs, lda, stride_a, ldb, stride_b, batch_size);
34453489
return scratchpad_size;

‎deps/src/onemkl.h

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1906,6 +1906,22 @@ int onemklZungqr_batch(syclQueue_t device_queue, int64_t m, int64_t n, int64_t k
19061906
int64_t lda, int64_t stride_a, double _Complex *tau, int64_t stride_tau,
19071907
int64_t batch_size, double _Complex *scratchpad, int64_t scratchpad_size);
19081908

1909+
int onemklSgetri_batch(syclQueue_t device_queue, int64_t n, float *a, int64_t lda, int64_t stride_a,
1910+
int64_t *ipiv, int64_t stride_ipiv, int64_t batch_size, float *scratchpad,
1911+
int64_t scratchpad_size);
1912+
1913+
int onemklDgetri_batch(syclQueue_t device_queue, int64_t n, double *a, int64_t lda, int64_t
1914+
stride_a, int64_t *ipiv, int64_t stride_ipiv, int64_t batch_size, double
1915+
*scratchpad, int64_t scratchpad_size);
1916+
1917+
int onemklCgetri_batch(syclQueue_t device_queue, int64_t n, float _Complex *a, int64_t lda, int64_t
1918+
stride_a, int64_t *ipiv, int64_t stride_ipiv, int64_t batch_size, float
1919+
_Complex *scratchpad, int64_t scratchpad_size);
1920+
1921+
int onemklZgetri_batch(syclQueue_t device_queue, int64_t n, double _Complex *a, int64_t lda, int64_t
1922+
stride_a, int64_t *ipiv, int64_t stride_ipiv, int64_t batch_size, double
1923+
_Complex *scratchpad, int64_t scratchpad_size);
1924+
19091925
int onemklSgels_batch(syclQueue_t device_queue, onemklTranspose trans, int64_t m, int64_t n,
19101926
int64_t nrhs, float *a, int64_t lda, int64_t stridea, float *b, int64_t ldb,
19111927
int64_t strideb, int64_t batchsize, float *scratchpad, int64_t
@@ -1986,6 +2002,22 @@ int64_t onemklZungqr_batch_scratchpad_size(syclQueue_t device_queue, int64_t m,
19862002
int64_t k, int64_t lda, int64_t stride_a, int64_t
19872003
stride_tau, int64_t batch_size);
19882004

2005+
int64_t onemklSgetri_batch_scratchpad_size(syclQueue_t device_queue, int64_t n, int64_t lda,
2006+
int64_t stride_a, int64_t stride_ipiv, int64_t
2007+
batch_size);
2008+
2009+
int64_t onemklDgetri_batch_scratchpad_size(syclQueue_t device_queue, int64_t n, int64_t lda,
2010+
int64_t stride_a, int64_t stride_ipiv, int64_t
2011+
batch_size);
2012+
2013+
int64_t onemklCgetri_batch_scratchpad_size(syclQueue_t device_queue, int64_t n, int64_t lda,
2014+
int64_t stride_a, int64_t stride_ipiv, int64_t
2015+
batch_size);
2016+
2017+
int64_t onemklZgetri_batch_scratchpad_size(syclQueue_t device_queue, int64_t n, int64_t lda,
2018+
int64_t stride_a, int64_t stride_ipiv, int64_t
2019+
batch_size);
2020+
19892021
int64_t onemklSgels_batch_scratchpad_size(syclQueue_t device_queue, onemklTranspose trans,
19902022
int64_t m, int64_t n, int64_t nrhs, int64_t lda, int64_t
19912023
stride_a, int64_t ldb, int64_t stride_b, int64_t

0 commit comments

Comments
 (0)