Skip to content

Commit 926dfaa

Browse files
Added calls to get_device_global_mem_size, get_device_local_mem_size
1 parent 982fcc5 commit 926dfaa

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

examples/pybind11/use_dpctl_syclqueue/example.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,13 @@
2525

2626
# Pass dpctl.SyclQueue to Pybind11 extension
2727
eu_count = eg.get_max_compute_units(q)
28+
global_mem_size = eg.get_device_global_mem_size(q.sycl_device)
29+
local_mem_size = eg.get_device_local_mem_size(q.sycl_device)
2830

2931
print(f"EU count returned by Pybind11 extension {eu_count}")
3032
print("EU count computed by dpctl {}".format(q.sycl_device.max_compute_units))
33+
print("Device's global memory size: {} bytes".format(global_mem_size))
34+
print("Device's local memory size: {} bytes".format(local_mem_size))
3135

3236
print("")
3337
print("Computing modular reduction using SYCL on a NumPy array")

0 commit comments

Comments
 (0)