ttm: Include the 'struct dev' when using the DMA API.

This makes the accounting when using 'debug_dma_dump_mappings()'
and CONFIG_DMA_API_DEBUG=y be assigned to the correct device
instead of 'fallback'.

No functional change - just cosmetic.

Signed-off-by: Konrad Rzeszutek Wilk <konrad.wilk@oracle.com>

2 files changed, 8 insertions, 7 deletions

diff --git a/drivers/gpu/drm/ttm/ttm_page_alloc.c b/drivers/gpu/drm/ttm/ttm_page_alloc.c
index 737a2a2e46a5..35849dbf3ab5 100644
--- a/drivers/gpu/drm/ttm/ttm_page_alloc.c
+++ b/drivers/gpu/drm/ttm/ttm_page_alloc.c
@@ -664,7 +664,7 @@ out:
  */
 int ttm_get_pages(struct list_head *pages, int flags,
                   enum ttm_caching_state cstate, unsigned count,
-                  dma_addr_t *dma_address)
+                  dma_addr_t *dma_address, struct device *dev)
 {
         struct ttm_page_pool *pool = ttm_get_pool(flags, cstate);
         struct page *p = NULL;
@@ -685,7 +685,7 @@ int ttm_get_pages(struct list_head *pages, int flags,
                 for (r = 0; r < count; ++r) {
                         if ((flags & TTM_PAGE_FLAG_DMA32) && dma_address) {
                                 void *addr;
-                                addr = dma_alloc_coherent(NULL, PAGE_SIZE,
+                                addr = dma_alloc_coherent(dev, PAGE_SIZE,
                                                           &dma_address[r],
                                                           gfp_flags);
                                 if (addr == NULL)
@@ -730,7 +730,7 @@ int ttm_get_pages(struct list_head *pages, int flags,
                         printk(KERN_ERR TTM_PFX
                                "Failed to allocate extra pages "
                                "for large request.");
-                        ttm_put_pages(pages, 0, flags, cstate, NULL);
+                        ttm_put_pages(pages, 0, flags, cstate, NULL, NULL);
                         return r;
                 }
         }
@@ -741,7 +741,8 @@ int ttm_get_pages(struct list_head *pages, int flags,
 
 /* Put all pages in pages list to correct pool to wait for reuse */
 void ttm_put_pages(struct list_head *pages, unsigned page_count, int flags,
-                   enum ttm_caching_state cstate, dma_addr_t *dma_address)
+                   enum ttm_caching_state cstate, dma_addr_t *dma_address,
+                   struct device *dev)
 {
         unsigned long irq_flags;
         struct ttm_page_pool *pool = ttm_get_pool(flags, cstate);
@@ -757,7 +758,7 @@ void ttm_put_pages(struct list_head *pages, unsigned page_count, int flags,
                                 void *addr = page_address(p);
                                 WARN_ON(!addr || !dma_address[r]);
                                 if (addr)
-                                        dma_free_coherent(NULL, PAGE_SIZE,
+                                        dma_free_coherent(dev, PAGE_SIZE,
                                                           addr,
                                                           dma_address[r]);
                                 dma_address[r] = 0;
diff --git a/drivers/gpu/drm/ttm/ttm_tt.c b/drivers/gpu/drm/ttm/ttm_tt.c
index 86d5b1745a45..0f8fc9ff0c53 100644
--- a/drivers/gpu/drm/ttm/ttm_tt.c
+++ b/drivers/gpu/drm/ttm/ttm_tt.c
@@ -110,7 +110,7 @@ static struct page *__ttm_tt_get_page(struct ttm_tt *ttm, int index)
                 INIT_LIST_HEAD(&h);
 
                 ret = ttm_get_pages(&h, ttm->page_flags, ttm->caching_state, 1,
-                                    &ttm->dma_address[index]);
+                                    &ttm->dma_address[index], ttm->be->bdev->dev);
 
                 if (ret != 0)
                         return NULL;
@@ -304,7 +304,7 @@ static void ttm_tt_free_alloced_pages(struct ttm_tt *ttm)
                 }
         }
         ttm_put_pages(&h, count, ttm->page_flags, ttm->caching_state,
-                      ttm->dma_address);
+                      ttm->dma_address, ttm->be->bdev->dev);
         ttm->state = tt_unpopulated;
         ttm->first_himem_page = ttm->num_pages;
         ttm->last_lomem_page = -1;