Skip to content

Commit 271a78a

Browse files
authored
fix eig (#919)
1 parent e88413b commit 271a78a

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

dpnp/backend/kernels/dpnp_krnl_common.cpp

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -169,10 +169,12 @@ void dpnp_eig_c(const void* array_in, void* result1, void* result2, size_t size)
169169
}
170170

171171
cl::sycl::event event;
172-
DPNPC_ptr_adapter<_DataType> input1_ptr(array_in, size);
172+
DPNPC_ptr_adapter<_DataType> input1_ptr(array_in, size * size, true);
173+
DPNPC_ptr_adapter<_ResultType> result1_ptr(result1, size, true, true);
174+
DPNPC_ptr_adapter<_ResultType> result2_ptr(result2, size * size, true, true);
173175
const _DataType* array = input1_ptr.get_ptr();
174-
_ResultType* result_val = reinterpret_cast<_ResultType*>(result1);
175-
_ResultType* result_vec = reinterpret_cast<_ResultType*>(result2);
176+
_ResultType* result_val = result1_ptr.get_ptr();
177+
_ResultType* result_vec = result2_ptr.get_ptr();
176178

177179
double* result_val_kern = reinterpret_cast<double*>(dpnp_memory_alloc_c(size * sizeof(double)));
178180
double* result_vec_kern = reinterpret_cast<double*>(dpnp_memory_alloc_c(size * size * sizeof(double)));
@@ -231,9 +233,10 @@ void dpnp_eigvals_c(const void* array_in, void* result1, size_t size)
231233
}
232234

233235
cl::sycl::event event;
234-
DPNPC_ptr_adapter<_DataType> input1_ptr(array_in, size);
236+
DPNPC_ptr_adapter<_DataType> input1_ptr(array_in, size * size, true);
237+
DPNPC_ptr_adapter<_ResultType> result1_ptr(result1, size, true, true);
235238
const _DataType* array = input1_ptr.get_ptr();
236-
_ResultType* result_val = reinterpret_cast<_ResultType*>(result1);
239+
_ResultType* result_val = result1_ptr.get_ptr();
237240

238241
double* result_val_kern = reinterpret_cast<double*>(dpnp_memory_alloc_c(size * sizeof(double)));
239242
double* result_vec_kern = reinterpret_cast<double*>(dpnp_memory_alloc_c(size * size * sizeof(double)));

0 commit comments

Comments
 (0)