1 files changed, 5 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c b/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
index 75df4ce5..ce23c59a 100644
--- a/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
@@ -92,4 +92,8 @@ void gk20a_ce2_nonstall_isr(struct gk20a *g)
 
         return;
 }
-
+void gk20a_init_ce2(struct gpu_ops *gops)
+{
+        gops->ce2.isr_stall = gk20a_ce2_isr;
+        gops->ce2.isr_nonstall = gk20a_ce2_nonstall_isr;
+}