File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change 27
27
28
28
#include < dpnp_iface.hpp>
29
29
#include " dpnp_fptr.hpp"
30
+ #include " dpnpc_memory_adapter.hpp"
30
31
#include " queue_sycl.hpp"
31
32
32
33
template <typename _DataType, typename _idx_DataType>
@@ -35,7 +36,8 @@ class dpnp_argmax_c_kernel;
35
36
template <typename _DataType, typename _idx_DataType>
36
37
void dpnp_argmax_c (void * array1_in, void * result1, size_t size)
37
38
{
38
- _DataType* array_1 = reinterpret_cast <_DataType*>(array1_in);
39
+ DPNPC_ptr_adapter<_DataType> input1_ptr (array1_in, size);
40
+ _DataType* array_1 = input1_ptr.get_ptr ();
39
41
_idx_DataType* result = reinterpret_cast <_idx_DataType*>(result1);
40
42
41
43
auto policy =
@@ -56,7 +58,8 @@ class dpnp_argmin_c_kernel;
56
58
template <typename _DataType, typename _idx_DataType>
57
59
void dpnp_argmin_c (void * array1_in, void * result1, size_t size)
58
60
{
59
- _DataType* array_1 = reinterpret_cast <_DataType*>(array1_in);
61
+ DPNPC_ptr_adapter<_DataType> input1_ptr (array1_in, size);
62
+ _DataType* array_1 = input1_ptr.get_ptr ();
60
63
_idx_DataType* result = reinterpret_cast <_idx_DataType*>(result1);
61
64
62
65
auto policy =
You can’t perform that action at this time.
0 commit comments