1 files changed, 10 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/include/nvgpu/nvgpu_mem.h b/drivers/gpu/nvgpu/include/nvgpu/nvgpu_mem.h
index 6feacff7..2b8b7015 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/nvgpu_mem.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/nvgpu_mem.h
@@ -62,6 +62,11 @@ struct nvgpu_sgt_ops {
         u64   (*sgl_gpu_addr)(struct gk20a *g, void *sgl,
                               struct nvgpu_gmmu_attrs *attrs);
         /*
+         * If left NULL then iommuable is assumed to be false.
+         */
+        bool  (*sgt_iommuable)(struct gk20a *g, struct nvgpu_sgt *sgt);
+
+        /*
          * Note: this operates on the whole SGT not a specific SGL entry.
          */
         void  (*sgt_free)(struct gk20a *g, struct nvgpu_sgt *sgt);
@@ -232,9 +237,12 @@ void *nvgpu_sgt_get_next(struct nvgpu_sgt *sgt, void *sgl);
 u64 nvgpu_sgt_get_phys(struct nvgpu_sgt *sgt, void *sgl);
 u64 nvgpu_sgt_get_dma(struct nvgpu_sgt *sgt, void *sgl);
 u64 nvgpu_sgt_get_length(struct nvgpu_sgt *sgt, void *sgl);
-u64 nvgpu_sgt_get_gpu_addr(struct nvgpu_sgt *sgt, struct gk20a *g, void *sgl,
+u64 nvgpu_sgt_get_gpu_addr(struct gk20a *g, struct nvgpu_sgt *sgt, void *sgl,
                            struct nvgpu_gmmu_attrs *attrs);
-void nvgpu_sgt_free(struct nvgpu_sgt *sgt, struct gk20a *g);
+void nvgpu_sgt_free(struct gk20a *g, struct nvgpu_sgt *sgt);
+
+bool nvgpu_sgt_iommuable(struct gk20a *g, struct nvgpu_sgt *sgt);
+u64 nvgpu_sgt_alignment(struct gk20a *g, struct nvgpu_sgt *sgt);
 
 /**
  * nvgpu_mem_create_from_mem - Create a new nvgpu_mem struct from an old one.