stgraph package

Subpackages

• stgraph.compiler package
  • Subpackages
    • stgraph.compiler.backend package
      • Subpackages
      • Submodules
      • stgraph.compiler.backend.callback module
      • stgraph.compiler.backend.kernel_wrapper module
      • Module contents
    • stgraph.compiler.code_gen package
      • Subpackages
      • Submodules
      • 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
      • Module contents
    • stgraph.compiler.debugging package
      • Submodules
      • stgraph.compiler.debugging.print_variables module
      • stgraph.compiler.debugging.stgraph_logger module
      • Module contents
    • stgraph.compiler.op package
      • Subpackages
      • Submodules
      • stgraph.compiler.op.op module
      • stgraph.compiler.op.op_factory module
      • Module contents
    • stgraph.compiler.passes package
      • Submodules
      • 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
      • Module contents
    • stgraph.compiler.val package
      • Subpackages
      • Submodules
      • stgraph.compiler.val.val module
      • stgraph.compiler.val.val_factory module
      • Module contents
  • Submodules
  • stgraph.compiler.autodiff module
    • diff()
  • stgraph.compiler.execution_unit module
    • ExecutionUnit
      • ExecutionUnit.add_parent_unit()
      • ExecutionUnit.add_ret_val()
      • ExecutionUnit.all_rets()
      • ExecutionUnit.calculate_kernel_launch_params()
      • ExecutionUnit.calculate_kernel_params()
      • ExecutionUnit.calculate_kernel_params_fa()
      • ExecutionUnit.compiled
      • ExecutionUnit.compute_tile_sizes()
      • ExecutionUnit.create_context()
      • ExecutionUnit.depends_on()
      • ExecutionUnit.feature_size()
      • ExecutionUnit.first_pow2_less_than_n()
      • ExecutionUnit.get_all_args()
      • ExecutionUnit.get_all_vars()
      • ExecutionUnit.has_parent()
      • ExecutionUnit.is_child_of()
      • ExecutionUnit.kernel_args()
      • ExecutionUnit.kernel_name
      • ExecutionUnit.kernel_run()
      • ExecutionUnit.materilized_vars()
      • ExecutionUnit.max_dims()
      • ExecutionUnit.max_ret_id()
      • ExecutionUnit.merge_with_independent_unit()
      • ExecutionUnit.nearest_pow2()
      • ExecutionUnit.parallel_mode()
      • ExecutionUnit.prepare_compiled_kernel()
      • ExecutionUnit.program
      • ExecutionUnit.reset_graph_info()
      • ExecutionUnit.set_parallel_mode()
      • ExecutionUnit.tmps
      • ExecutionUnit.unit_args()
      • ExecutionUnit.unit_count
      • ExecutionUnit.unit_rets()
      • ExecutionUnit.use_fa_tmpl()
    • FeatureAdaptiveKernel
    • Kernel
      • Kernel.reset_graph_info()
      • Kernel.run()
    • V2Kernel
      • V2Kernel.scalar_args
      • V2Kernel.launch_config
  • stgraph.compiler.executor module
    • ExeState
      • ExeState.all_bu_executed()
      • ExeState.clear_current_tensor_state()
      • ExeState.is_executed_bu()
      • ExeState.reset()
      • ExeState.track_executed_bu()
      • ExeState.track_tensor()
    • Executor
      • Executor.backward_cb()
      • Executor.construct_backward_mappping()
      • Executor.create_tensor_for_grad_vars()
      • Executor.create_tensor_for_vars()
      • Executor.execute()
      • Executor.execute_compiled()
      • Executor.execute_prog()
      • Executor.execute_unit()
      • Executor.forward_cb()
      • Executor.merge_units()
      • Executor.restart()
      • Executor.set_new_zeros_cb()
      • Executor.set_raw_ptr_cb()
    • MergedUnit
      • MergedUnit.append()
      • MergedUnit.compiled()
      • MergedUnit.joint_args()
      • MergedUnit.joint_inputs()
      • MergedUnit.joint_rets()
      • MergedUnit.kernel_arg_list()
      • MergedUnit.last()
      • MergedUnit.union_of_rets()
    • Stack
      • Stack.pop()
      • Stack.print()
      • Stack.push()
      • Stack.top()
  • stgraph.compiler.node module
    • CentralNode
      • CentralNode.update_allnode()
    • NbEdge
    • NbNode
  • stgraph.compiler.program module
    • Program
      • Program.append_stmt()
      • Program.begin()
      • Program.clear_stmts()
      • Program.copy_append_prog()
      • Program.copy_append_stmt()
      • Program.copy_append_stmts()
      • Program.copy_prepend_prog()
      • Program.copy_prepend_stmt()
      • Program.deepcopy()
      • Program.empty()
      • Program.end()
      • Program.find_ret_var_by_id()
      • Program.find_var_by_id()
      • Program.has_stmt()
      • Program.input_vars()
      • Program.insert_stmts_before()
      • Program.last_stmt()
      • Program.prepend_stmt()
      • Program.resort_vars()
      • Program.set_input()
    • Stmt
      • Stmt.StmtInfo
      • Stmt.copy()
      • Stmt.create_add_stmt()
      • Stmt.create_binary_bcast_stmt()
      • Stmt.create_mul_stmt()
      • Stmt.create_stmt()
      • Stmt.depends_on()
      • Stmt.execute()
      • Stmt.gen_code()
      • Stmt.grad()
      • Stmt.insert_after()
      • Stmt.insert_before()
      • Stmt.is_agg()
      • Stmt.is_dst()
      • Stmt.is_edgewise()
      • Stmt.is_element_wise_fusable()
      • Stmt.is_nodewise()
      • Stmt.is_src()
      • Stmt.is_supported()
      • Stmt.op_name
      • Stmt.print_stmt_args()
      • Stmt.remove_cur()
      • Stmt.replace_arg_with()
      • Stmt.shape_propogation()
      • Stmt.stmt_info()
      • Stmt.type_eq()
    • Var
      • Var.add_user()
      • Var.copy()
      • Var.create_var()
      • Var.detach_from_stmt()
      • Var.device
      • Var.id
      • Var.int_id
      • Var.is_dstvar()
      • Var.is_edgevar()
      • Var.is_nodevar()
      • Var.is_srcvar()
      • Var.replace_all_uses_with()
      • Var.replace_grad()
      • Var.requires_grad
      • Var.rmv_user()
      • Var.set_grad()
      • Var.set_to_be_grad_of()
      • Var.stmt
      • Var.used_by()
      • Var.users
      • Var.val_type
      • Var.var_dtype
      • Var.var_shape
  • stgraph.compiler.registry module
    • AddOp
      • AddOp.gen_code()
    • AggMaxOp
      • AggMaxOp.gen_code()
      • AggMaxOp.gen_init()
      • AggMaxOp.grad_impl()
    • AggSumOp
      • AggSumOp.gen_code()
      • AggSumOp.gen_init()
      • AggSumOp.grad_impl()
    • BackwardAMaxOp
      • BackwardAMaxOp.gen_code()
      • BackwardAMaxOp.grad_impl()
    • BackwardLeakyReluOp
      • BackwardLeakyReluOp.gen_code()
      • BackwardLeakyReluOp.grad_impl()
    • BackwardReluOp
      • BackwardReluOp.gen_code()
      • BackwardReluOp.grad_impl()
    • BinaryOpImpl
      • BinaryOpImpl.grad_impl()
    • ExpOp
      • ExpOp.gen_code()
      • ExpOp.grad_impl()
    • GTypeCastOp
      • GTypeCastOp.gen_code()
      • GTypeCastOp.grad_impl()
    • GradInfo
      • GradInfo.args
      • GradInfo.grad_x
      • GradInfo.op_schema
      • GradInfo.targ
    • LeakyReluOp
      • LeakyReluOp.gen_code()
      • LeakyReluOp.grad_impl()
    • MulOp
      • MulOp.gen_code()
    • OpImpl
      • OpImpl.args
      • OpImpl.create_var_like()
      • OpImpl.gen_agg_info_map()
      • OpImpl.gen_code()
      • OpImpl.gen_edge_info_map()
      • OpImpl.gen_load()
      • OpImpl.gen_var()
      • OpImpl.gen_write()
      • OpImpl.grad()
      • OpImpl.grad_impl()
      • OpImpl.multiply_grad()
      • OpImpl.op_schema
      • OpImpl.ret
    • ReluOp
      • ReluOp.gen_code()
      • ReluOp.grad_impl()
    • SubOp
      • SubOp.gen_code()
    • TrueDivOp
      • TrueDivOp.gen_code()
    • look_up_registry()
    • register_ops()
    • register_or_look_up_backend_cb()
  • stgraph.compiler.schema module
    • Schema
  • stgraph.compiler.stgraph module
    • Context
    • STGraph
      • STGraph.compile()
  • stgraph.compiler.utils module
    • EdgeDirection
      • EdgeDirection.IN
      • EdgeDirection.OUT
    • FusionType
      • FusionType.NEAN
      • FusionType.NN
      • FusionType.NOT_FUSIBLE
    • OpType
      • OpType.A
      • OpType.D
      • OpType.E
      • OpType.S
    • ParallelMode
      • ParallelMode.DstParallel
      • ParallelMode.SrcParallel
    • ValType
      • ValType.DEST
      • ValType.EDGE
      • ValType.PARAM
      • ValType.SRC
    • WriteLocation
      • WriteLocation.INNER
      • WriteLocation.NONE
      • WriteLocation.OUTER
    • WriteType
      • WriteType.ADD
      • WriteType.ASSIGN
      • WriteType.ATOMIC
      • WriteType.NONE
    • any_var()
    • bcast_dim()
    • infer_op_type_from_args()
    • infer_val_type()
    • is_const_scalar()
  • Module contents
• stgraph.dataset package
  • Subpackages
    • stgraph.dataset.dynamic package
      • Submodules
      • stgraph.dataset.dynamic.england_covid_dataloader module
      • stgraph.dataset.dynamic.stgraph_dynamic_dataset module
      • Module contents
    • stgraph.dataset.static package
      • Submodules
      • stgraph.dataset.static.cora_dataloader module
      • stgraph.dataset.static.stgraph_static_dataset module
      • Module contents
    • stgraph.dataset.temporal package
      • Submodules
      • 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
      • Module contents
  • Submodules
  • stgraph.dataset.stgraph_dataset module
    • STGraphDataset
  • Module contents
• stgraph.graph package
  • Subpackages
    • stgraph.graph.dynamic package
      • Subpackages
      • Submodules
      • stgraph.graph.dynamic.dynamic_graph module
      • Module contents
    • stgraph.graph.static package
      • Submodules
      • stgraph.graph.static.csr module
      • stgraph.graph.static.static_graph module
      • Module contents
  • Submodules
  • stgraph.graph.stgraph_base module
    • STGraphBase
      • STGraphBase.get_ndata()
      • STGraphBase.get_num_edges()
      • STGraphBase.get_num_nodes()
      • STGraphBase.graph_type
      • STGraphBase.set_ndata()
  • Module contents
• stgraph.nn package
  • Subpackages
    • stgraph.nn.pytorch package
      • Subpackages
      • Module contents
  • Module contents
• stgraph.utils package
  • Submodules
  • stgraph.utils.constants module
    • SizeConstants
      • SizeConstants.NODE_NORM_SIZE
  • stgraph.utils.data_table module
    • DataTable
      • DataTable.title
      • DataTable.col_name_list
      • DataTable.add_row()
      • DataTable.display()
  • Module contents

Module contents

STGraph Module