diff --git a/gasket_core.c b/gasket_core.c
index 65b3ab2..ae3f9c7 100644
--- a/gasket_core.c
+++ b/gasket_core.c
@@ -215,6 +215,7 @@ static int gasket_alloc_dev(struct gasket_internal_desc *internal_desc,
 	gasket_dev->dev_idx = dev_idx;
 	snprintf(gasket_dev->kobj_name, GASKET_NAME_MAX, "%s", parent_name);
 	gasket_dev->dev = get_device(parent);
+	gasket_dev->dma_dev = get_device(parent);
 	/* gasket_bar_data is uninitialized. */
 	gasket_dev->num_page_tables = driver_desc->num_page_tables;
 	/* max_page_table_size and *page table are uninit'ed */
@@ -247,6 +248,7 @@ static void gasket_free_dev(struct gasket_dev *gasket_dev)
 	internal_desc->devs[gasket_dev->dev_idx] = NULL;
 	mutex_unlock(&internal_desc->mutex);
 	put_device(gasket_dev->dev);
+	put_device(gasket_dev->dma_dev);
 	kfree(gasket_dev);
 }
 
@@ -1643,6 +1645,14 @@ void gasket_platform_remove_device(struct platform_device *pdev)
 }
 EXPORT_SYMBOL(gasket_platform_remove_device);
 
+void gasket_set_dma_device(struct gasket_dev *gasket_dev,
+			   struct device *dma_dev)
+{
+	put_device(gasket_dev->dma_dev);
+	gasket_dev->dma_dev = get_device(dma_dev);
+}
+EXPORT_SYMBOL(gasket_set_dma_device);
+
 /**
  * Lookup a name by number in a num_name table.
  * @num: Number to lookup.
diff --git a/gasket_core.h b/gasket_core.h
index 75ad11b..7301166 100644
--- a/gasket_core.h
+++ b/gasket_core.h
@@ -266,6 +266,9 @@ struct gasket_dev {
 	/* Device info */
 	struct device *dev;
 
+	/* DMA device to use, may be same as above or a parent */
+	struct device *dma_dev;
+
 	/* PCI device pointer for PCI devices */
 	struct pci_dev *pci_dev;
 
@@ -550,6 +553,10 @@ int gasket_platform_add_device(struct platform_device *pdev,
 /* Remove a platform gasket device. */
 void gasket_platform_remove_device(struct platform_device *pdev);
 
+/* Set DMA device to use (if different from PCI/platform device) */
+void gasket_set_dma_device(struct gasket_dev *gasket_dev,
+			   struct device *dma_dev);
+
 /* Enable a Gasket device. */
 int gasket_enable_device(struct gasket_dev *gasket_dev);