Skip to content
Snippets Groups Projects
Commit 72ba61c7 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Adapt test test_address_of to use create_type

parent e0a3c928
No related merge requests found
Pipeline #16689 passed with stage
in 3 minutes and 41 seconds
...@@ -3,19 +3,18 @@ ...@@ -3,19 +3,18 @@
Test of pystencils.data_types.address_of Test of pystencils.data_types.address_of
""" """
from pystencils.data_types import address_of, cast_func, PointerType
import pystencils import pystencils
from pystencils.data_types import PointerType, address_of, cast_func, create_type
from pystencils.simp.simplifications import sympy_cse from pystencils.simp.simplifications import sympy_cse
import sympy
def test_address_of(): def test_address_of():
x, y = pystencils.fields('x,y: int64[2d]') x, y = pystencils.fields('x,y: int64[2d]')
s = pystencils.TypedSymbol('s', PointerType('int64')) s = pystencils.TypedSymbol('s', PointerType(create_type('int64')))
assignments = pystencils.AssignmentCollection({ assignments = pystencils.AssignmentCollection({
s: address_of(x[0, 0]), s: address_of(x[0, 0]),
y[0, 0]: cast_func(s, 'int64') y[0, 0]: cast_func(s, create_type('int64'))
}, {}) }, {})
ast = pystencils.create_kernel(assignments) ast = pystencils.create_kernel(assignments)
...@@ -23,7 +22,7 @@ def test_address_of(): ...@@ -23,7 +22,7 @@ def test_address_of():
print(code) print(code)
assignments = pystencils.AssignmentCollection({ assignments = pystencils.AssignmentCollection({
y[0, 0]: cast_func(address_of(x[0, 0]), 'int64') y[0, 0]: cast_func(address_of(x[0, 0]), create_type('int64'))
}, {}) }, {})
ast = pystencils.create_kernel(assignments) ast = pystencils.create_kernel(assignments)
...@@ -33,11 +32,11 @@ def test_address_of(): ...@@ -33,11 +32,11 @@ def test_address_of():
def test_address_of_with_cse(): def test_address_of_with_cse():
x, y = pystencils.fields('x,y: int64[2d]') x, y = pystencils.fields('x,y: int64[2d]')
s = pystencils.TypedSymbol('s', PointerType('int64')) s = pystencils.TypedSymbol('s', PointerType(create_type('int64')))
assignments = pystencils.AssignmentCollection({ assignments = pystencils.AssignmentCollection({
y[0, 0]: cast_func(address_of(x[0, 0]), 'int64'), y[0, 0]: cast_func(address_of(x[0, 0]), create_type('int64')) + s,
x[0, 0]: cast_func(address_of(x[0, 0]), 'int64') + 1 x[0, 0]: cast_func(address_of(x[0, 0]), create_type('int64')) + 1
}, {}) }, {})
ast = pystencils.create_kernel(assignments) ast = pystencils.create_kernel(assignments)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment