1 files changed, 8 insertions, 7 deletions

diff --git a/include/litmus/gpu_affinity.h b/include/litmus/gpu_affinity.h
index 6b3fb8b28745..d64a15cbf2a5 100644
--- a/include/litmus/gpu_affinity.h
+++ b/include/litmus/gpu_affinity.h
@@ -31,17 +31,18 @@ static inline lt_t get_gpu_time(struct task_struct* t)
 
 static inline lt_t get_gpu_estimate(struct task_struct* t, gpu_migration_dist_t dist)
 {
-        int i;
-        fpbuf_t temp = _fp_to_integer(t->rt_param.gpu_migration_est[dist].est);
-        lt_t val = (temp >= 0) ? temp : 0;  // never allow negative estimates...
+//      int i;
+//      fpbuf_t temp = _fp_to_integer(t->rt_param.gpu_migration_est[dist].est);
+//      lt_t val = (temp >= 0) ? temp : 0;  // never allow negative estimates...
+        lt_t val = t->rt_param.gpu_migration_est[dist].avg;
 
-        WARN_ON(temp < 0);
+//      WARN_ON(temp < 0);
 
         // lower-bound a distant migration to be at least equal to the level
         // below it.
-        for(i = dist-1; (val == 0) && (i >= MIG_LOCAL); --i) {
-                val = _fp_to_integer(t->rt_param.gpu_migration_est[i].est);
-        }
+//      for(i = dist-1; (val == 0) && (i >= MIG_LOCAL); --i) {
+//              val = _fp_to_integer(t->rt_param.gpu_migration_est[i].est);
+//      }
 
         return ((val > 0) ? val : dist+1);
 }