1 files changed, 4 insertions, 2 deletions

diff --git a/drivers/gpu/drm/etnaviv/etnaviv_gem_prime.c b/drivers/gpu/drm/etnaviv/etnaviv_gem_prime.c
index 7e8fdb1859dd..62b47972a52e 100644
--- a/drivers/gpu/drm/etnaviv/etnaviv_gem_prime.c
+++ b/drivers/gpu/drm/etnaviv/etnaviv_gem_prime.c
@@ -23,10 +23,12 @@
 struct sg_table *etnaviv_gem_prime_get_sg_table(struct drm_gem_object *obj)
 {
         struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
+        int npages = obj->size >> PAGE_SHIFT;
 
-        BUG_ON(!etnaviv_obj->sgt);  /* should have already pinned! */
+        if (WARN_ON(!etnaviv_obj->pages))  /* should have already pinned! */
+                return NULL;
 
-        return etnaviv_obj->sgt;
+        return drm_prime_pages_to_sg(etnaviv_obj->pages, npages);
 }
 
 void *etnaviv_gem_prime_vmap(struct drm_gem_object *obj)