1 files changed, 12 insertions, 3 deletions

diff --git a/include/litmus/rt_param.h b/include/litmus/rt_param.h
index 0198884eab86..a441badd30cc 100644
--- a/include/litmus/rt_param.h
+++ b/include/litmus/rt_param.h
@@ -144,6 +144,17 @@ typedef struct feedback_est{
         fp_t accum_err;
 } feedback_est_t;
 
+
+#define AVG_EST_WINDOW_SIZE 20
+
+typedef struct avg_est{
+        lt_t history[AVG_EST_WINDOW_SIZE];
+        uint16_t count;
+        uint16_t idx;
+        lt_t sum;
+        lt_t avg;
+} avg_est_t;
+
 /*      RT task parameters for scheduling extensions
  *      These parameters are inherited during clone and therefore must
  *      be explicitly set up before the task set is launched.
@@ -190,12 +201,10 @@ struct rt_param {
         long unsigned int                       held_gpus;  // bitmap of held GPUs.
 
 #ifdef CONFIG_LITMUS_AFFINITY_LOCKING
-        fp_t    gpu_fb_param_a[MIG_LAST+1];
-        fp_t    gpu_fb_param_b[MIG_LAST+1];
+        avg_est_t gpu_migration_est[MIG_LAST+1];
 
         gpu_migration_dist_t    gpu_migration;
         int                             last_gpu;
-        feedback_est_t  gpu_migration_est[MIG_LAST+1]; // local, near, med, far
 
         lt_t accum_gpu_time;
         lt_t gpu_time_stamp;