2 files changed, 9 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/include/nvgpu/vgpu/tegra_vgpu.h b/drivers/gpu/nvgpu/include/nvgpu/vgpu/tegra_vgpu.h
index 3890249d..3e4372eb 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/vgpu/tegra_vgpu.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/vgpu/tegra_vgpu.h
@@ -509,6 +509,12 @@ struct tegra_vgpu_constants_params {
         u32 l2_en_mask[TEGRA_VGPU_L2_EN_MASK];
 };
 
+enum {
+        TEGRA_VGPU_CYCLE_STATS_SNAPSHOT_CMD_FLUSH = 0,
+        TEGRA_VGPU_CYCLE_STATS_SNAPSHOT_CMD_ATTACH = 1,
+        TEGRA_VGPU_CYCLE_STATS_SNAPSHOT_CMD_DETACH = 2,
+};
+
 struct tegra_vgpu_channel_cyclestats_snapshot_params {
         u64 handle;
         u32 perfmon_start;
diff --git a/drivers/gpu/nvgpu/include/nvgpu/vgpu/vgpu_ivm.h b/drivers/gpu/nvgpu/include/nvgpu/vgpu/vgpu_ivm.h
index ba2cfc56..cecdd516 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/vgpu/vgpu_ivm.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/vgpu/vgpu_ivm.h
@@ -31,5 +31,7 @@ struct tegra_hv_ivm_cookie *vgpu_ivm_mempool_reserve(unsigned int id);
 int vgpu_ivm_mempool_unreserve(struct tegra_hv_ivm_cookie *cookie);
 u64 vgpu_ivm_get_ipa(struct tegra_hv_ivm_cookie *cookie);
 u64 vgpu_ivm_get_size(struct tegra_hv_ivm_cookie *cookie);
-
+void *vgpu_ivm_mempool_map(struct tegra_hv_ivm_cookie *cookie);
+void vgpu_ivm_mempool_unmap(struct tegra_hv_ivm_cookie *cookie,
+                void *addr);
 #endif