Skip to content

Commit 49edafc

Browse files
author
u75572
committed
looks like mkl requires memory on host
1 parent 01c6b58 commit 49edafc

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

dpnp/backend/kernels/dpnp_krnl_common.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -316,11 +316,12 @@ void dpnp_matmul_c(void* array1_in, void* array2_in, void* result1, size_t size_
316316
}
317317

318318
cl::sycl::event event;
319-
DPNPC_ptr_adapter<_DataType> input1_ptr(array1_in, size_m * size_k);
320-
DPNPC_ptr_adapter<_DataType> input2_ptr(array2_in, size_k * size_n);
319+
DPNPC_ptr_adapter<_DataType> input1_ptr(array1_in, size_m * size_k, true);
320+
DPNPC_ptr_adapter<_DataType> input2_ptr(array2_in, size_k * size_n, true);
321+
DPNPC_ptr_adapter<_DataType> result_ptr(result1, size_m * size_n, true, true);
321322
_DataType* array_1 = input1_ptr.get_ptr();
322323
_DataType* array_2 = input2_ptr.get_ptr();
323-
_DataType* result = reinterpret_cast<_DataType*>(result1);
324+
_DataType* result = result_ptr.get_ptr();
324325

325326
if constexpr (std::is_same<_DataType, double>::value || std::is_same<_DataType, float>::value)
326327
{

0 commit comments

Comments
 (0)