def METHOD_NAME(shape, dtype, usm_type):
    """ "Use device only in dpnp.ones_like() inside dpjit."""
    device = dpctl.SyclDevice().filter_string
    @dpjit
    def func(x):
        y = dpnp.ones_like(x, dtype=dtype, usm_type=usm_type, device=device)
        return y
    try:
        a = dpnp.zeros(shape, dtype=dtype, usm_type=usm_type, device=device)
        c = func(a)
    except Exception:
        pytest.fail("Calling dpnp.ones_like() inside dpjit failed.")
    if len(c.shape) == 1:
        assert c.shape[0] == a.shape[0]
    else:
        assert c.shape == a.shape
    assert c.dtype == a.dtype
    assert c.usm_type == a.usm_type
    assert c.sycl_device.filter_string == device
    if c.sycl_queue != dpctl._sycl_queue_manager.get_device_cached_queue(
        device
    ):
        pytest.xfail(
            "Returned queue does not have the same queue as cached against the device."
        )
    assert (c.asnumpy() == 1).all()