Add ODS counter for number of accelerators

Differential Revision: D52448660

below/src/main.rs

@@ -973,7 +973,7 @@ fn record(
         compress_opts.to_compression_mode()?,
         store::Format::Cbor,
     )?;
-    let mut stats = statistics::Statistics::new();
+    let mut stats = statistics::Statistics::new(init.clone());
 
     let (exit_buffer, bpf_errs) = if disable_exitstats {
         (Arc::new(Mutex::new(procfs::PidMap::new())), None)
@@ -1058,13 +1058,17 @@ fn record(
 
         match collected_sample {
             Ok(s) => {
-                match store.put(post_collect_sys_time, &DataFrame { sample: s }) {
+                let frame = DataFrame { sample: s };
+                match store.put(post_collect_sys_time, &frame) {
                     Ok(/* new shard */ true) => {
                         cleanup_store(&store, &logger, store_size_limit, /* retention */ None)?
                     }
                     Ok(/* new shard */ false) => {}
                     Err(e) => error!(logger, "{:#}", e),
                 }
+                if below_config.enable_gpu_stats {
+                    stats.report_nr_accelerators(&frame.sample);
+                }
             }
             Err(e) => {
                 // Handle cgroupfs errors

below/src/open_source/statistics.rs

@@ -12,16 +12,18 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-use std::path::Path;
+use crate::init;
 
 pub struct Statistics {}
 
 impl Statistics {
-    pub fn new() -> Statistics {
+    pub fn new(_init_token: init::InitToken) -> Statistics {
         Statistics {}
     }
 
     pub fn report_store_size<P: AsRef<Path>>(&mut self, _dir: P) {}
 
     pub fn report_collection_skew(&mut self) {}
+
+    pub fn report_nr_accelerators(&mut self, &model::Sample: sample) {}
 }