1 files changed, 2 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/common/pramin.c b/drivers/gpu/nvgpu/common/pramin.c
index 9d59f61d..4f7d6248 100644
--- a/drivers/gpu/nvgpu/common/pramin.c
+++ b/drivers/gpu/nvgpu/common/pramin.c
@@ -105,7 +105,8 @@ void nvgpu_pramin_access_batched(struct gk20a *g, struct nvgpu_mem *mem,
 
         alloc = get_vidmem_page_alloc(mem->priv.sgt->sgl);
         sgt = &alloc->sgt;
-        for (sgl = sgt->sgl; sgl; sgl = nvgpu_sgt_get_next(sgt, sgl)) {
+
+        nvgpu_sgt_for_each_sgl(sgl, sgt) {
                 if (offset >= nvgpu_sgt_get_length(sgt, sgl))
                         offset -= nvgpu_sgt_get_length(sgt, sgl);
                 else