1 files changed, 4 insertions, 1 deletions

diff --git a/drivers/misc/sgi-gru/grukservices.c b/drivers/misc/sgi-gru/grukservices.c
index d8bd7d84a7cf..900f7aad2286 100644
--- a/drivers/misc/sgi-gru/grukservices.c
+++ b/drivers/misc/sgi-gru/grukservices.c
@@ -672,7 +672,10 @@ int gru_kservices_init(struct gru_state *gru)
         cch->tlb_int_enable = 0;
         cch->tfm_done_bit_enable = 0;
         cch->unmap_enable = 1;
-        err = cch_allocate(cch, 0, 0, cbr_map, dsr_map);
+        cch->dsr_allocation_map = dsr_map;
+        cch->cbr_allocation_map = cbr_map;
+
+        err = cch_allocate(cch);
         if (err) {
                 gru_dbg(grudev,
                         "Unable to allocate kernel CCH: gid %d, err %d\n",