def global_tracking()

in optimum_benchmark/scenarios/inference/scenario.py [0:0]

• 16 lines of code
• 7 McCabe index (conditional complexity)

    def global_tracking(self, target: str):
        with ExitStack() as context_stack:
            if self.config.energy:
                context_stack.enter_context(self.energy_tracker.track(task_name=target))
            if self.config.memory:
                context_stack.enter_context(self.memory_tracker.track())
            if self.config.latency:
                context_stack.enter_context(self.latency_tracker.session())
                context_stack.enter_context(self.latency_tracker.track())

            yield

        if self.config.latency:
            getattr(self.report, target).latency = self.latency_tracker.get_latency()
        if self.config.memory:
            getattr(self.report, target).memory = self.memory_tracker.get_max_memory()
        if self.config.energy:
            getattr(self.report, target).energy = self.energy_tracker.get_energy()