Skip to content

Commit ef1ce6d

Browse files
committed
Update liboneapi_support.jl
1 parent 37cd05d commit ef1ce6d

File tree

1 file changed

+52
-0
lines changed

1 file changed

+52
-0
lines changed

lib/support/liboneapi_support.jl

Lines changed: 52 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -180,6 +180,14 @@ end
180180
ONEMKL_MATMAT_REQUEST_FINALIZE = 8
181181
end
182182

183+
@cenum onemklOmatconvertAlg::UInt32 begin
184+
ONEMKL_OMATCONVERT_DEFAULT_ALG = 0
185+
end
186+
187+
@cenum onemklOmataddAlg::UInt32 begin
188+
ONEMKL_OMATADD_DEFAULT_ALG = 0
189+
end
190+
183191
mutable struct matrix_handle end
184192

185193
const matrix_handle_t = Ptr{matrix_handle}
@@ -188,6 +196,14 @@ mutable struct matmat_descr end
188196

189197
const matmat_descr_t = Ptr{matmat_descr}
190198

199+
mutable struct omatconvert_descr end
200+
201+
const omatconvert_descr_t = Ptr{omatconvert_descr}
202+
203+
mutable struct omatadd_descr end
204+
205+
const omatadd_descr_t = Ptr{omatadd_descr}
206+
191207
function onemklHgemm_batch(device_queue, transa, transb, m, n, k, alpha, a, lda, b, ldb,
192208
beta, c, ldc, group_count, group_size)
193209
@ccall liboneapi_support.onemklHgemm_batch(device_queue::syclQueue_t,
@@ -6596,6 +6612,26 @@ function onemklXsparse_release_matmat_descr(p_desc)
65966612
@ccall liboneapi_support.onemklXsparse_release_matmat_descr(p_desc::Ptr{matmat_descr_t})::Cint
65976613
end
65986614

6615+
function onemklXsparse_init_omatconvert_descr(device_queue, p_descr)
6616+
@ccall liboneapi_support.onemklXsparse_init_omatconvert_descr(device_queue::syclQueue_t,
6617+
p_descr::Ptr{omatconvert_descr_t})::Cint
6618+
end
6619+
6620+
function onemklXsparse_release_omatconvert_descr(device_queue, descr)
6621+
@ccall liboneapi_support.onemklXsparse_release_omatconvert_descr(device_queue::syclQueue_t,
6622+
descr::omatconvert_descr_t)::Cint
6623+
end
6624+
6625+
function onemklXsparse_init_omatadd_descr(device_queue, p_omatadd_desc)
6626+
@ccall liboneapi_support.onemklXsparse_init_omatadd_descr(device_queue::syclQueue_t,
6627+
p_omatadd_desc::Ptr{omatadd_descr_t})::Cint
6628+
end
6629+
6630+
function onemklXsparse_release_omatadd_descr(device_queue, omatadd_desc)
6631+
@ccall liboneapi_support.onemklXsparse_release_omatadd_descr(device_queue::syclQueue_t,
6632+
omatadd_desc::omatadd_descr_t)::Cint
6633+
end
6634+
65996635
function onemklXsparse_omatcopy(device_queue, transpose_val, spMat_in, spMat_out)
66006636
@ccall liboneapi_support.onemklXsparse_omatcopy(device_queue::syclQueue_t,
66016637
transpose_val::onemklTranspose,
@@ -6658,6 +6694,22 @@ function onemklXsparse_optimize_trsv(device_queue, uplo_val, opA, diag_val, A)
66586694
A::matrix_handle_t)::Cint
66596695
end
66606696

6697+
function onemklXsparse_optimize_gemm(device_queue, opA, A)
6698+
@ccall liboneapi_support.onemklXsparse_optimize_gemm(device_queue::syclQueue_t,
6699+
opA::onemklTranspose,
6700+
A::matrix_handle_t)::Cint
6701+
end
6702+
6703+
function onemklXsparse_optimize_gemm_advanced(device_queue, layout_val, opA, opB, A,
6704+
columns)
6705+
@ccall liboneapi_support.onemklXsparse_optimize_gemm_advanced(device_queue::syclQueue_t,
6706+
layout_val::onemklLayout,
6707+
opA::onemklTranspose,
6708+
opB::onemklTranspose,
6709+
A::matrix_handle_t,
6710+
columns::Int64)::Cint
6711+
end
6712+
66616713
function onemklXsparse_optimize_trsm(device_queue, uplo_val, opA, diag_val, A)
66626714
@ccall liboneapi_support.onemklXsparse_optimize_trsm(device_queue::syclQueue_t,
66636715
uplo_val::onemklUplo,

0 commit comments

Comments
 (0)