def test_dpctl_api(filter_str):
device = dpctl.SyclDevice(filter_str)
with dpctl.device_context(device):
dpctl.lsplatform()
dpctl.get_current_queue()
dpctl.get_num_activated_queues()
dpctl.is_in_device_context()
def test_dpctl_api(filter_str):
if skip_test(filter_str):
pytest.skip()
device = dpctl.SyclDevice(filter_str)
with dppy.offload_to_sycl_device(device):
dpctl.lsplatform()
dpctl.get_current_queue()
dpctl.get_num_activated_queues()
dpctl.is_in_device_context()
def test_is_in_device_context_inside_nested_device_ctxt_cpu():
cpu = dpctl.SyclDevice("cpu")
n = cpu.max_compute_units
n_half = n // 2
try:
d0, d1 = cpu.create_subdevices(partition=[n_half, n - n_half])
except Exception:
pytest.skip("Could not create subdevices")
assert 0 == dpctl.get_num_activated_queues()
with dpctl.device_context(d0):
assert 1 == dpctl.get_num_activated_queues()
with dpctl.device_context(d1):
assert 2 == dpctl.get_num_activated_queues()
assert dpctl.is_in_device_context()
assert dpctl.is_in_device_context()
assert 1 == dpctl.get_num_activated_queues()
assert not dpctl.is_in_device_context()
assert 0 == dpctl.get_num_activated_queues()
def test_num_current_queues_inside_threads(self):
from threading import Thread
def SessionThread(self):
self.assertEqual(dpctl.get_num_activated_queues(), 0)
with dpctl.device_context("opencl:gpu:0"):
self.assertEqual(dpctl.get_num_activated_queues(), 1)
Session1 = Thread(target=SessionThread(self))
Session2 = Thread(target=SessionThread(self))
with dpctl.device_context("opencl:cpu:0"):
self.assertEqual(dpctl.get_num_activated_queues(), 1)
Session1.start()
Session2.start()
def test_num_current_queues_inside_threads():
from threading import Thread
def SessionThread():
assert dpctl.get_num_activated_queues() == 0
with dpctl.device_context("opencl:gpu:0"):
assert dpctl.get_num_activated_queues() == 1
Session1 = Thread(target=SessionThread())
Session2 = Thread(target=SessionThread())
with dpctl.device_context("opencl:cpu:0"):
assert dpctl.get_num_activated_queues() == 1
Session1.start()
Session2.start()
def test_num_current_queues_inside_with_clause(self):
with dpctl.device_context("opencl:cpu:0"):
self.assertEqual(dpctl.get_num_activated_queues(), 1)
with dpctl.device_context("opencl:gpu:0"):
self.assertEqual(dpctl.get_num_activated_queues(), 2)
self.assertEqual(dpctl.get_num_activated_queues(), 0)
def test_num_current_queues_outside_with_clause(self):
self.assertEqual(dpctl.get_num_activated_queues(), 0)
def SessionThread(self):
self.assertEqual(dpctl.get_num_activated_queues(), 0)
with dpctl.device_context("opencl:gpu:0"):
self.assertEqual(dpctl.get_num_activated_queues(), 1)
def SessionThread():
assert dpctl.get_num_activated_queues() == 0
with dpctl.device_context("opencl:gpu:0"):
assert dpctl.get_num_activated_queues() == 1
def test_num_current_queues_inside_with_clause():
with dpctl.device_context("opencl:cpu:0"):
assert 1 == dpctl.get_num_activated_queues()
with dpctl.device_context("opencl:gpu:0"):
assert 2 == dpctl.get_num_activated_queues()
assert 0 == dpctl.get_num_activated_queues()
def test_num_current_queues_outside_with_clause():
assert 0 == dpctl.get_num_activated_queues()
