gigl.src.common.modeling_task_specs.utils.profiler_wrapper#

Attributes#

`TMP_PROFILER_LOG_DIR_NAME`
`logger`

Classes#

TorchProfiler

Module Contents#

class gigl.src.common.modeling_task_specs.utils.profiler_wrapper.TorchProfiler(**kwargs)[source]#

profiler_context()[source]#

Return type:: torch.profiler.profile

active[source]#

profile_memory = True[source]#

record_shapes = True[source]#

repeat[source]#

trace_handler[source]#

tracing_schedule[source]#

wait[source]#

warmup[source]#

with_stack = True[source]#

gigl.src.common.modeling_task_specs.utils.profiler_wrapper.TMP_PROFILER_LOG_DIR_NAME[source]#

gigl.src.common.modeling_task_specs.utils.profiler_wrapper.logger[source]#