remove compute_follows_data imports from dpctl.utils __init__

Author: ndgrigorian

dpctl/tests/test_utils.py

@@ -22,96 +22,6 @@
 import dpctl.utils
 
 
-def test_get_execution_queue_input_validation():
-    with pytest.raises(TypeError):
-        dpctl.utils.get_execution_queue(dict())
-
-
-def test_get_execution_queue():
-    try:
-        q = dpctl.SyclQueue()
-        q2 = dpctl.SyclQueue()
-    except dpctl.SyclQueueCreationError:
-        pytest.skip("Queue could not be create for default device")
-
-    exec_q = dpctl.utils.get_execution_queue(())
-    assert exec_q is None
-
-    exec_q = dpctl.utils.get_execution_queue([q])
-    assert exec_q is q
-
-    exec_q = dpctl.utils.get_execution_queue([q, q, q, q])
-    assert exec_q is q
-
-    exec_q = dpctl.utils.get_execution_queue((q, q, None, q))
-    assert exec_q is None
-
-    exec_q = dpctl.utils.get_execution_queue(
-        (
-            q,
-            q2,
-            q,
-        )
-    )
-    assert exec_q is None
-    q_c = dpctl.SyclQueue(q._get_capsule())
-    assert q == q_c
-    exec_q = dpctl.utils.get_execution_queue(
-        (
-            q,
-            q_c,
-            q,
-        )
-    )
-    assert exec_q == q
-
-
-def test_get_execution_queue_nonequiv():
-    try:
-        q = dpctl.SyclQueue("cpu")
-        d1, d2 = q.sycl_device.create_sub_devices(partition=[1, 1])
-        ctx = dpctl.SyclContext([q.sycl_device, d1, d2])
-        q1 = dpctl.SyclQueue(ctx, d1)
-        q2 = dpctl.SyclQueue(ctx, d2)
-    except dpctl.SyclQueueCreationError:
-        pytest.skip("Queue could not be create for default device")
-
-    exec_q = dpctl.utils.get_execution_queue((q, q1, q2))
-    assert exec_q is None
-
-
-def test_get_coerced_usm_type():
-    _t = ["device", "shared", "host"]
-
-    for i1 in range(len(_t)):
-        for i2 in range(len(_t)):
-            assert (
-                dpctl.utils.get_coerced_usm_type([_t[i1], _t[i2]])
-                == _t[min(i1, i2)]
-            )
-
-    assert dpctl.utils.get_coerced_usm_type([]) is None
-    with pytest.raises(TypeError):
-        dpctl.utils.get_coerced_usm_type(dict())
-
-
-def validate_usm_type_arg():
-    _t = ["device", "shared", "host"]
-
-    for i in range(len(_t)):
-        dpctl.utils.validate_usm_type(_t[i])
-        dpctl.utils.validate_usm_type(_t[i], allow_none=False)
-    dpctl.utils.validate_usm_type(None, allow_none=True)
-    with pytest.raises(TypeError):
-        dpctl.utils.validate_usm_type(dict(), allow_none=True)
-    with pytest.raises(TypeError):
-        dpctl.utils.validate_usm_type(dict(), allow_none=False)
-    with pytest.raises(ValueError):
-        dpctl.utils.validate_usm_type("inv", allow_none=True)
-    with pytest.raises(ValueError):
-        dpctl.utils.validate_usm_type("inv", allow_none=False)
-
-
 @pytest.mark.filterwarnings("ignore:.*:RuntimeWarning")
 def test_onetrace_enabled():
     import os

dpctl/utils/__init__.py

@@ -18,22 +18,12 @@
 A collection of utility functions.
 """
 
-from ._compute_follows_data import (
-    ExecutionPlacementError,
-    get_coerced_usm_type,
-    get_execution_queue,
-    validate_usm_type,
-)
 from ._intel_device_info import intel_device_info
 from ._onetrace_context import onetrace_enabled
 from ._order_manager import SequentialOrderManager
 
 __all__ = [
-    "get_execution_queue",
-    "get_coerced_usm_type",
-    "validate_usm_type",
     "onetrace_enabled",
     "intel_device_info",
-    "ExecutionPlacementError",
     "SequentialOrderManager",
 ]