1 files changed, 3 insertions, 2 deletions

diff --git a/drivers/gpu/ion/ion_chunk_heap.c b/drivers/gpu/ion/ion_chunk_heap.c
index f65274d3306..687af63b9f5 100644
--- a/drivers/gpu/ion/ion_chunk_heap.c
+++ b/drivers/gpu/ion/ion_chunk_heap.c
@@ -104,8 +104,9 @@ static void ion_chunk_heap_free(struct ion_buffer *buffer)
         ion_heap_buffer_zero(buffer);
 
         for_each_sg(table->sgl, sg, table->nents, i) {
-                __dma_page_cpu_to_dev(sg_page(sg), 0, sg_dma_len(sg),
-                                      DMA_BIDIRECTIONAL);
+                if (ion_buffer_cached(buffer))
+                        __dma_page_cpu_to_dev(sg_page(sg), 0, sg_dma_len(sg),
+                                              DMA_BIDIRECTIONAL);
                 gen_pool_free(chunk_heap->pool, page_to_phys(sg_page(sg)),
                               sg_dma_len(sg));
         }