V2Kernel¶

class stgraph.compiler.execution_unit.V2Kernel(num_nodes, row_offsets_ptr, col_indices_ptr, eids_ptr, max_dims, kernel_name, compiled_module, launch_config, tile_sizes)[source]¶

Bases: Kernel

The Version 2 Kernel

This class contains the parameters for the second version of the kernel written for STGraph

Parameters:

num_nodes (int) – Number of nodes present in the graph
row_offsets_ptr (c_type) – Pointer to the row offset array
col_indices_ptr (c_type) – Pointer to the column indicies array

scalar_args¶

List of the scalar arguments passed to the kernel

Type:: list[c_types]

launch_config¶

List of the kernel launch configurations

Type:: list[int]