import dpctl

def get_execution_queue(qs):
    """ Given a list of :class:`dpctl.SyclQueue` objects
    returns the execution queue under compute follows data paradigm, or returns `None`
    if queues are not equivalent.
    """
    def cmp(q1, q2):
        return isinstance(q1, dpctl.SyclQueue) and isinstance(q2, dpctl.SyclQueue) and (
            (q1 is q2) or (
            (q1.sycl_context == q2.sycl_context) and 
            (q1.sycl_device == q2.sycl_device) and 
            (q1.is_in_order, q1.has_enable_profiling) == (q2.is_in_order, q2.has_enable_profiling)
         ))
    if not isinstance(qs, (list, tuple)):
        raise TypeError
    if len(qs) == 0:
        return None
    elif len(qs) == 1:
        return qs[0] 
    for q1, q2 in zip(qs, qs[1:]):
        if not cmp(q1, q2):
             return None
    return qs[0]

Now:

In [5]: get_execution_queue((dpctl.SyclQueue("gpu"), dpctl.SyclQueue("gpu")))
Out[5]: <dpctl.SyclQueue at 0x7f3aa9bb6c80>

In [6]: get_execution_queue((dpctl.SyclQueue("gpu"), dpctl.SyclQueue("gpu", property="in_order")))

In [7]: get_execution_queue((dpctl.SyclQueue("gpu"), dpctl.SyclQueue("gpu", property="in_order"))) is None
Out[7]: True

In [8]: get_execution_queue((dpctl.SyclQueue("gpu", property="in_order"), dpctl.SyclQueue("gpu", property="in_order"))) is None
Out[8]: False

dpctl should expose a function to check the whether two SYCL queues are equivalent or not #629

Description

Metadata

Metadata

Assignees

Labels

Type

Projects

Milestone

Relationships

Development

Issue actions