diff --git a/src/pystencils/backend/platforms/sycl.py b/src/pystencils/backend/platforms/sycl.py index 52f61a92454fea8e360db0e1cab75fe26b85160f..4cbd186c0536aa210addefdfa72e1fd7d22cb291 100644 --- a/src/pystencils/backend/platforms/sycl.py +++ b/src/pystencils/backend/platforms/sycl.py @@ -36,6 +36,7 @@ class SyclPlatform(Platform): self, math_function: PsMathFunction, dtype: PsType ) -> CFunction: func = math_function.func + arg_types = (dtype,) * func.num_args if isinstance(dtype, PsIeeeFloatType) and dtype.width in (16, 32, 64): match func: case ( @@ -45,9 +46,9 @@ class SyclPlatform(Platform): | MathFunctions.Tan | MathFunctions.Pow ): - return CFunction(f"sycl::{func.function_name}", func.num_args) + return CFunction(f"sycl::{func.function_name}", arg_types, dtype) case MathFunctions.Abs | MathFunctions.Min | MathFunctions.Max: - return CFunction(f"sycl::f{func.function_name}", func.num_args) + return CFunction(f"sycl::f{func.function_name}", arg_types, dtype) raise MaterializationError( f"No implementation available for function {math_function} on data type {dtype}"