diff --git a/drivers/gpu/nvgpu/common/pmu/clk/clk_arb.c b/drivers/gpu/nvgpu/common/pmu/clk/clk_arb.c
index 7d56203ad..d057ba9fd 100644
--- a/drivers/gpu/nvgpu/common/pmu/clk/clk_arb.c
+++ b/drivers/gpu/nvgpu/common/pmu/clk/clk_arb.c
@@ -43,7 +43,7 @@
 
 int nvgpu_clk_notification_queue_alloc(struct gk20a *g,
 				struct nvgpu_clk_notification_queue *queue,
-				size_t events_number) {
+				u32 events_number) {
 	queue->notifications = nvgpu_kcalloc(g, events_number,
 		sizeof(struct nvgpu_clk_notification));
 	if (!queue->notifications)
diff --git a/drivers/gpu/nvgpu/include/nvgpu/clk_arb.h b/drivers/gpu/nvgpu/include/nvgpu/clk_arb.h
index 161ac9516..063d5d7c9 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/clk_arb.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/clk_arb.h
@@ -40,7 +40,7 @@ struct gk20a;
 #include <nvgpu/pmu/lpwr.h>
 
 #define MAX_F_POINTS 256
-#define DEFAULT_EVENT_NUMBER 32
+#define DEFAULT_EVENT_NUMBER 32U
 
 struct nvgpu_clk_dev;
 struct nvgpu_clk_arb_target;
@@ -360,7 +360,7 @@ u32 nvgpu_clk_arb_notify(struct nvgpu_clk_dev *dev,
 
 int nvgpu_clk_notification_queue_alloc(struct gk20a *g,
 				struct nvgpu_clk_notification_queue *queue,
-				size_t events_number);
+				u32 events_number);
 
 void nvgpu_clk_notification_queue_free(struct gk20a *g,
 		struct nvgpu_clk_notification_queue *queue);