collector

AbstractPerformanceCollector

Bases: Protocol

total_timer instance-attribute

timestep_timer instance-attribute

collect_performance()

write_out_performance(backend, is_orchestrated, dt_atmos)

write_out_rank_0(backend, is_orchestrated, dt_atmos, sim_status)

start_cuda_profiler() classmethod

stop_cuda_profiler() classmethod

mark_cuda_profiler(message) classmethod

PerformanceCollector

Bases: AbstractPerformanceCollector

times_per_step = [] instance-attribute

hits_per_step = [] instance-attribute

timestep_timer = Timer() instance-attribute

total_timer = Timer() instance-attribute

experiment_name = experiment_name instance-attribute

comm = comm instance-attribute

__init__(experiment_name, comm)

clear()

collect_performance()

Take the accumulated timings and flush them into a new entry in times_per_step and hits_per_step.

write_out_rank_0(backend, is_orchestrated, dt_atmos, sim_status)

write_out_performance(backend, is_orchestrated, dt_atmos)

NullPerformanceCollector

Bases: AbstractPerformanceCollector

total_timer = NullTimer() instance-attribute

timestep_timer = NullTimer() instance-attribute

__init__()

collect_performance()

write_out_performance(backend, is_orchestrated, dt_atmos)

write_out_rank_0(backend, is_orchestrated, dt_atmos, sim_status)