Skip to content
Snippets Groups Projects

Extend Support for CUDA and HIP kernel invocations

Merged Frederik Hennig requested to merge fhennig/cuda-invoke into master
Compare and Show latest version
14 files
+ 173
122
Preferences
Compare changes
Files
14
import pystencilssfg
import pystencilssfg
from pystencilssfg.config import SfgConfig
from pystencilssfg.config import SfgConfig
 
from os.path import splitext
 
class DocsPatchedGenerator(pystencilssfg.SourceFileGenerator):
class DocsPatchedGenerator(pystencilssfg.SourceFileGenerator):
"""Mockup wrapper around SourceFileGenerator for use in documentation
"""Mockup wrapper around SourceFileGenerator for use in documentation
@@ -28,23 +30,23 @@ class DocsPatchedGenerator(pystencilssfg.SourceFileGenerator):
@@ -28,23 +30,23 @@ class DocsPatchedGenerator(pystencilssfg.SourceFileGenerator):
def __exit__(self, exc_type, exc_value, traceback):
def __exit__(self, exc_type, exc_value, traceback):
if exc_type is None:
if exc_type is None:
self._finish_files()
self._finish_files()
 
emitter = self._get_emitter()
header_code = self._emitter.dumps(self._header_file)
header_code = emitter.dumps(self._header_file)
impl_code = (
header_ext = splitext(self._header_file.name)[1]
None
if self._impl_file is None
else self._emitter.dumps(self._impl_file)
)
mdcode = ":::::{tab-set}\n"
mdcode = ":::::{tab-set}\n"
mdcode += "::::{tab-item} Generated Header (.hpp)\n"
mdcode += f"::::{{tab-item}} Generated Header ({header_ext})\n"
mdcode += ":::{code-block} C++\n\n"
mdcode += ":::{code-block} C++\n\n"
mdcode += header_code
mdcode += header_code
mdcode += "\n:::\n::::\n"
mdcode += "\n:::\n::::\n"
if impl_code:
if self._impl_file is not None:
mdcode += "::::{tab-item} Generated Implementation (.cpp)\n"
impl_code = emitter.dumps(self._impl_file)
 
impl_ext = splitext(self._impl_file.name)[1]
 
 
mdcode += f"::::{{tab-item}} Generated Implementation ({impl_ext})\n"
mdcode += ":::{code-block} C++\n\n"
mdcode += ":::{code-block} C++\n\n"
mdcode += impl_code
mdcode += impl_code
mdcode += "\n:::\n::::\n"
mdcode += "\n:::\n::::\n"