stgraph package¶
Subpackages¶
- stgraph.compiler package
- Subpackages
- stgraph.compiler.backend package
- 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
- stgraph.compiler.op package
- 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
- Submodules
- stgraph.compiler.autodiff module
- stgraph.compiler.execution_unit module
ExecutionUnitExecutionUnit.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.compiledExecutionUnit.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_nameExecutionUnit.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.programExecutionUnit.reset_graph_info()ExecutionUnit.set_parallel_mode()ExecutionUnit.tmpsExecutionUnit.unit_args()ExecutionUnit.unit_countExecutionUnit.unit_rets()ExecutionUnit.use_fa_tmpl()
FeatureAdaptiveKernelKernelV2Kernel
- stgraph.compiler.executor module
ExeStateExecutorExecutor.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()
MergedUnitStack
- stgraph.compiler.node module
- stgraph.compiler.program module
ProgramProgram.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()
StmtStmt.StmtInfoStmt.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_nameStmt.print_stmt_args()Stmt.remove_cur()Stmt.replace_arg_with()Stmt.shape_propogation()Stmt.stmt_info()Stmt.type_eq()
VarVar.add_user()Var.copy()Var.create_var()Var.detach_from_stmt()Var.deviceVar.idVar.int_idVar.is_dstvar()Var.is_edgevar()Var.is_nodevar()Var.is_srcvar()Var.replace_all_uses_with()Var.replace_grad()Var.requires_gradVar.rmv_user()Var.set_grad()Var.set_to_be_grad_of()Var.stmtVar.used_by()Var.usersVar.val_typeVar.var_dtypeVar.var_shape
- stgraph.compiler.registry module
- stgraph.compiler.schema module
- stgraph.compiler.stgraph module
- stgraph.compiler.utils module
- Module contents
- Subpackages
- stgraph.dataset package
- Subpackages
- stgraph.dataset.dynamic package
- stgraph.dataset.static package
- 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
- Module contents
- Subpackages
- stgraph.graph package
- stgraph.nn package
- stgraph.utils package
Module contents¶
STGraph Module