Skip to content

Commit 3bc0372

Browse files
committed
enqueue_copy for SVM: Accept no-op {src,dest}_offset, byte_count parameters
1 parent 338169c commit 3bc0372

2 files changed

Lines changed: 4 additions & 0 deletions

File tree

pyopencl/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1780,6 +1780,9 @@ def enqueue_copy(queue, dest, src, **kwargs):
17801780
src = SVM(src)
17811781

17821782
is_blocking = kwargs.pop("is_blocking", True)
1783+
assert kwargs.pop("src_offset", 0) == 0
1784+
assert kwargs.pop("dest_offset", 0) == 0
1785+
assert "byte_count" not in kwargs or kwargs.pop("byte_count") == src._size()
17831786
return _cl._enqueue_svm_memcpy(queue, is_blocking, dest, src, **kwargs)
17841787

17851788
else:

src/wrap_cl_part_2.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -299,6 +299,7 @@ void pyopencl_expose_part_2(py::module &m)
299299
typedef svm_arg_wrapper cls;
300300
py::class_<cls>(m, "SVM", py::dynamic_attr())
301301
.def(py::init<py::object>())
302+
.def("_size", &cls::size)
302303
;
303304
}
304305

0 commit comments

Comments
 (0)