1 files changed, 3 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/common/linux/pci.c b/drivers/gpu/nvgpu/common/linux/pci.c
index 905a3d39..6ebe8dda 100644
--- a/drivers/gpu/nvgpu/common/linux/pci.c
+++ b/drivers/gpu/nvgpu/common/linux/pci.c
@@ -646,8 +646,10 @@ static int nvgpu_pci_probe(struct pci_dev *pdev,
 
         np = nvgpu_get_node(g);
 
-        if (of_dma_is_coherent(np))
+        if (of_dma_is_coherent(np)) {
                 __nvgpu_set_enabled(g, NVGPU_DMA_COHERENT, true);
+                __nvgpu_set_enabled(g, NVGPU_SUPPORT_IO_COHERENCE, true);
+        }
 
         return 0;
 }