1 files changed, 35 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gm206/mm_gm206.c b/drivers/gpu/nvgpu/gm206/mm_gm206.c
new file mode 100644
index 00000000..c3763d58
--- /dev/null
+++ b/drivers/gpu/nvgpu/gm206/mm_gm206.c
@@ -0,0 +1,35 @@
+/*
+ * GM206 memory management
+ *
+ * Copyright (c) 2016, 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,
+ * version 2, as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
+ * more details.
+ */
+
+#include "gk20a/gk20a.h"
+#include "gm20b/mm_gm20b.h"
+
+#include "hw_fbpa_gm206.h"
+#include "hw_top_gm206.h"
+
+static size_t gm206_mm_get_vidmem_size(struct gk20a *g)
+{
+        u32 fbpas = top_num_fbpas_value_v(
+                        gk20a_readl(g, top_num_fbpas_r()));
+        u32 ram = fbpa_cstatus_ramamount_v(
+                        gk20a_readl(g, fbpa_cstatus_r()));
+        return (size_t)fbpas * ram * SZ_1M;
+}
+
+void gm206_init_mm(struct gpu_ops *gops)
+{
+        gm20b_init_mm(gops);
+        gops->mm.get_vidmem_size = gm206_mm_get_vidmem_size;
+}