Prepare rust side for l3 awareness

Author: tommy-u

scheds/rust/scx_mitosis/src/bpf/intf.h

@@ -46,6 +46,14 @@ enum cell_stat_idx {
 	NR_CSTATS,
 };
 
+/* Function invocation counters */
+enum counter_idx {
+	COUNTER_SELECT_CPU,
+	COUNTER_ENQUEUE,
+	COUNTER_DISPATCH,
+	NR_COUNTERS,
+};
+
 struct cpu_ctx {
 	u64 cstats[MAX_CELLS][NR_CSTATS];
 	u64 cell_cycles[MAX_CELLS];

scheds/rust/scx_mitosis/src/bpf/mitosis.bpf.c

@@ -49,6 +49,11 @@ UEI_DEFINE(uei);
 */
 struct cpu_to_l3_map cpu_to_l3 SEC(".maps");
 struct l3_to_cpus_map l3_to_cpus SEC(".maps");
+
+/*
+ * Maps for statistics
+*/
+struct function_counters_map function_counters SEC(".maps");
 struct steal_stats_map steal_stats SEC(".maps");
 
 /*

scheds/rust/scx_mitosis/src/bpf/mitosis.bpf.h

@@ -37,15 +37,16 @@
  */
 #define MAX_CPUMASK_ENTRIES (4)
 
-extern const volatile u32 nr_l3;
-
 /*
  * We don't know how big struct cpumask is at compile time, so just allocate a
  * large space and check that it is big enough at runtime
+ * TODO: This should be deduplicated with the rust code and put in intf.h
  */
 #define CPUMASK_LONG_ENTRIES (128)
 #define CPUMASK_SIZE (sizeof(long) * CPUMASK_LONG_ENTRIES)
 
+extern const volatile u32 nr_l3;
+
 enum mitosis_constants {
 	/* Invalid/unset L3 value */
 	INVALID_L3_ID = -1,
@@ -102,3 +103,11 @@ struct task_ctx {
 // These could go in mitosis.bpf.h, but we'll cross that bridge when we get
 static inline struct cell *lookup_cell(int idx);
 static inline const struct cpumask *lookup_cell_cpumask(int idx);
+
+/* MAP TYPES */
+struct function_counters_map {
+	__uint(type, BPF_MAP_TYPE_PERCPU_ARRAY);
+	__type(key, u32);
+	__type(value, u64);
+	__uint(max_entries, NR_COUNTERS);
+};