ml-energy / zeus

Deep Learning Energy Measurement and Optimization
https://ml.energy/zeus
Apache License 2.0
195 stars 25 forks source link

Lazily initialize RAPL wraparound monitor processes #121

Open jaywonchung opened 1 week ago

jaywonchung commented 1 week ago

Currently, whenever get_cpus is called, it will instantiate a RAPLFile for every CPU/DRAM domain, which will in turn spawn a wraparound monitor process. However, the user may not intend to use RAPL anyway, which makes the processes poll RAPL for nothing.

One good way is to start those processes lazily, meaning that starting the wraparound monitor processes is deferred as much as possible. That is, only when RAPL metrics are actually needed (e.g., ZeusMonitor that tracks a non-empty cpu_indices calls begin_window), the processes will be spawned only then.

@wbjin WDYT?

wbjin commented 1 week ago

I agree, we could wait until the very first read() on a RAPLFile to start the monitoring process. I think we could do something like this Initialize self.wraparound_tracker = None. In read() initialize RaplWraparoundTracker if self.wraparound_tracker == None