1 files changed, 1 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gv11b/ce_gv11b.c b/drivers/gpu/nvgpu/gv11b/ce_gv11b.c
index 8bf636b1..4f611799 100644
--- a/drivers/gpu/nvgpu/gv11b/ce_gv11b.c
+++ b/drivers/gpu/nvgpu/gv11b/ce_gv11b.c
@@ -36,7 +36,7 @@ u32 gv11b_ce_get_num_pce(struct gk20a *g)
         u32 num_pce;
         u32 ce_pce_map = gk20a_readl(g, ce_pce_map_r());
 
-        num_pce = get_count_order(ce_pce_map) + 1;
+        num_pce = hweight32(ce_pce_map);
         nvgpu_log_info(g, "num PCE: %d", num_pce);
         return num_pce;
 }