diff --git a/loopy/auto_test.py b/loopy/auto_test.py index 627a218bf147e6d49cba4fed64972e455232a334..2ef5295c5cb157179a3eeaf952cebe4e65f2d5f2 100644 --- a/loopy/auto_test.py +++ b/loopy/auto_test.py @@ -101,7 +101,7 @@ def make_ref_args(kernel, impl_arg_info, queue, parameters): argv_dtype = None if argv_dtype != arg.dtype: - arg_value = arg.dtype.type(arg_value) + arg_value = arg.dtype.numpy_dtype.type(arg_value) ref_args[arg.name] = arg_value @@ -208,7 +208,7 @@ def make_args(kernel, impl_arg_info, queue, ref_arg_data, parameters): argv_dtype = None if argv_dtype != arg.dtype: - arg_value = arg.dtype.type(arg_value) + arg_value = arg.dtype.numpy_dtype.type(arg_value) args[arg.name] = arg_value