1 files changed, 10 insertions, 13 deletions

diff --git a/drivers/gpu/nvgpu/include/nvgpu/linux/kmem.h b/drivers/gpu/nvgpu/include/nvgpu/linux/kmem.h
index 611854f2..660aac9f 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/linux/kmem.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/linux/kmem.h
@@ -21,14 +21,11 @@ struct gk20a;
 struct device;
 
 #ifdef CONFIG_NVGPU_TRACK_MEM_USAGE
-void *__nvgpu_track_vmalloc(struct gk20a *g, unsigned long size,
-                            unsigned long ip);
-void *__nvgpu_track_vzalloc(struct gk20a *g, unsigned long size,
-                            unsigned long ip);
-void *__nvgpu_track_kmalloc(struct gk20a *g, size_t size, unsigned long ip);
-void *__nvgpu_track_kzalloc(struct gk20a *g, size_t size, unsigned long ip);
-void *__nvgpu_track_kcalloc(struct gk20a *g, size_t n, size_t size,
-                            unsigned long ip);
+void *__nvgpu_track_vmalloc(struct gk20a *g, unsigned long size, void *ip);
+void *__nvgpu_track_vzalloc(struct gk20a *g, unsigned long size, void *ip);
+void *__nvgpu_track_kmalloc(struct gk20a *g, size_t size, void *ip);
+void *__nvgpu_track_kzalloc(struct gk20a *g, size_t size, void *ip);
+void *__nvgpu_track_kcalloc(struct gk20a *g, size_t n, size_t size, void *ip);
 void  __nvgpu_track_vfree(struct gk20a *g, void *addr);
 void  __nvgpu_track_kfree(struct gk20a *g, void *addr);
 #endif
@@ -39,11 +36,11 @@ void  __nvgpu_track_kfree(struct gk20a *g, void *addr);
  * These are the Linux implementations of the various kmem functions defined by
  * nvgpu. This should not be included directly - instead include <nvgpu/kmem.h>.
  */
-void *__nvgpu_kmalloc(struct gk20a *g, size_t size, unsigned long ip);
-void *__nvgpu_kzalloc(struct gk20a *g, size_t size, unsigned long ip);
-void *__nvgpu_kcalloc(struct gk20a *g, size_t n, size_t size, unsigned long ip);
-void *__nvgpu_vmalloc(struct gk20a *g, unsigned long size, unsigned long ip);
-void *__nvgpu_vzalloc(struct gk20a *g, unsigned long size, unsigned long ip);
+void *__nvgpu_kmalloc(struct gk20a *g, size_t size, void *ip);
+void *__nvgpu_kzalloc(struct gk20a *g, size_t size, void *ip);
+void *__nvgpu_kcalloc(struct gk20a *g, size_t n, size_t size, void *ip);
+void *__nvgpu_vmalloc(struct gk20a *g, unsigned long size, void *ip);
+void *__nvgpu_vzalloc(struct gk20a *g, unsigned long size, void *ip);
 void __nvgpu_kfree(struct gk20a *g, void *addr);
 void __nvgpu_vfree(struct gk20a *g, void *addr);