Skip to content
Snippets Groups Projects

Add support for sycl accessors

Merged Christoph Alt requested to merge ob28imeq/pystencils-sfg:sycl_accessor into master
All threads resolved!
Viewing commit f02641fe
Prev
Show latest version
1 file
+ 3
8
Preferences
Compare changes
@@ -10,13 +10,11 @@ import pystencils as ps
@@ -10,13 +10,11 @@ import pystencils as ps
def test_spatial_field(data_type, dim):
def test_spatial_field(data_type, dim):
f = ps.fields(f"f:{data_type}[{dim}D]")
f = ps.fields(f"f:{data_type}[{dim}D]")
ref = sycl_accessor_ref(f)
ref = sycl_accessor_ref(f)
assert (
assert f"sycl::accessor< {data_type}, {dim} > &" in str(ref.get_dtype())
f"sycl::accessor< {data_type}, {dim} > &" in ref.as_variable().name_and_type()
)
@pytest.mark.parametrize("data_type", ["double", "float"])
@pytest.mark.parametrize("data_type", ["double", "float"])
def test_to_large_dim(data_type):
def test_too_large_dim(data_type):
dim = 4
dim = 4
f = ps.fields(f"f:{data_type}[{dim}D]")
f = ps.fields(f"f:{data_type}[{dim}D]")
with pytest.raises(ValueError):
with pytest.raises(ValueError):
@@ -32,10 +30,7 @@ def test_index_field(data_type, spatial_dim, index_dims):
@@ -32,10 +30,7 @@ def test_index_field(data_type, spatial_dim, index_dims):
f = ps.fields(f"f({', '.join(index_shape)}):{data_type}[{spatial_dim}D]")
f = ps.fields(f"f({', '.join(index_shape)}):{data_type}[{spatial_dim}D]")
if total_dims <= 3:
if total_dims <= 3:
ref = sycl_accessor_ref(f)
ref = sycl_accessor_ref(f)
assert (
assert f"sycl::accessor< {data_type}, {total_dims} > &" in str(ref.get_dtype())
f"sycl::accessor< {data_type}, {total_dims} > &"
in ref.as_variable().name_and_type()
)
else:
else:
with pytest.raises(ValueError):
with pytest.raises(ValueError):
sycl_accessor_ref(f)
sycl_accessor_ref(f)