diff --git a/pyopencl/tools.py b/pyopencl/tools.py index ad26be72752d6119cbb6ef34cc3835f7f78e3ee5..6a8bf4e78be3308afca5029d5f832289cece1f4e 100644 --- a/pyopencl/tools.py +++ b/pyopencl/tools.py @@ -258,12 +258,13 @@ class ScalarArg(DtypedArgument): def declarator(self): return "%s %s" % (dtype_to_ctype(self.dtype), self.name) -class OtherArgument(Argument): - def __init__(self, declarator): - self.declarator = declarator +class OtherArg(Argument): + def __init__(self, declarator, name): + self.decl = declarator + self.name = name def declarator(self): - return self.declarator + return self.decl