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
1 file
+ 10
1
Compare changes
  • Side-by-side
  • Inline
@@ -5,9 +5,18 @@ from pystencils import Assignment, create_kernel, fields
@@ -5,9 +5,18 @@ from pystencils import Assignment, create_kernel, fields
def test_single_precision():
def test_single_precision():
s = fields("s: float32[1D]")
s = fields("s: double[1D]")
assignments = [Assignment(x, y), Assignment(s[0], x)]
assignments = [Assignment(x, y), Assignment(s[0], x)]
ast = create_kernel(assignments, data_type="float")
ast = create_kernel(assignments, data_type="float")
assert ast.body.args[0].lhs.dtype.numpy_dtype == np.dtype("float32")
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[0].rhs.dtype.numpy_dtype == np.dtype("float32")
assert ast.body.args[1].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")
Loading