1 files changed, 7 insertions, 0 deletions

diff --git a/drivers/misc/sgi-gru/grumain.c b/drivers/misc/sgi-gru/grumain.c
index 347004e4f591..0c20be007b20 100644
--- a/drivers/misc/sgi-gru/grumain.c
+++ b/drivers/misc/sgi-gru/grumain.c
@@ -321,6 +321,7 @@ struct gru_thread_state *gru_alloc_gts(struct vm_area_struct *vma,
         gts->ts_tsid = tsid;
         gts->ts_ctxnum = NULLCTX;
         gts->ts_tlb_int_select = -1;
+        gts->ts_cch_req_slice = -1;
         gts->ts_sizeavail = GRU_SIZEAVAIL(PAGE_SHIFT);
         if (vma) {
                 gts->ts_mm = current->mm;
@@ -566,6 +567,12 @@ void gru_load_context(struct gru_thread_state *gts)
                 gts->ts_tlb_int_select = gru_cpu_fault_map_id();
                 cch->tlb_int_select = gts->ts_tlb_int_select;
         }
+        if (gts->ts_cch_req_slice >= 0) {
+                cch->req_slice_set_enable = 1;
+                cch->req_slice = gts->ts_cch_req_slice;
+        } else {
+                cch->req_slice_set_enable =0;
+        }
         cch->tfm_done_bit_enable = 0;
         cch->dsr_allocation_map = gts->ts_dsr_map;
         cch->cbr_allocation_map = gts->ts_cbr_map;