File tree Expand file tree Collapse file tree 1 file changed +4
-6
lines changed
examples/pybind11/use_dpctl_syclqueue/use_queue_device Expand file tree Collapse file tree 1 file changed +4
-6
lines changed Original file line number Diff line number Diff line change @@ -45,7 +45,7 @@ uint64_t get_device_local_mem_size(sycl::device &d)
45
45
}
46
46
47
47
py::array_t <int64_t >
48
- offloaded_array_mod (sycl::queue & q,
48
+ offloaded_array_mod (sycl::queue q,
49
49
py::array_t <int64_t , py::array::c_style> array,
50
50
int64_t mod)
51
51
{
@@ -86,13 +86,11 @@ offloaded_array_mod(sycl::queue &q,
86
86
87
87
PYBIND11_MODULE (_use_queue_device, m)
88
88
{
89
- // Import the dpctl extensions
90
- import_dpctl ();
91
89
m.def (
92
90
" get_max_compute_units" ,
93
- [=](sycl::queue & q) -> size_t {
94
- return q.get_device ()
95
- .get_info <sycl::info::device::max_compute_units>();
91
+ [=](sycl::queue q) -> size_t {
92
+ sycl::device d = q.get_device ();
93
+ return d .get_info <sycl::info::device::max_compute_units>();
96
94
},
97
95
" Computes max_compute_units property of the device underlying given "
98
96
" dpctl.SyclQueue" );
You can’t perform that action at this time.
0 commit comments