1 files changed, 3 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gp106/mm_gp106.h b/drivers/gpu/nvgpu/gp106/mm_gp106.h
index 36a89a11..73b128a6 100644
--- a/drivers/gpu/nvgpu/gp106/mm_gp106.h
+++ b/drivers/gpu/nvgpu/gp106/mm_gp106.h
@@ -1,7 +1,7 @@
 /*
  * GP106 memory management
  *
- * Copyright (c) 2016, NVIDIA CORPORATION.  All rights reserved.
+ * Copyright (c) 2016-2017, NVIDIA CORPORATION.  All rights reserved.
  *
  * This program is free software; you can redistribute it and/or modify it
  * under the terms and conditions of the GNU General Public License,
@@ -16,8 +16,8 @@
 #ifndef MM_GP106_H
 #define MM_GP106_H
 
-struct gpu_ops;
+struct gk20a;
 
-void gp106_init_mm(struct gpu_ops *gops);
+size_t gp106_mm_get_vidmem_size(struct gk20a *g);
 
 #endif