1 files changed, 15 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/common/linux/vgpu/vgpu_ivm.c b/drivers/gpu/nvgpu/common/linux/vgpu/vgpu_ivm.c
index 82475cac..90089de8 100644
--- a/drivers/gpu/nvgpu/common/linux/vgpu/vgpu_ivm.c
+++ b/drivers/gpu/nvgpu/common/linux/vgpu/vgpu_ivm.c
@@ -15,8 +15,11 @@
  */
 
 #include <nvgpu/vgpu/vgpu_ivm.h>
+
 #include <linux/tegra-ivc.h>
 
+#include "common/linux/os_linux.h"
+
 struct tegra_hv_ivm_cookie *vgpu_ivm_mempool_reserve(unsigned int id)
 {
         return tegra_hv_mempool_reserve(id);
@@ -36,3 +39,15 @@ u64 vgpu_ivm_get_size(struct tegra_hv_ivm_cookie *cookie)
 {
         return cookie->size;
 }
+
+void *vgpu_ivm_mempool_map(struct tegra_hv_ivm_cookie *cookie)
+{
+        return ioremap_cache(vgpu_ivm_get_ipa(cookie),
+                                vgpu_ivm_get_size(cookie));
+}
+
+void vgpu_ivm_mempool_unmap(struct tegra_hv_ivm_cookie *cookie,
+                void *addr)
+{
+        iounmap(addr);
+}