Skip to content

Commit

Permalink
Print ref_c and disable dtypes other than float
Browse files Browse the repository at this point in the history
  • Loading branch information
ndgrigorian committed Jun 29, 2024
1 parent 1bb0621 commit daa88c0
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions dpctl/tests/test_sycl_kernel_submit.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,13 +32,13 @@
@pytest.mark.parametrize(
"ctype_str,dtype,ctypes_ctor",
[
("short", dpt.dtype("i2"), ctypes.c_short),
("int", dpt.dtype("i4"), ctypes.c_int),
("unsigned int", dpt.dtype("u4"), ctypes.c_uint),
("long", dpt.dtype(np.longlong), ctypes.c_longlong),
("unsigned long", dpt.dtype(np.ulonglong), ctypes.c_ulonglong),
# ("short", dpt.dtype("i2"), ctypes.c_short),
# ("int", dpt.dtype("i4"), ctypes.c_int),
# ("unsigned int", dpt.dtype("u4"), ctypes.c_uint),
# ("long", dpt.dtype(np.longlong), ctypes.c_longlong),
# ("unsigned long", dpt.dtype(np.ulonglong), ctypes.c_ulonglong),
("float", dpt.dtype("f4"), ctypes.c_float),
("double", dpt.dtype("f8"), ctypes.c_double),
# ("double", dpt.dtype("f8"), ctypes.c_double),
],
)
def test_create_program_from_source(ctype_str, dtype, ctypes_ctor):
Expand Down Expand Up @@ -77,6 +77,7 @@ def test_create_program_from_source(ctype_str, dtype, ctypes_ctor):
a_np = np.arange(n_elems, dtype=dtype)
b_np = np.arange(n_elems, stop=0, step=-1, dtype=dtype)
ref_c = a_np * np.asarray(d, dtype=dtype) + b_np
print(ref_c)

for r in (
[
Expand Down

0 comments on commit daa88c0

Please sign in to comment.