Commit dbf26cc8 authored by Martin Bauer's avatar Martin Bauer
Browse files

Merge branch 'bugfix-readd-launch-bounds' into 'master'

Bugfix: Readd __launch_bounds__ for dialect 'cuda'

See merge request pycodegen/pystencils!32
parents bda43459 8e6e15bb
......@@ -174,7 +174,7 @@ class CBackend:
def _print_KernelFunction(self, node):
function_arguments = ["%s %s" % (str(s.symbol.dtype), s.symbol.name) for s in node.get_parameters()]
launch_bounds = ""
if self.__class__ == 'cuda':
if self._dialect == 'cuda':
max_threads = node.indexing.max_threads_per_block()
if max_threads:
launch_bounds = "__launch_bounds__({}) ".format(max_threads)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment