Skip to content
Snippets Groups Projects
Commit db3a7bd0 authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

More flags fixes

parent b00e0888
No related branches found
No related tags found
No related merge requests found
......@@ -635,7 +635,6 @@ class CompiledKernel:
"""
self.context = context
self.kernel = kernel
self.codegen_kwargs = codegen_kwargs
self.options = list(options)
......@@ -655,6 +654,8 @@ class CompiledKernel:
self.flags = my_flags
self.kernel = kernel.copy(flags=my_flags)
self.packing_controller = SeparateArrayPackingController(kernel)
self.output_names = tuple(arg.name for arg in self.kernel.args
......
......@@ -965,17 +965,17 @@ class LoopKernel(Record):
# {{{ direct execution
@memoize_method
def get_compiled_kernel(self, ctx, options, iflags):
def get_compiled_kernel(self, ctx, options, flags):
from loopy.compiled import CompiledKernel
return CompiledKernel(ctx, self, options=options, iflags=iflags)
return CompiledKernel(ctx, self, options=options, flags=flags)
def __call__(self, queue, **kwargs):
iflags = kwargs.pop("iflags", None)
flags = kwargs.pop("flags", None)
options = kwargs.pop("options", ())
assert isinstance(options, tuple)
return self.get_compiled_kernel(queue.context, options, iflags)(
return self.get_compiled_kernel(queue.context, options, flags)(
queue, **kwargs)
# }}}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment