Index

A

A (stgraph.compiler.utils.OpType attribute) accept() (stgraph.compiler.passes.fusion.FusionStateMachine method) ADD (stgraph.compiler.utils.WriteType attribute) add_parent_unit() (stgraph.compiler.execution_unit.ExecutionUnit method) add_ret_val() (stgraph.compiler.execution_unit.ExecutionUnit method) add_row() (stgraph.utils.data_table.DataTable method) (stgraph.utils.DataTable method) add_user() (stgraph.compiler.program.Var method) AddOp (class in stgraph.compiler.registry) advance() (stgraph.compiler.passes.fusion.FusionStateMachine method) AggInfo (class in stgraph.compiler.code_gen.code_gen) AggMaxOp (class in stgraph.compiler.registry)

AggSumOp (class in stgraph.compiler.registry) all_bu_executed() (stgraph.compiler.executor.ExeState method) all_rets() (stgraph.compiler.execution_unit.ExecutionUnit method) any_var() (in module stgraph.compiler.utils) append() (stgraph.compiler.executor.MergedUnit method) append_stmt() (stgraph.compiler.program.Program method) ArgInfo (class in stgraph.compiler.code_gen.code_gen) args (stgraph.compiler.program.Stmt.StmtInfo attribute) (stgraph.compiler.registry.GradInfo attribute) (stgraph.compiler.registry.OpImpl property) ASSERT_DRV() (in module stgraph.compiler.code_gen.cuda_error) ASSIGN (stgraph.compiler.utils.WriteType attribute) ATOMIC (stgraph.compiler.utils.WriteType attribute)

B

backend (stgraph.compiler.val.pytorch.torch_val.TorchVal property) backend() (stgraph.compiler.val.val.Val method) backend_cb() (stgraph.compiler.backend.callback.STGraphBackend method) backend_key (stgraph.compiler.val.pytorch.torch_val.TorchVal property) backend_key() (stgraph.compiler.val.val.Val method) backward() (stgraph.compiler.backend.kernel_wrapper.KernelWrapper static method) backward_cb() (stgraph.compiler.executor.Executor method) BackwardAMaxOp (class in stgraph.compiler.registry)

BackwardLeakyReluOp (class in stgraph.compiler.registry) BackwardReluOp (class in stgraph.compiler.registry) bcast_dim() (in module stgraph.compiler.utils) begin() (stgraph.compiler.program.Program method) BinaryOpImpl (class in stgraph.compiler.registry) build_backward_csr() (in module stgraph.graph.dynamic.gpma.gpma) build_csr() (stgraph.graph.dynamic.pcsr.pcsr.PCSR method) build_reverse_csr() (stgraph.graph.dynamic.pcsr.pcsr.PCSR method)

C

calculate_kernel_launch_params() (stgraph.compiler.execution_unit.ExecutionUnit method) calculate_kernel_params() (stgraph.compiler.execution_unit.ExecutionUnit method) calculate_kernel_params_fa() (stgraph.compiler.execution_unit.ExecutionUnit method) CentralNode (class in stgraph.compiler.node) CF() (in module stgraph.compiler.passes.cf) clear_current_tensor_state() (stgraph.compiler.executor.ExeState method) clear_stmts() (stgraph.compiler.program.Program method) col_name_list (stgraph.utils.data_table.DataTable attribute) (stgraph.utils.DataTable attribute) column_indices_ptr (stgraph.graph.static.csr.CSR property) compile() (stgraph.compiler.stgraph.STGraph method) compile_cuda() (in module stgraph.compiler.code_gen.compiler) compile_with_nvcc() (in module stgraph.compiler.code_gen.compiler) compile_with_nvrtc() (in module stgraph.compiler.code_gen.compiler) compiled (stgraph.compiler.execution_unit.ExecutionUnit property) compiled() (stgraph.compiler.executor.MergedUnit method) compute (stgraph.compiler.code_gen.code_gen.AggInfo attribute) (stgraph.compiler.code_gen.code_gen.EdgeInfo attribute) (stgraph.compiler.code_gen.code_gen.NodeInfo attribute) compute_tile_sizes() (stgraph.compiler.execution_unit.ExecutionUnit method) construct_backward_mappping() (stgraph.compiler.executor.Executor method) Context (class in stgraph.compiler.stgraph)

copy() (stgraph.compiler.program.Stmt method) (stgraph.compiler.program.Var class method) copy_append_prog() (stgraph.compiler.program.Program method) copy_append_stmt() (stgraph.compiler.program.Program method) copy_append_stmts() (stgraph.compiler.program.Program method) copy_prepend_prog() (stgraph.compiler.program.Program method) copy_prepend_stmt() (stgraph.compiler.program.Program method) CoraDataLoader (class in stgraph.dataset) (class in stgraph.dataset.static.cora_dataloader) create() (stgraph.compiler.op.op_factory.OpFactory method) (stgraph.compiler.val.val_factory.ValFactory method) create_add_stmt() (stgraph.compiler.program.Stmt class method) create_binary_bcast_stmt() (stgraph.compiler.program.Stmt class method) create_context() (stgraph.compiler.execution_unit.ExecutionUnit method) create_mul_stmt() (stgraph.compiler.program.Stmt class method) create_stmt() (stgraph.compiler.program.Stmt class method) create_tensor_for_grad_vars() (stgraph.compiler.executor.Executor method) create_tensor_for_vars() (stgraph.compiler.executor.Executor method) create_var() (stgraph.compiler.program.Var class method) create_var_like() (stgraph.compiler.registry.OpImpl method) CSE() (in module stgraph.compiler.passes.cse) CSR (class in stgraph.graph.static.csr) current_fusion_type() (stgraph.compiler.passes.fusion.FusionStateMachine method)

D

D (stgraph.compiler.utils.OpType attribute) DataTable (class in stgraph.utils) (class in stgraph.utils.data_table) DCE() (in module stgraph.compiler.passes.dce) deepcopy() (stgraph.compiler.program.Program method) dep_program() (in module stgraph.compiler.passes.dependency_analysis) depends_on() (stgraph.compiler.execution_unit.ExecutionUnit method) (stgraph.compiler.program.Stmt method) DEST (stgraph.compiler.utils.ValType attribute) detach_from_stmt() (stgraph.compiler.program.Var method) device (stgraph.compiler.program.Var property) (stgraph.compiler.val.pytorch.torch_val.TorchVal property)

device() (stgraph.compiler.val.val.Val method) DeviceInfo (class in stgraph.compiler.code_gen.device_info) diff() (in module stgraph.compiler.autodiff) display() (stgraph.utils.data_table.DataTable method) (stgraph.utils.DataTable method) DstParallel (stgraph.compiler.utils.ParallelMode attribute) dtype (stgraph.compiler.val.pytorch.torch_val.TorchVal property) dtype() (stgraph.compiler.val.val.Val method) DynamicGraph (class in stgraph.graph) (class in stgraph.graph.dynamic.dynamic_graph)

E

E (stgraph.compiler.utils.OpType attribute) EDGE (stgraph.compiler.utils.ValType attribute) edge_count (stgraph.graph.dynamic.pcsr.pcsr.PCSR property) edge_update_list() (stgraph.graph.dynamic.pcsr.pcsr.PCSR method) edge_update_t() (in module stgraph.graph.dynamic.gpma.gpma) EdgeDirection (class in stgraph.compiler.utils) EdgeInfo (class in stgraph.compiler.code_gen.code_gen) eids_ptr (stgraph.graph.static.csr.CSR property) empty() (stgraph.compiler.program.Program method) end() (stgraph.compiler.program.Program method) EnglandCovidDataLoader (class in stgraph.dataset) (class in stgraph.dataset.dynamic.england_covid_dataloader)

eq_dim() (stgraph.compiler.code_gen.kernel_context.KernelContext method) execute() (stgraph.compiler.executor.Executor method) (stgraph.compiler.program.Stmt method) execute_compiled() (stgraph.compiler.executor.Executor method) execute_prog() (stgraph.compiler.executor.Executor method) execute_sym_program() (in module stgraph.compiler.passes.peephole) execute_unit() (stgraph.compiler.executor.Executor method) ExecutionUnit (class in stgraph.compiler.execution_unit) Executor (class in stgraph.compiler.executor) ExeState (class in stgraph.compiler.executor) ExpOp (class in stgraph.compiler.registry)

F

feature_size() (stgraph.compiler.execution_unit.ExecutionUnit method) FeatureAdaptiveKernel (class in stgraph.compiler.execution_unit) find_ret_var_by_id() (stgraph.compiler.program.Program method) find_var() (in module stgraph.compiler.passes.fusion) find_var_by_id() (stgraph.compiler.program.Program method) first_pow2_less_than_n() (stgraph.compiler.execution_unit.ExecutionUnit method) forward() (stgraph.compiler.backend.kernel_wrapper.KernelWrapper static method) (stgraph.nn.GCNConv method) (stgraph.nn.pytorch.static.gat_conv.GATConv method) (stgraph.nn.pytorch.static.gcn_conv.GCNConv method) (stgraph.nn.pytorch.temporal.tgcn.TGCN method)

forward_cb() (stgraph.compiler.executor.Executor method) free_backward_csr() (in module stgraph.graph.dynamic.gpma.gpma) fusable() (in module stgraph.compiler.passes.fusion) fuse() (in module stgraph.compiler.passes.fusion) FusionStateMachine (class in stgraph.compiler.passes.fusion) FusionType (class in stgraph.compiler.utils)

G

GATConv (class in stgraph.nn.pytorch.static.gat_conv) GCNConv (class in stgraph.nn) (class in stgraph.nn.pytorch.static.gcn_conv) gdata (stgraph.dataset.CoraDataLoader attribute) (stgraph.dataset.dynamic.england_covid_dataloader.EnglandCovidDataLoader attribute) (stgraph.dataset.EnglandCovidDataLoader attribute) (stgraph.dataset.HungaryCPDataLoader attribute) (stgraph.dataset.METRLADataLoader attribute) (stgraph.dataset.MontevideoBusDataLoader attribute) (stgraph.dataset.PedalMeDataLoader attribute) (stgraph.dataset.static.cora_dataloader.CoraDataLoader attribute) (stgraph.dataset.temporal.hungarycp_dataloader.HungaryCPDataLoader attribute) (stgraph.dataset.temporal.metrla_dataloader.METRLADataLoader attribute) (stgraph.dataset.temporal.montevideobus_dataloader.MontevideoBusDataLoader attribute) (stgraph.dataset.temporal.pedalme_dataloader.PedalMeDataLoader attribute) (stgraph.dataset.temporal.wikimath_dataloader.WikiMathDataLoader attribute) (stgraph.dataset.temporal.windmilloutput_dataloader.WindmillOutputDataLoader attribute) (stgraph.dataset.WikiMathDataLoader attribute) (stgraph.dataset.WindmillOutputDataLoader attribute) gen_agg_info() (in module stgraph.compiler.code_gen.code_gen) gen_agg_info_map() (stgraph.compiler.registry.OpImpl method) gen_arg_info() (in module stgraph.compiler.code_gen.code_gen) gen_code() (in module stgraph.compiler.code_gen.code_gen) (stgraph.compiler.program.Stmt method) (stgraph.compiler.registry.AddOp method) (stgraph.compiler.registry.AggMaxOp method) (stgraph.compiler.registry.AggSumOp method) (stgraph.compiler.registry.BackwardAMaxOp method) (stgraph.compiler.registry.BackwardLeakyReluOp method) (stgraph.compiler.registry.BackwardReluOp method) (stgraph.compiler.registry.ExpOp method) (stgraph.compiler.registry.GTypeCastOp method) (stgraph.compiler.registry.LeakyReluOp method) (stgraph.compiler.registry.MulOp method) (stgraph.compiler.registry.OpImpl method) (stgraph.compiler.registry.ReluOp method) (stgraph.compiler.registry.SubOp method) (stgraph.compiler.registry.TrueDivOp method) gen_cuda() (in module stgraph.compiler.code_gen.code_gen) gen_edge_info() (in module stgraph.compiler.code_gen.code_gen) gen_edge_info_map() (stgraph.compiler.registry.OpImpl method) gen_init() (stgraph.compiler.registry.AggMaxOp method) (stgraph.compiler.registry.AggSumOp method) gen_load() (stgraph.compiler.registry.OpImpl method) gen_node_info() (in module stgraph.compiler.code_gen.code_gen) gen_var() (stgraph.compiler.registry.OpImpl method) gen_write() (stgraph.compiler.registry.OpImpl method) generate_stmts_from_expr() (in module stgraph.compiler.passes.peephole) get_all_args() (stgraph.compiler.execution_unit.ExecutionUnit method) get_all_features() (stgraph.dataset.CoraDataLoader method) (stgraph.dataset.dynamic.england_covid_dataloader.EnglandCovidDataLoader method) (stgraph.dataset.EnglandCovidDataLoader method) (stgraph.dataset.METRLADataLoader method) (stgraph.dataset.MontevideoBusDataLoader method) (stgraph.dataset.static.cora_dataloader.CoraDataLoader method) (stgraph.dataset.temporal.metrla_dataloader.METRLADataLoader method) (stgraph.dataset.temporal.montevideobus_dataloader.MontevideoBusDataLoader method) get_all_targets() (stgraph.dataset.CoraDataLoader method) (stgraph.dataset.dynamic.england_covid_dataloader.EnglandCovidDataLoader method) (stgraph.dataset.EnglandCovidDataLoader method) (stgraph.dataset.HungaryCPDataLoader method) (stgraph.dataset.METRLADataLoader method) (stgraph.dataset.MontevideoBusDataLoader method) (stgraph.dataset.PedalMeDataLoader method) (stgraph.dataset.static.cora_dataloader.CoraDataLoader method) (stgraph.dataset.temporal.hungarycp_dataloader.HungaryCPDataLoader method) (stgraph.dataset.temporal.metrla_dataloader.METRLADataLoader method) (stgraph.dataset.temporal.montevideobus_dataloader.MontevideoBusDataLoader method) (stgraph.dataset.temporal.pedalme_dataloader.PedalMeDataLoader method) (stgraph.dataset.temporal.wikimath_dataloader.WikiMathDataLoader method) (stgraph.dataset.temporal.windmilloutput_dataloader.WindmillOutputDataLoader method) (stgraph.dataset.WikiMathDataLoader method) (stgraph.dataset.WindmillOutputDataLoader method) get_all_vars() (stgraph.compiler.execution_unit.ExecutionUnit method) get_array() (in module stgraph.graph.static.csr) get_backward_graph() (stgraph.graph.dynamic.dynamic_graph.DynamicGraph method) (stgraph.graph.DynamicGraph method) get_csr_ptrs() (in module stgraph.graph.dynamic.gpma.gpma) (stgraph.graph.dynamic.pcsr.pcsr.PCSR method) get_edge_weights() (stgraph.dataset.dynamic.england_covid_dataloader.EnglandCovidDataLoader method) (stgraph.dataset.EnglandCovidDataLoader method) (stgraph.dataset.HungaryCPDataLoader method) (stgraph.dataset.METRLADataLoader method) (stgraph.dataset.MontevideoBusDataLoader method) (stgraph.dataset.PedalMeDataLoader method) (stgraph.dataset.temporal.hungarycp_dataloader.HungaryCPDataLoader method) (stgraph.dataset.temporal.metrla_dataloader.METRLADataLoader method) (stgraph.dataset.temporal.montevideobus_dataloader.MontevideoBusDataLoader method) (stgraph.dataset.temporal.pedalme_dataloader.PedalMeDataLoader method) (stgraph.dataset.temporal.wikimath_dataloader.WikiMathDataLoader method) (stgraph.dataset.temporal.windmilloutput_dataloader.WindmillOutputDataLoader method) (stgraph.dataset.WikiMathDataLoader method) (stgraph.dataset.WindmillOutputDataLoader method)

get_edges() (stgraph.dataset.CoraDataLoader method) (stgraph.dataset.dynamic.england_covid_dataloader.EnglandCovidDataLoader method) (stgraph.dataset.EnglandCovidDataLoader method) (stgraph.dataset.HungaryCPDataLoader method) (stgraph.dataset.METRLADataLoader method) (stgraph.dataset.MontevideoBusDataLoader method) (stgraph.dataset.PedalMeDataLoader method) (stgraph.dataset.static.cora_dataloader.CoraDataLoader method) (stgraph.dataset.temporal.hungarycp_dataloader.HungaryCPDataLoader method) (stgraph.dataset.temporal.metrla_dataloader.METRLADataLoader method) (stgraph.dataset.temporal.montevideobus_dataloader.MontevideoBusDataLoader method) (stgraph.dataset.temporal.pedalme_dataloader.PedalMeDataLoader method) (stgraph.dataset.temporal.wikimath_dataloader.WikiMathDataLoader method) (stgraph.dataset.temporal.windmilloutput_dataloader.WindmillOutputDataLoader method) (stgraph.dataset.WikiMathDataLoader method) (stgraph.dataset.WindmillOutputDataLoader method) (stgraph.graph.dynamic.pcsr.pcsr.PCSR method) get_gpma_edge_list() (in module stgraph.graph.dynamic.gpma.gpma) get_graph() (stgraph.graph.dynamic.dynamic_graph.DynamicGraph method) (stgraph.graph.DynamicGraph method) get_graph_attr() (in module stgraph.graph.dynamic.gpma.gpma) get_in_degrees() (in module stgraph.graph.dynamic.gpma.gpma) get_n() (stgraph.graph.dynamic.pcsr.pcsr.PCSR method) get_ndata() (stgraph.graph.dynamic.dynamic_graph.DynamicGraph method) (stgraph.graph.DynamicGraph method) (stgraph.graph.static.static_graph.StaticGraph method) (stgraph.graph.StaticGraph method) (stgraph.graph.stgraph_base.STGraphBase method) (stgraph.graph.STGraphBase method) get_node_ids() (in module stgraph.graph.dynamic.gpma.gpma) get_num_edges() (stgraph.graph.dynamic.dynamic_graph.DynamicGraph method) (stgraph.graph.DynamicGraph method) (stgraph.graph.static.static_graph.StaticGraph method) (stgraph.graph.StaticGraph method) (stgraph.graph.stgraph_base.STGraphBase method) (stgraph.graph.STGraphBase method) get_num_nodes() (stgraph.graph.dynamic.dynamic_graph.DynamicGraph method) (stgraph.graph.DynamicGraph method) (stgraph.graph.static.static_graph.StaticGraph method) (stgraph.graph.StaticGraph method) (stgraph.graph.stgraph_base.STGraphBase method) (stgraph.graph.STGraphBase method) get_offset_key() (stgraph.compiler.code_gen.kernel_context.KernelContext method) get_out_degrees() (in module stgraph.graph.dynamic.gpma.gpma) get_reverse_csr_edge_list() (in module stgraph.graph.dynamic.gpma.gpma) get_val_backend() (stgraph.compiler.val.val_factory.ValFactory method) GPMA (class in stgraph.graph.dynamic.gpma.gpma) GPMAGraph (class in stgraph.graph.dynamic.gpma.gpma_graph) GPUDevice (class in stgraph.compiler.code_gen.cuda_check) grad() (stgraph.compiler.program.Stmt method) (stgraph.compiler.registry.OpImpl method) grad_impl() (stgraph.compiler.registry.AggMaxOp method) (stgraph.compiler.registry.AggSumOp method) (stgraph.compiler.registry.BackwardAMaxOp method) (stgraph.compiler.registry.BackwardLeakyReluOp method) (stgraph.compiler.registry.BackwardReluOp method) (stgraph.compiler.registry.BinaryOpImpl method) (stgraph.compiler.registry.ExpOp method) (stgraph.compiler.registry.GTypeCastOp method) (stgraph.compiler.registry.LeakyReluOp method) (stgraph.compiler.registry.OpImpl method) (stgraph.compiler.registry.ReluOp method) grad_x (stgraph.compiler.registry.GradInfo attribute) GradInfo (class in stgraph.compiler.registry) graph_type (stgraph.graph.stgraph_base.STGraphBase property) (stgraph.graph.STGraphBase property) graph_type() (stgraph.graph.dynamic.gpma.gpma_graph.GPMAGraph method) (stgraph.graph.dynamic.naive.naive_graph.NaiveGraph method) (stgraph.graph.dynamic.pcsr.pcsr_graph.PCSRGraph method) (stgraph.graph.static.static_graph.StaticGraph method) (stgraph.graph.StaticGraph method) graph_type_key() (stgraph.compiler.code_gen.kernel_context.LinearizedKernelContext method) GTypeCastOp (class in stgraph.compiler.registry)

H

has_parent() (stgraph.compiler.execution_unit.ExecutionUnit method) has_stmt() (stgraph.compiler.program.Program method)

HungaryCPDataLoader (class in stgraph.dataset) (class in stgraph.dataset.temporal.hungarycp_dataloader)

I

id (stgraph.compiler.program.Var property) (stgraph.compiler.val.val.Val property) IN (stgraph.compiler.utils.EdgeDirection attribute) in_degrees (stgraph.graph.dynamic.pcsr.pcsr.PCSR property) (stgraph.graph.static.csr.CSR property) in_degrees() (stgraph.graph.dynamic.dynamic_graph.DynamicGraph method) (stgraph.graph.dynamic.gpma.gpma_graph.GPMAGraph method) (stgraph.graph.dynamic.naive.naive_graph.NaiveGraph method) (stgraph.graph.dynamic.pcsr.pcsr_graph.PCSRGraph method) (stgraph.graph.DynamicGraph method) (stgraph.graph.static.static_graph.StaticGraph method) (stgraph.graph.StaticGraph method) infer_op_type_from_args() (in module stgraph.compiler.utils) infer_val_type() (in module stgraph.compiler.utils) init (stgraph.compiler.code_gen.code_gen.AggInfo attribute) init_gpma() (in module stgraph.graph.dynamic.gpma.gpma) init_graph_updates() (in module stgraph.graph.dynamic.gpma.gpma) init_offset_cache() (stgraph.compiler.code_gen.kernel_context.KernelContext method) INNER (stgraph.compiler.utils.WriteLocation attribute) inner_write (stgraph.compiler.code_gen.code_gen.AggInfo attribute) (stgraph.compiler.code_gen.code_gen.EdgeInfo attribute) (stgraph.compiler.code_gen.code_gen.NodeInfo attribute)

input_vars() (stgraph.compiler.program.Program method) insert_after() (stgraph.compiler.program.Stmt method) insert_before() (stgraph.compiler.program.Stmt method) insert_stmts_before() (stgraph.compiler.program.Program method) int_id (stgraph.compiler.program.Var property) is_agg() (stgraph.compiler.program.Stmt method) is_child_of() (stgraph.compiler.execution_unit.ExecutionUnit method) is_const_scalar() (in module stgraph.compiler.utils) is_dst() (stgraph.compiler.program.Stmt method) is_dstvar() (stgraph.compiler.program.Var method) is_edgevar() (stgraph.compiler.program.Var method) is_edgewise() (stgraph.compiler.program.Stmt method) is_element_wise_fusable() (stgraph.compiler.program.Stmt method) is_executed_bu() (stgraph.compiler.executor.ExeState method) is_nodevar() (stgraph.compiler.program.Var method) is_nodewise() (stgraph.compiler.program.Stmt method) is_ptr (stgraph.compiler.code_gen.code_gen.ArgInfo attribute) is_src() (stgraph.compiler.program.Stmt method) is_srcvar() (stgraph.compiler.program.Var method) is_supported() (stgraph.compiler.program.Stmt method)

J

joint_args() (stgraph.compiler.executor.MergedUnit method) joint_inputs() (stgraph.compiler.executor.MergedUnit method)

joint_optimize() (in module stgraph.compiler.passes) joint_rets() (stgraph.compiler.executor.MergedUnit method)

K

Kernel (class in stgraph.compiler.execution_unit) kernel_arg_list() (stgraph.compiler.executor.MergedUnit method) kernel_args() (stgraph.compiler.execution_unit.ExecutionUnit method) kernel_argument_used_in_stmt() (stgraph.compiler.code_gen.kernel_context.KernelContext method) kernel_arguments (stgraph.compiler.code_gen.kernel_context.StmtGenCtx attribute)

kernel_name (stgraph.compiler.execution_unit.ExecutionUnit property) kernel_run() (stgraph.compiler.execution_unit.ExecutionUnit method) KernelContext (class in stgraph.compiler.code_gen.kernel_context) KernelWrapper (class in stgraph.compiler.backend.kernel_wrapper) KernelWrapperTorch (class in stgraph.compiler.backend.pytorch.torch_kernel_wrapper)

L

label_edges() (in module stgraph.graph.dynamic.gpma.gpma) (stgraph.graph.dynamic.pcsr.pcsr.PCSR method) last() (stgraph.compiler.executor.MergedUnit method) last_stmt() (stgraph.compiler.program.Program method) launch_config (stgraph.compiler.execution_unit.V2Kernel attribute), [1] layout (stgraph.compiler.val.pytorch.torch_val.TorchVal property) layout() (stgraph.compiler.val.val.Val method)

LeakyReluOp (class in stgraph.compiler.registry) LinearizedKernelContext (class in stgraph.compiler.code_gen.kernel_context) load (stgraph.compiler.code_gen.code_gen.EdgeInfo attribute) (stgraph.compiler.code_gen.code_gen.NodeInfo attribute) log() (stgraph.compiler.code_gen.cuda_check.GPUDevice method) (stgraph.compiler.code_gen.device_info.DeviceInfo method) look_up_registry() (in module stgraph.compiler.registry)

M

materilized_vars() (stgraph.compiler.execution_unit.ExecutionUnit method) matrix_var_offset() (stgraph.compiler.code_gen.kernel_context.KernelContext method) max_dims() (stgraph.compiler.execution_unit.ExecutionUnit method) max_ret_id() (stgraph.compiler.execution_unit.ExecutionUnit method) mem_planning() (in module stgraph.compiler.passes.mem_planning) mergable() (in module stgraph.compiler.passes.fusion) merge_independent() (in module stgraph.compiler.passes.fusion) merge_program() (in module stgraph.compiler.passes.fusion) merge_stmt() (in module stgraph.compiler.passes.fusion) merge_units() (stgraph.compiler.executor.Executor method) merge_with_independent_unit() (stgraph.compiler.execution_unit.ExecutionUnit method) MergedUnit (class in stgraph.compiler.executor) METRLADataLoader (class in stgraph.dataset) (class in stgraph.dataset.temporal.metrla_dataloader) module stgraph stgraph.compiler, [1] stgraph.compiler.autodiff stgraph.compiler.backend stgraph.compiler.backend.callback stgraph.compiler.backend.kernel_wrapper stgraph.compiler.backend.mxnet stgraph.compiler.backend.pytorch stgraph.compiler.backend.pytorch.torch_callback stgraph.compiler.backend.pytorch.torch_kernel_wrapper stgraph.compiler.backend.tensorflow stgraph.compiler.code_gen stgraph.compiler.code_gen.code_gen stgraph.compiler.code_gen.compiler stgraph.compiler.code_gen.cuda_check stgraph.compiler.code_gen.cuda_driver stgraph.compiler.code_gen.cuda_error stgraph.compiler.code_gen.device_info stgraph.compiler.code_gen.kernel_context stgraph.compiler.code_gen.templates stgraph.compiler.code_gen.templates.fa stgraph.compiler.code_gen.templates.v2 stgraph.compiler.debugging stgraph.compiler.debugging.print_variables stgraph.compiler.debugging.stgraph_logger stgraph.compiler.execution_unit stgraph.compiler.executor stgraph.compiler.node stgraph.compiler.op stgraph.compiler.op.agg stgraph.compiler.op.mxnet stgraph.compiler.op.op stgraph.compiler.op.op_factory stgraph.compiler.op.pytorch stgraph.compiler.op.pytorch.torch_op stgraph.compiler.op.tensorflow stgraph.compiler.passes stgraph.compiler.passes.cf stgraph.compiler.passes.cse stgraph.compiler.passes.dce stgraph.compiler.passes.dependency_analysis stgraph.compiler.passes.fusion stgraph.compiler.passes.mem_planning stgraph.compiler.passes.peephole stgraph.compiler.passes.visualize stgraph.compiler.program stgraph.compiler.registry stgraph.compiler.schema stgraph.compiler.stgraph stgraph.compiler.utils stgraph.compiler.val stgraph.compiler.val.mxnet stgraph.compiler.val.pytorch stgraph.compiler.val.pytorch.torch_val stgraph.compiler.val.tensorflow stgraph.compiler.val.val stgraph.compiler.val.val_factory stgraph.dataset, [1] stgraph.dataset.dynamic stgraph.dataset.dynamic.england_covid_dataloader stgraph.dataset.dynamic.stgraph_dynamic_dataset stgraph.dataset.static stgraph.dataset.static.cora_dataloader stgraph.dataset.static.stgraph_static_dataset stgraph.dataset.stgraph_dataset stgraph.dataset.temporal stgraph.dataset.temporal.hungarycp_dataloader stgraph.dataset.temporal.metrla_dataloader stgraph.dataset.temporal.montevideobus_dataloader stgraph.dataset.temporal.pedalme_dataloader stgraph.dataset.temporal.stgraph_temporal_dataset stgraph.dataset.temporal.wikimath_dataloader stgraph.dataset.temporal.windmilloutput_dataloader stgraph.graph, [1] stgraph.graph.dynamic stgraph.graph.dynamic.dynamic_graph stgraph.graph.dynamic.gpma stgraph.graph.dynamic.gpma.gpma stgraph.graph.dynamic.gpma.gpma_graph stgraph.graph.dynamic.naive stgraph.graph.dynamic.naive.naive_graph stgraph.graph.dynamic.pcsr stgraph.graph.dynamic.pcsr.pcsr stgraph.graph.dynamic.pcsr.pcsr_graph stgraph.graph.static stgraph.graph.static.csr stgraph.graph.static.static_graph stgraph.graph.stgraph_base stgraph.nn, [1] stgraph.nn.pytorch stgraph.nn.pytorch.static stgraph.nn.pytorch.static.gat_conv stgraph.nn.pytorch.static.gcn_conv stgraph.nn.pytorch.temporal stgraph.nn.pytorch.temporal.tgcn stgraph.utils, [1] stgraph.utils.constants stgraph.utils.data_table

MontevideoBusDataLoader (class in stgraph.dataset) (class in stgraph.dataset.temporal.montevideobus_dataloader) MulOp (class in stgraph.compiler.registry) multiply_grad() (stgraph.compiler.registry.OpImpl method)

N

NaiveGraph (class in stgraph.graph.dynamic.naive.naive_graph) name (stgraph.compiler.code_gen.code_gen.ArgInfo attribute) (stgraph.dataset.CoraDataLoader attribute) (stgraph.dataset.dynamic.england_covid_dataloader.EnglandCovidDataLoader attribute) (stgraph.dataset.EnglandCovidDataLoader attribute) (stgraph.dataset.HungaryCPDataLoader attribute) (stgraph.dataset.METRLADataLoader attribute) (stgraph.dataset.MontevideoBusDataLoader attribute) (stgraph.dataset.PedalMeDataLoader attribute) (stgraph.dataset.static.cora_dataloader.CoraDataLoader attribute) (stgraph.dataset.temporal.hungarycp_dataloader.HungaryCPDataLoader attribute) (stgraph.dataset.temporal.metrla_dataloader.METRLADataLoader attribute) (stgraph.dataset.temporal.montevideobus_dataloader.MontevideoBusDataLoader attribute) (stgraph.dataset.temporal.pedalme_dataloader.PedalMeDataLoader attribute) (stgraph.dataset.temporal.wikimath_dataloader.WikiMathDataLoader attribute) (stgraph.dataset.temporal.windmilloutput_dataloader.WindmillOutputDataLoader attribute) (stgraph.dataset.WikiMathDataLoader attribute) (stgraph.dataset.WindmillOutputDataLoader attribute)

NbEdge (class in stgraph.compiler.node) NbNode (class in stgraph.compiler.node) NEAN (stgraph.compiler.utils.FusionType attribute) nearest_pow2() (stgraph.compiler.execution_unit.ExecutionUnit method) new_zeros_call_back() (stgraph.compiler.backend.callback.STGraphBackend method) (stgraph.compiler.backend.pytorch.torch_callback.STGraphBackendTorch method) NN (stgraph.compiler.utils.FusionType attribute) node_ids_ptr (stgraph.graph.static.csr.CSR property) NODE_NORM_SIZE (stgraph.utils.constants.SizeConstants attribute) NodeInfo (class in stgraph.compiler.code_gen.code_gen) NONE (stgraph.compiler.utils.WriteLocation attribute) (stgraph.compiler.utils.WriteType attribute) NOT_FUSIBLE (stgraph.compiler.utils.FusionType attribute)

O

Op (class in stgraph.compiler.op.op) op_name (stgraph.compiler.program.Stmt property) op_schema (stgraph.compiler.program.Stmt.StmtInfo attribute) (stgraph.compiler.registry.GradInfo attribute) (stgraph.compiler.registry.OpImpl property) OpFactory (class in stgraph.compiler.op.op_factory) OpImpl (class in stgraph.compiler.registry) optimize() (in module stgraph.compiler.passes) OpType (class in stgraph.compiler.utils) OUT (stgraph.compiler.utils.EdgeDirection attribute)

out_degrees (stgraph.graph.dynamic.pcsr.pcsr.PCSR property) (stgraph.graph.static.csr.CSR property) out_degrees() (stgraph.graph.dynamic.dynamic_graph.DynamicGraph method) (stgraph.graph.dynamic.gpma.gpma_graph.GPMAGraph method) (stgraph.graph.dynamic.naive.naive_graph.NaiveGraph method) (stgraph.graph.dynamic.pcsr.pcsr_graph.PCSRGraph method) (stgraph.graph.DynamicGraph method) (stgraph.graph.static.static_graph.StaticGraph method) (stgraph.graph.StaticGraph method) OUTER (stgraph.compiler.utils.WriteLocation attribute) outter_write (stgraph.compiler.code_gen.code_gen.AggInfo attribute)

P

parallel_mode() (stgraph.compiler.execution_unit.ExecutionUnit method) ParallelMode (class in stgraph.compiler.utils) PARAM (stgraph.compiler.utils.ValType attribute) PCSR (class in stgraph.graph.dynamic.pcsr.pcsr) PCSRGraph (class in stgraph.graph.dynamic.pcsr.pcsr_graph) PedalMeDataLoader (class in stgraph.dataset) (class in stgraph.dataset.temporal.pedalme_dataloader) PH() (in module stgraph.compiler.passes.peephole) plog_program() (in module stgraph.compiler.passes.visualize) plot_exec_units() (in module stgraph.compiler.passes.visualize)

plot_programs() (in module stgraph.compiler.passes.visualize) pop() (stgraph.compiler.executor.Stack method) prepare_compiled_kernel() (stgraph.compiler.execution_unit.ExecutionUnit method) prepend_stmt() (stgraph.compiler.program.Program method) print() (stgraph.compiler.executor.Stack method) print_gpma_info() (in module stgraph.graph.dynamic.gpma.gpma) print_log() (in module stgraph.compiler.debugging.stgraph_logger) print_stmt_args() (stgraph.compiler.program.Stmt method) Program (class in stgraph.compiler.program) program (stgraph.compiler.execution_unit.ExecutionUnit property) push() (stgraph.compiler.executor.Stack method)

Q

query_offset() (stgraph.compiler.code_gen.kernel_context.KernelContext method) (stgraph.compiler.code_gen.kernel_context.LinearizedKernelContext method)

R

read_gpu_csr() (in module stgraph.graph.dynamic.pcsr.pcsr) register_ops() (in module stgraph.compiler.registry) register_or_look_up_backend_cb() (in module stgraph.compiler.registry) ReluOp (class in stgraph.compiler.registry) remove_cur() (stgraph.compiler.program.Stmt method) render_template() (in module stgraph.compiler.code_gen.code_gen) replace_all_uses_with() (stgraph.compiler.program.Var method) replace_arg_with() (stgraph.compiler.program.Stmt method) replace_grad() (stgraph.compiler.program.Var method) requires_grad (stgraph.compiler.program.Var property) (stgraph.compiler.val.pytorch.torch_val.TorchVal property) requires_grad() (stgraph.compiler.val.val.Val method) reset() (stgraph.compiler.executor.ExeState method)

reset_graph() (stgraph.graph.dynamic.dynamic_graph.DynamicGraph method) (stgraph.graph.DynamicGraph method) reset_graph_info() (stgraph.compiler.execution_unit.ExecutionUnit method) (stgraph.compiler.execution_unit.Kernel method) reset_parameters() (stgraph.nn.GCNConv method) (stgraph.nn.pytorch.static.gat_conv.GATConv method) (stgraph.nn.pytorch.static.gcn_conv.GCNConv method) resort_vars() (stgraph.compiler.program.Program method) restart() (stgraph.compiler.executor.Executor method) ret (stgraph.compiler.registry.OpImpl property) rmv_user() (stgraph.compiler.program.Var method) row_offset_ptr (stgraph.graph.static.csr.CSR property) run() (stgraph.compiler.execution_unit.Kernel method)

S

S (stgraph.compiler.utils.OpType attribute) scalar_args (stgraph.compiler.execution_unit.V2Kernel attribute), [1] scalar_var_offset() (stgraph.compiler.code_gen.kernel_context.KernelContext method) Schema (class in stgraph.compiler.schema) set_grad() (stgraph.compiler.program.Var method) set_input() (stgraph.compiler.program.Program method) set_ndata() (stgraph.graph.dynamic.dynamic_graph.DynamicGraph method) (stgraph.graph.DynamicGraph method) (stgraph.graph.static.static_graph.StaticGraph method) (stgraph.graph.StaticGraph method) (stgraph.graph.stgraph_base.STGraphBase method) (stgraph.graph.STGraphBase method) set_new_zeros_cb() (stgraph.compiler.executor.Executor method) set_parallel_mode() (stgraph.compiler.execution_unit.ExecutionUnit method) set_raw_ptr_cb() (stgraph.compiler.executor.Executor method) set_stmt_ctx() (stgraph.compiler.code_gen.kernel_context.KernelContext method) set_to_be_grad_of() (stgraph.compiler.program.Var method) setup_context() (stgraph.compiler.backend.kernel_wrapper.KernelWrapper static method) shape_propogation() (in module stgraph.compiler.passes.peephole) (stgraph.compiler.program.Stmt method) size (stgraph.compiler.val.pytorch.torch_val.TorchVal property) size() (stgraph.compiler.val.val.Val method) SizeConstants (class in stgraph.utils) (class in stgraph.utils.constants) SRC (stgraph.compiler.utils.ValType attribute) SrcParallel (stgraph.compiler.utils.ParallelMode attribute) Stack (class in stgraph.compiler.executor) state_trans (stgraph.compiler.passes.fusion.FusionStateMachine attribute) StaticGraph (class in stgraph.graph) (class in stgraph.graph.static.static_graph) stgraph module STGraph (class in stgraph.compiler.stgraph) stgraph.compiler module, [1] stgraph.compiler.autodiff module stgraph.compiler.backend module stgraph.compiler.backend.callback module stgraph.compiler.backend.kernel_wrapper module stgraph.compiler.backend.mxnet module stgraph.compiler.backend.pytorch module stgraph.compiler.backend.pytorch.torch_callback module stgraph.compiler.backend.pytorch.torch_kernel_wrapper module stgraph.compiler.backend.tensorflow module stgraph.compiler.code_gen module stgraph.compiler.code_gen.code_gen module stgraph.compiler.code_gen.compiler module stgraph.compiler.code_gen.cuda_check module stgraph.compiler.code_gen.cuda_driver module stgraph.compiler.code_gen.cuda_error module stgraph.compiler.code_gen.device_info module stgraph.compiler.code_gen.kernel_context module stgraph.compiler.code_gen.templates module stgraph.compiler.code_gen.templates.fa module stgraph.compiler.code_gen.templates.v2 module stgraph.compiler.debugging module stgraph.compiler.debugging.print_variables module stgraph.compiler.debugging.stgraph_logger module stgraph.compiler.execution_unit module stgraph.compiler.executor module stgraph.compiler.node module stgraph.compiler.op module stgraph.compiler.op.agg module stgraph.compiler.op.mxnet module stgraph.compiler.op.op module stgraph.compiler.op.op_factory module stgraph.compiler.op.pytorch module stgraph.compiler.op.pytorch.torch_op module stgraph.compiler.op.tensorflow module stgraph.compiler.passes module stgraph.compiler.passes.cf module stgraph.compiler.passes.cse module stgraph.compiler.passes.dce module stgraph.compiler.passes.dependency_analysis module stgraph.compiler.passes.fusion module stgraph.compiler.passes.mem_planning module stgraph.compiler.passes.peephole module stgraph.compiler.passes.visualize module stgraph.compiler.program module stgraph.compiler.registry module

stgraph.compiler.schema module stgraph.compiler.stgraph module stgraph.compiler.utils module stgraph.compiler.val module stgraph.compiler.val.mxnet module stgraph.compiler.val.pytorch module stgraph.compiler.val.pytorch.torch_val module stgraph.compiler.val.tensorflow module stgraph.compiler.val.val module stgraph.compiler.val.val_factory module stgraph.dataset module, [1] stgraph.dataset.dynamic module stgraph.dataset.dynamic.england_covid_dataloader module stgraph.dataset.dynamic.stgraph_dynamic_dataset module stgraph.dataset.static module stgraph.dataset.static.cora_dataloader module stgraph.dataset.static.stgraph_static_dataset module stgraph.dataset.stgraph_dataset module stgraph.dataset.temporal module stgraph.dataset.temporal.hungarycp_dataloader module stgraph.dataset.temporal.metrla_dataloader module stgraph.dataset.temporal.montevideobus_dataloader module stgraph.dataset.temporal.pedalme_dataloader module stgraph.dataset.temporal.stgraph_temporal_dataset module stgraph.dataset.temporal.wikimath_dataloader module stgraph.dataset.temporal.windmilloutput_dataloader module stgraph.graph module, [1] stgraph.graph.dynamic module stgraph.graph.dynamic.dynamic_graph module stgraph.graph.dynamic.gpma module stgraph.graph.dynamic.gpma.gpma module stgraph.graph.dynamic.gpma.gpma_graph module stgraph.graph.dynamic.naive module stgraph.graph.dynamic.naive.naive_graph module stgraph.graph.dynamic.pcsr module stgraph.graph.dynamic.pcsr.pcsr module stgraph.graph.dynamic.pcsr.pcsr_graph module stgraph.graph.static module stgraph.graph.static.csr module stgraph.graph.static.static_graph module stgraph.graph.stgraph_base module stgraph.nn module, [1] stgraph.nn.pytorch module stgraph.nn.pytorch.static module stgraph.nn.pytorch.static.gat_conv module stgraph.nn.pytorch.static.gcn_conv module stgraph.nn.pytorch.temporal module stgraph.nn.pytorch.temporal.tgcn module stgraph.utils module, [1] stgraph.utils.constants module stgraph.utils.data_table module STGraphBackend (class in stgraph.compiler.backend.callback) STGraphBackendTorch (class in stgraph.compiler.backend.pytorch.torch_callback) STGraphBase (class in stgraph.graph) (class in stgraph.graph.stgraph_base) STGraphDataset (class in stgraph.dataset) (class in stgraph.dataset.stgraph_dataset) STGraphDynamicDataset (class in stgraph.dataset) (class in stgraph.dataset.dynamic.stgraph_dynamic_dataset) STGraphStaticDataset (class in stgraph.dataset) (class in stgraph.dataset.static.stgraph_static_dataset) STGraphTemporalDataset (class in stgraph.dataset) (class in stgraph.dataset.temporal.stgraph_temporal_dataset) Stmt (class in stgraph.compiler.program) stmt (stgraph.compiler.program.Var property) Stmt.StmtInfo (class in stgraph.compiler.program) stmt_info() (stgraph.compiler.program.Stmt method) stmt_to_trans() (stgraph.compiler.passes.fusion.FusionStateMachine static method) StmtGenCtx (class in stgraph.compiler.code_gen.kernel_context) SubOp (class in stgraph.compiler.registry) sum() (stgraph.compiler.val.pytorch.torch_val.TorchVal method) (stgraph.compiler.val.val.Val method) sum_propogation() (in module stgraph.compiler.passes.peephole)

T

targ (stgraph.compiler.registry.GradInfo attribute) tensor_raw_ptr() (stgraph.compiler.backend.callback.STGraphBackend method) (stgraph.compiler.backend.pytorch.torch_callback.STGraphBackendTorch method) TGCN (class in stgraph.nn.pytorch.temporal.tgcn) title (stgraph.utils.data_table.DataTable attribute) (stgraph.utils.DataTable attribute) tmps (stgraph.compiler.execution_unit.ExecutionUnit property) to_schema() (stgraph.compiler.op.op.Op method) (stgraph.compiler.op.pytorch.torch_op.TorchOp method) TODO (stgraph.graph.dynamic.gpma.gpma_graph.GPMAGraph attribute) (stgraph.graph.dynamic.naive.naive_graph.NaiveGraph attribute) (stgraph.graph.dynamic.pcsr.pcsr_graph.PCSRGraph attribute)

top() (stgraph.compiler.executor.Stack method) TorchOp (class in stgraph.compiler.op.pytorch.torch_op) TorchVal (class in stgraph.compiler.val.pytorch.torch_val) track_executed_bu() (stgraph.compiler.executor.ExeState method) track_tensor() (stgraph.compiler.executor.ExeState method) training (stgraph.nn.pytorch.static.gat_conv.GATConv attribute) (stgraph.nn.pytorch.static.gcn_conv.GCNConv attribute) (stgraph.nn.pytorch.temporal.tgcn.TGCN attribute) TrueDivOp (class in stgraph.compiler.registry) type (stgraph.compiler.code_gen.code_gen.ArgInfo attribute) type_eq() (stgraph.compiler.program.Stmt method)

U

union_of_rets() (stgraph.compiler.executor.MergedUnit method) unit_args() (stgraph.compiler.execution_unit.ExecutionUnit method) unit_count (stgraph.compiler.execution_unit.ExecutionUnit attribute) unit_independent() (in module stgraph.compiler.passes.fusion)

unit_rets() (stgraph.compiler.execution_unit.ExecutionUnit method) update_allnode() (stgraph.compiler.node.CentralNode method) use_fa_tmpl() (stgraph.compiler.execution_unit.ExecutionUnit method) used_by() (stgraph.compiler.program.Var method) users (stgraph.compiler.program.Var property)

V

v (stgraph.compiler.val.val.Val property) V2Kernel (class in stgraph.compiler.execution_unit), [1] Val (class in stgraph.compiler.val.val) val_type (stgraph.compiler.program.Var property) (stgraph.compiler.val.pytorch.torch_val.TorchVal property) val_type() (stgraph.compiler.val.val.Val method) ValFactory (class in stgraph.compiler.val.val_factory)

ValType (class in stgraph.compiler.utils) Var (class in stgraph.compiler.program) var_dtype (stgraph.compiler.program.Var property) var_shape (stgraph.compiler.program.Var property) vector_var_offset() (stgraph.compiler.code_gen.kernel_context.KernelContext method) view() (stgraph.compiler.val.pytorch.torch_val.TorchVal method) (stgraph.compiler.val.val.Val method)

W

weighted_in_degrees() (stgraph.graph.static.static_graph.StaticGraph method) (stgraph.graph.StaticGraph method) weighted_out_degrees (stgraph.graph.static.csr.CSR property) WikiMathDataLoader (class in stgraph.dataset) (class in stgraph.dataset.temporal.wikimath_dataloader) WindmillOutputDataLoader (class in stgraph.dataset) (class in stgraph.dataset.temporal.windmilloutput_dataloader)

write_inner() (stgraph.compiler.code_gen.kernel_context.KernelContext method) write_location (stgraph.compiler.code_gen.kernel_context.StmtGenCtx attribute) write_type (stgraph.compiler.code_gen.kernel_context.StmtGenCtx attribute) WriteLocation (class in stgraph.compiler.utils) WriteType (class in stgraph.compiler.utils)