loki.backend.cudagen module

cudagen(ir, **kwargs)

Generate standardized CUDA code from one or many IR objects/trees.

class CudaCodegen(depth=0, indent='  ', linewidth=90, **kwargs)

Bases: CppCodegen

Tree visitor to generate standardized CUDA code from IR.

standard_imports = ['stdio.h', 'stdbool.h', 'float.h', 'math.h', 'cuda.h', 'cuda_runtime.h']
visit_CallStatement(o, **kwargs)
Format call statement as

<name>(<args>)

class CudaCodeMapper(intrinsic_type_mapper, *args, **kwargs)

Bases: CppCodeMapper

A StringifyMapper-derived visitor for Pymbolic expression trees that converts an expression to a string adhering to standardized CUDA.