1 files changed, 15 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gv11b/ce_gv11b.c b/drivers/gpu/nvgpu/gv11b/ce_gv11b.c
index 2eea11df..035f2ef9 100644
--- a/drivers/gpu/nvgpu/gv11b/ce_gv11b.c
+++ b/drivers/gpu/nvgpu/gv11b/ce_gv11b.c
@@ -1,7 +1,7 @@
 /*
  * Volta GPU series Copy Engine.
  *
- * Copyright (c) 2016-2018, NVIDIA CORPORATION.  All rights reserved.
+ * Copyright (c) 2016-2019, NVIDIA CORPORATION.  All rights reserved.
  *
  * Permission is hereby granted, free of charge, to any person obtaining a
  * copy of this software and associated documentation files (the "Software"),
@@ -108,3 +108,17 @@ void gv11b_ce_mthd_buffer_fault_in_bar2_fault(struct gk20a *g)
                 }
         }
 }
+
+void gv11b_ce_init_prod_values(struct gk20a *g)
+{
+        u32 reg_val;
+        u32 num_lce, lce;
+
+        num_lce = gv11b_ce_get_num_lce(g);
+
+        for (lce = 0U; lce < num_lce; lce++) {
+                reg_val = nvgpu_readl(g, ce_lce_opt_r(lce));
+                reg_val |= ce_lce_opt_force_barriers_npl__prod_f();
+                nvgpu_writel(g, ce_lce_opt_r(lce), reg_val);
+        }
+}