Skip to content
Snippets Groups Projects

fixed create_kernel parameter data_type="float" to procucde single precision

Merged Christoph Alt requested to merge ob28imeq/pystencils:fix_single_precision into master
Viewing commit c7e8e21c
Show latest version
1 file
+ 10
1
Preferences
Compare changes
@@ -5,9 +5,18 @@ from pystencils import Assignment, create_kernel, fields
def test_single_precision():
s = fields("s: float32[1D]")
s = fields("s: double[1D]")
assignments = [Assignment(x, y), Assignment(s[0], x)]
ast = create_kernel(assignments, data_type="float")
assert ast.body.args[0].lhs.dtype.numpy_dtype == np.dtype("float32")
assert ast.body.args[0].rhs.dtype.numpy_dtype == np.dtype("float32")
assert ast.body.args[1].body.args[0].rhs.dtype.numpy_dtype == np.dtype("float32")
def test_double_precision():
s = fields("s: double[1D]")
assignments = [Assignment(x, y), Assignment(s[0], x)]
ast = create_kernel(assignments, data_type="double")
assert ast.body.args[0].lhs.dtype.numpy_dtype == np.dtype("float64")
assert ast.body.args[0].rhs.dtype.numpy_dtype == np.dtype("float64")
assert ast.body.args[1].body.args[0].rhs.dtype.numpy_dtype == np.dtype("float64")