]> git.proxmox.com Git - mirror_ubuntu-hirsute-kernel.git/commitdiff
drm/panfrost: Implement per FD address spaces
authorRob Herring <robh@kernel.org>
Tue, 13 Aug 2019 15:01:15 +0000 (09:01 -0600)
committerRob Herring <robh@kernel.org>
Mon, 19 Aug 2019 16:34:57 +0000 (11:34 -0500)
Up until now, a single shared GPU address space was used. This is not
ideal as there's no protection between processes and doesn't work for
supporting the same GPU/CPU VA feature. Most importantly, this will
hopefully mitigate Alyssa's fear of WebGL, whatever that is.

Most of the changes here are moving struct drm_mm and struct
panfrost_mmu objects from the per device struct to the per FD struct.
The critical function is panfrost_mmu_as_get() which handles allocating
and switching the h/w address spaces.

There's 3 states an AS can be in: free, allocated, and in use. When a
job runs, it requests an address space and then marks it not in use when
job is complete(but stays assigned). The first time thru, we find a free
AS in the alloc_mask and assign the AS to the FD. Then the next time
thru, we most likely already have our AS and we just mark it in use with
a ref count. We need a ref count because we have multiple job slots. If
the job/FD doesn't have an AS assigned and there are no free ones, then
we pick an allocated one not in use from our LRU list and switch the AS
from the old FD to the new one.

Cc: Tomeu Vizoso <tomeu.vizoso@collabora.com>
Cc: David Airlie <airlied@linux.ie>
Cc: Daniel Vetter <daniel@ffwll.ch>
Cc: Robin Murphy <robin.murphy@arm.com>
Cc: Steven Price <steven.price@arm.com>
Signed-off-by: Rob Herring <robh@kernel.org>
Acked-by: Alyssa Rosenzweig <alyssa.rosenzweig@collabora.com>
Reviewed-by: Steven Price <steven.price@arm.com>
Link: https://patchwork.freedesktop.org/patch/msgid/20190813150115.30338-1-robh@kernel.org
drivers/gpu/drm/panfrost/TODO
drivers/gpu/drm/panfrost/panfrost_device.c
drivers/gpu/drm/panfrost/panfrost_device.h
drivers/gpu/drm/panfrost/panfrost_drv.c
drivers/gpu/drm/panfrost/panfrost_gem.c
drivers/gpu/drm/panfrost/panfrost_gem.h
drivers/gpu/drm/panfrost/panfrost_job.c
drivers/gpu/drm/panfrost/panfrost_mmu.c
drivers/gpu/drm/panfrost/panfrost_mmu.h

index e7727b292355acb77e88c319fa0b8f714a4f16d0..536a0d4f8d29355dd7915856a37c9d595c6bff48 100644 (file)
@@ -6,10 +6,6 @@
   - Bifrost specific feature and issue handling
   - Coherent DMA support
 
-- Per FD address space support. The h/w supports multiple addresses spaces.
-  The hard part is handling when more address spaces are needed than what
-  the h/w provides.
-
 - Support userspace controlled GPU virtual addresses. Needed for Vulkan. (Tomeu)
 
 - Compute job support. So called 'compute only' jobs need to be plumbed up to
index 9814f4ccbd263c229d0ca732cdd3b6af6d0736f4..4da71bb56c2030974ee4f7e6f85128a297be7e25 100644 (file)
@@ -123,8 +123,10 @@ int panfrost_device_init(struct panfrost_device *pfdev)
        mutex_init(&pfdev->sched_lock);
        mutex_init(&pfdev->reset_lock);
        INIT_LIST_HEAD(&pfdev->scheduled_jobs);
+       INIT_LIST_HEAD(&pfdev->as_lru_list);
 
        spin_lock_init(&pfdev->hwaccess_lock);
+       spin_lock_init(&pfdev->as_lock);
 
        err = panfrost_clk_init(pfdev);
        if (err) {
index 4e5641db9c7e58f0e31a42da31f18b81cad73710..f503c566e99f33ebdc00542f83122c9a75611629 100644 (file)
@@ -5,6 +5,8 @@
 #ifndef __PANFROST_DEVICE_H__
 #define __PANFROST_DEVICE_H__
 
+#include <linux/atomic.h>
+#include <linux/io-pgtable.h>
 #include <linux/spinlock.h>
 #include <drm/drm_device.h>
 #include <drm/drm_mm.h>
@@ -63,9 +65,6 @@ struct panfrost_device {
 
        spinlock_t hwaccess_lock;
 
-       struct drm_mm mm;
-       spinlock_t mm_lock;
-
        void __iomem *iomem;
        struct clk *clock;
        struct clk *bus_clock;
@@ -74,7 +73,11 @@ struct panfrost_device {
 
        struct panfrost_features features;
 
-       struct panfrost_mmu *mmu;
+       spinlock_t as_lock;
+       unsigned long as_in_use_mask;
+       unsigned long as_alloc_mask;
+       struct list_head as_lru_list;
+
        struct panfrost_job_slot *js;
 
        struct panfrost_job *jobs[NUM_JOB_SLOTS];
@@ -98,10 +101,23 @@ struct panfrost_device {
        } devfreq;
 };
 
+struct panfrost_mmu {
+       struct io_pgtable_cfg pgtbl_cfg;
+       struct io_pgtable_ops *pgtbl_ops;
+       struct mutex lock;
+       int as;
+       atomic_t as_count;
+       struct list_head list;
+};
+
 struct panfrost_file_priv {
        struct panfrost_device *pfdev;
 
        struct drm_sched_entity sched_entity[NUM_JOB_SLOTS];
+
+       struct panfrost_mmu mmu;
+       struct drm_mm mm;
+       spinlock_t mm_lock;
 };
 
 static inline struct panfrost_device *to_panfrost_device(struct drm_device *ddev)
index b417546586814930b5fd90e46d86c517c13f1808..ae70200be44a3b50b8c55c1b7ef8341ee3918cfb 100644 (file)
@@ -403,6 +403,7 @@ static void panfrost_drm_mm_color_adjust(const struct drm_mm_node *node,
 static int
 panfrost_open(struct drm_device *dev, struct drm_file *file)
 {
+       int ret;
        struct panfrost_device *pfdev = dev->dev_private;
        struct panfrost_file_priv *panfrost_priv;
 
@@ -413,7 +414,28 @@ panfrost_open(struct drm_device *dev, struct drm_file *file)
        panfrost_priv->pfdev = pfdev;
        file->driver_priv = panfrost_priv;
 
-       return panfrost_job_open(panfrost_priv);
+       spin_lock_init(&panfrost_priv->mm_lock);
+
+       /* 4G enough for now. can be 48-bit */
+       drm_mm_init(&panfrost_priv->mm, SZ_32M >> PAGE_SHIFT, (SZ_4G - SZ_32M) >> PAGE_SHIFT);
+       panfrost_priv->mm.color_adjust = panfrost_drm_mm_color_adjust;
+
+       ret = panfrost_mmu_pgtable_alloc(panfrost_priv);
+       if (ret)
+               goto err_pgtable;
+
+       ret = panfrost_job_open(panfrost_priv);
+       if (ret)
+               goto err_job;
+
+       return 0;
+
+err_job:
+       panfrost_mmu_pgtable_free(panfrost_priv);
+err_pgtable:
+       drm_mm_takedown(&panfrost_priv->mm);
+       kfree(panfrost_priv);
+       return ret;
 }
 
 static void
@@ -424,6 +446,8 @@ panfrost_postclose(struct drm_device *dev, struct drm_file *file)
        panfrost_perfcnt_close(panfrost_priv);
        panfrost_job_close(panfrost_priv);
 
+       panfrost_mmu_pgtable_free(panfrost_priv);
+       drm_mm_takedown(&panfrost_priv->mm);
        kfree(panfrost_priv);
 }
 
@@ -496,14 +520,9 @@ static int panfrost_probe(struct platform_device *pdev)
        ddev->dev_private = pfdev;
        pfdev->ddev = ddev;
 
-       spin_lock_init(&pfdev->mm_lock);
        mutex_init(&pfdev->shrinker_lock);
        INIT_LIST_HEAD(&pfdev->shrinker_list);
 
-       /* 4G enough for now. can be 48-bit */
-       drm_mm_init(&pfdev->mm, SZ_32M >> PAGE_SHIFT, (SZ_4G - SZ_32M) >> PAGE_SHIFT);
-       pfdev->mm.color_adjust = panfrost_drm_mm_color_adjust;
-
        pm_runtime_use_autosuspend(pfdev->dev);
        pm_runtime_set_autosuspend_delay(pfdev->dev, 50); /* ~3 frames */
        pm_runtime_enable(pfdev->dev);
index e71f27c4041ea737097be79cb90b9fa517d4309b..e084bc4e908343971a34f186e0063e1393bfc662 100644 (file)
@@ -47,8 +47,8 @@ static int panfrost_gem_open(struct drm_gem_object *obj, struct drm_file *file_p
        size_t size = obj->size;
        u64 align;
        struct panfrost_gem_object *bo = to_panfrost_bo(obj);
-       struct panfrost_device *pfdev = obj->dev->dev_private;
        unsigned long color = bo->noexec ? PANFROST_BO_NOEXEC : 0;
+       struct panfrost_file_priv *priv = file_priv->driver_priv;
 
        /*
         * Executable buffers cannot cross a 16MB boundary as the program
@@ -61,8 +61,9 @@ static int panfrost_gem_open(struct drm_gem_object *obj, struct drm_file *file_p
        else
                align = size >= SZ_2M ? SZ_2M >> PAGE_SHIFT : 0;
 
-       spin_lock(&pfdev->mm_lock);
-       ret = drm_mm_insert_node_generic(&pfdev->mm, &bo->node,
+       bo->mmu = &priv->mmu;
+       spin_lock(&priv->mm_lock);
+       ret = drm_mm_insert_node_generic(&priv->mm, &bo->node,
                                         size >> PAGE_SHIFT, align, color, 0);
        if (ret)
                goto out;
@@ -73,22 +74,22 @@ static int panfrost_gem_open(struct drm_gem_object *obj, struct drm_file *file_p
                        drm_mm_remove_node(&bo->node);
        }
 out:
-       spin_unlock(&pfdev->mm_lock);
+       spin_unlock(&priv->mm_lock);
        return ret;
 }
 
 static void panfrost_gem_close(struct drm_gem_object *obj, struct drm_file *file_priv)
 {
        struct panfrost_gem_object *bo = to_panfrost_bo(obj);
-       struct panfrost_device *pfdev = obj->dev->dev_private;
+       struct panfrost_file_priv *priv = file_priv->driver_priv;
 
        if (bo->is_mapped)
                panfrost_mmu_unmap(bo);
 
-       spin_lock(&pfdev->mm_lock);
+       spin_lock(&priv->mm_lock);
        if (drm_mm_node_allocated(&bo->node))
                drm_mm_remove_node(&bo->node);
-       spin_unlock(&pfdev->mm_lock);
+       spin_unlock(&priv->mm_lock);
 }
 
 static int panfrost_gem_pin(struct drm_gem_object *obj)
index e10f58316915eb85e67665004b42b2eafb4a2ad5..50920819cc1698fce981f9e4fdd104ab631a9d3f 100644 (file)
@@ -7,10 +7,13 @@
 #include <drm/drm_gem_shmem_helper.h>
 #include <drm/drm_mm.h>
 
+struct panfrost_mmu;
+
 struct panfrost_gem_object {
        struct drm_gem_shmem_object base;
        struct sg_table *sgts;
 
+       struct panfrost_mmu *mmu;
        struct drm_mm_node node;
        bool is_mapped          :1;
        bool noexec             :1;
index 0fc4539fd08d9264490f3a964c7047e8af0bad17..05c85f45a0dea9eb660c2fd506a4c5e44af109b7 100644 (file)
@@ -153,6 +153,8 @@ static void panfrost_job_hw_submit(struct panfrost_job *job, int js)
        if (WARN_ON(job_read(pfdev, JS_COMMAND_NEXT(js))))
                goto end;
 
+       cfg = panfrost_mmu_as_get(pfdev, &job->file_priv->mmu);
+
        panfrost_devfreq_record_transition(pfdev, js);
        spin_lock_irqsave(&pfdev->hwaccess_lock, flags);
 
@@ -163,8 +165,7 @@ static void panfrost_job_hw_submit(struct panfrost_job *job, int js)
 
        /* start MMU, medium priority, cache clean/flush on end, clean/flush on
         * start */
-       /* TODO: different address spaces */
-       cfg = JS_CONFIG_THREAD_PRI(8) |
+       cfg |= JS_CONFIG_THREAD_PRI(8) |
                JS_CONFIG_START_FLUSH_CLEAN_INVALIDATE |
                JS_CONFIG_END_FLUSH_CLEAN_INVALIDATE;
 
@@ -377,8 +378,9 @@ static void panfrost_job_timedout(struct drm_sched_job *sched_job)
        if (dma_fence_is_signaled(job->done_fence))
                return;
 
-       dev_err(pfdev->dev, "gpu sched timeout, js=%d, status=0x%x, head=0x%x, tail=0x%x, sched_job=%p",
+       dev_err(pfdev->dev, "gpu sched timeout, js=%d, config=0x%x, status=0x%x, head=0x%x, tail=0x%x, sched_job=%p",
                js,
+               job_read(pfdev, JS_CONFIG(js)),
                job_read(pfdev, JS_STATUS(js)),
                job_read(pfdev, JS_HEAD_LO(js)),
                job_read(pfdev, JS_TAIL_LO(js)),
@@ -448,8 +450,12 @@ static irqreturn_t panfrost_job_irq_handler(int irq, void *data)
                }
 
                if (status & JOB_INT_MASK_DONE(j)) {
+                       struct panfrost_job *job = pfdev->jobs[j];
+
+                       pfdev->jobs[j] = NULL;
+                       panfrost_mmu_as_put(pfdev, &job->file_priv->mmu);
                        panfrost_devfreq_record_transition(pfdev, j);
-                       dma_fence_signal(pfdev->jobs[j]->done_fence);
+                       dma_fence_signal(job->done_fence);
                }
 
                status &= ~mask;
index 06f1a563e9408d5555770cd2a4de2c3c5804f939..842bdd7cf6bed69ae49d9676d03b093f4b645361 100644 (file)
@@ -1,5 +1,6 @@
 // SPDX-License-Identifier:    GPL-2.0
 /* Copyright 2019 Linaro, Ltd, Rob Herring <robh@kernel.org> */
+#include <linux/atomic.h>
 #include <linux/bitfield.h>
 #include <linux/delay.h>
 #include <linux/dma-mapping.h>
 #define mmu_write(dev, reg, data) writel(data, dev->iomem + reg)
 #define mmu_read(dev, reg) readl(dev->iomem + reg)
 
-struct panfrost_mmu {
-       struct io_pgtable_cfg pgtbl_cfg;
-       struct io_pgtable_ops *pgtbl_ops;
-       struct mutex lock;
-};
-
 static int wait_ready(struct panfrost_device *pfdev, u32 as_nr)
 {
        int ret;
@@ -85,13 +80,19 @@ static void lock_region(struct panfrost_device *pfdev, u32 as_nr,
 }
 
 
-static int mmu_hw_do_operation(struct panfrost_device *pfdev, u32 as_nr,
-               u64 iova, size_t size, u32 op)
+static int mmu_hw_do_operation(struct panfrost_device *pfdev,
+                              struct panfrost_mmu *mmu,
+                              u64 iova, size_t size, u32 op)
 {
-       unsigned long flags;
-       int ret;
+       int ret, as_nr;
 
-       spin_lock_irqsave(&pfdev->hwaccess_lock, flags);
+       spin_lock(&pfdev->as_lock);
+       as_nr = mmu->as;
+
+       if (as_nr < 0) {
+               spin_unlock(&pfdev->as_lock);
+               return 0;
+       }
 
        if (op != AS_COMMAND_UNLOCK)
                lock_region(pfdev, as_nr, iova, size);
@@ -102,14 +103,15 @@ static int mmu_hw_do_operation(struct panfrost_device *pfdev, u32 as_nr,
        /* Wait for the flush to complete */
        ret = wait_ready(pfdev, as_nr);
 
-       spin_unlock_irqrestore(&pfdev->hwaccess_lock, flags);
+       spin_unlock(&pfdev->as_lock);
 
        return ret;
 }
 
-static void panfrost_mmu_enable(struct panfrost_device *pfdev, u32 as_nr)
+static void panfrost_mmu_enable(struct panfrost_device *pfdev, struct panfrost_mmu *mmu)
 {
-       struct io_pgtable_cfg *cfg = &pfdev->mmu->pgtbl_cfg;
+       int as_nr = mmu->as;
+       struct io_pgtable_cfg *cfg = &mmu->pgtbl_cfg;
        u64 transtab = cfg->arm_mali_lpae_cfg.transtab;
        u64 memattr = cfg->arm_mali_lpae_cfg.memattr;
 
@@ -136,9 +138,75 @@ static void mmu_disable(struct panfrost_device *pfdev, u32 as_nr)
        write_cmd(pfdev, as_nr, AS_COMMAND_UPDATE);
 }
 
+u32 panfrost_mmu_as_get(struct panfrost_device *pfdev, struct panfrost_mmu *mmu)
+{
+       int as;
+
+       spin_lock(&pfdev->as_lock);
+
+       as = mmu->as;
+       if (as >= 0) {
+               int en = atomic_inc_return(&mmu->as_count);
+               WARN_ON(en >= NUM_JOB_SLOTS);
+
+               list_move(&mmu->list, &pfdev->as_lru_list);
+               goto out;
+       }
+
+       /* Check for a free AS */
+       as = ffz(pfdev->as_alloc_mask);
+       if (!(BIT(as) & pfdev->features.as_present)) {
+               struct panfrost_mmu *lru_mmu;
+
+               list_for_each_entry_reverse(lru_mmu, &pfdev->as_lru_list, list) {
+                       if (!atomic_read(&lru_mmu->as_count))
+                               break;
+               }
+               WARN_ON(&lru_mmu->list == &pfdev->as_lru_list);
+
+               list_del_init(&lru_mmu->list);
+               as = lru_mmu->as;
+
+               WARN_ON(as < 0);
+               lru_mmu->as = -1;
+       }
+
+       /* Assign the free or reclaimed AS to the FD */
+       mmu->as = as;
+       set_bit(as, &pfdev->as_alloc_mask);
+       atomic_set(&mmu->as_count, 1);
+       list_add(&mmu->list, &pfdev->as_lru_list);
+
+       dev_dbg(pfdev->dev, "Assigned AS%d to mmu %p, alloc_mask=%lx", as, mmu, pfdev->as_alloc_mask);
+
+       panfrost_mmu_enable(pfdev, mmu);
+
+out:
+       spin_unlock(&pfdev->as_lock);
+       return as;
+}
+
+void panfrost_mmu_as_put(struct panfrost_device *pfdev, struct panfrost_mmu *mmu)
+{
+       atomic_dec(&mmu->as_count);
+       WARN_ON(atomic_read(&mmu->as_count) < 0);
+}
+
 void panfrost_mmu_reset(struct panfrost_device *pfdev)
 {
-       panfrost_mmu_enable(pfdev, 0);
+       struct panfrost_mmu *mmu, *mmu_tmp;
+
+       spin_lock(&pfdev->as_lock);
+
+       pfdev->as_alloc_mask = 0;
+
+       list_for_each_entry_safe(mmu, mmu_tmp, &pfdev->as_lru_list, list) {
+               mmu->as = -1;
+               atomic_set(&mmu->as_count, 0);
+               list_del_init(&mmu->list);
+       }
+
+       spin_unlock(&pfdev->as_lock);
 
        mmu_write(pfdev, MMU_INT_CLEAR, ~0);
        mmu_write(pfdev, MMU_INT_MASK, ~0);
@@ -152,21 +220,21 @@ static size_t get_pgsize(u64 addr, size_t size)
        return SZ_2M;
 }
 
-static int mmu_map_sg(struct panfrost_device *pfdev, u64 iova,
-                     int prot, struct sg_table *sgt)
+static int mmu_map_sg(struct panfrost_device *pfdev, struct panfrost_mmu *mmu,
+                     u64 iova, int prot, struct sg_table *sgt)
 {
        unsigned int count;
        struct scatterlist *sgl;
-       struct io_pgtable_ops *ops = pfdev->mmu->pgtbl_ops;
+       struct io_pgtable_ops *ops = mmu->pgtbl_ops;
        u64 start_iova = iova;
 
-       mutex_lock(&pfdev->mmu->lock);
+       mutex_lock(&mmu->lock);
 
        for_each_sg(sgt->sgl, sgl, sgt->nents, count) {
                unsigned long paddr = sg_dma_address(sgl);
                size_t len = sg_dma_len(sgl);
 
-               dev_dbg(pfdev->dev, "map: iova=%llx, paddr=%lx, len=%zx", iova, paddr, len);
+               dev_dbg(pfdev->dev, "map: as=%d, iova=%llx, paddr=%lx, len=%zx", mmu->as, iova, paddr, len);
 
                while (len) {
                        size_t pgsize = get_pgsize(iova | paddr, len);
@@ -178,10 +246,10 @@ static int mmu_map_sg(struct panfrost_device *pfdev, u64 iova,
                }
        }
 
-       mmu_hw_do_operation(pfdev, 0, start_iova, iova - start_iova,
+       mmu_hw_do_operation(pfdev, mmu, start_iova, iova - start_iova,
                            AS_COMMAND_FLUSH_PT);
 
-       mutex_unlock(&pfdev->mmu->lock);
+       mutex_unlock(&mmu->lock);
 
        return 0;
 }
@@ -208,7 +276,7 @@ int panfrost_mmu_map(struct panfrost_gem_object *bo)
        if (ret < 0)
                return ret;
 
-       mmu_map_sg(pfdev, bo->node.start << PAGE_SHIFT, prot, sgt);
+       mmu_map_sg(pfdev, bo->mmu, bo->node.start << PAGE_SHIFT, prot, sgt);
 
        pm_runtime_mark_last_busy(pfdev->dev);
        pm_runtime_put_autosuspend(pfdev->dev);
@@ -221,7 +289,7 @@ void panfrost_mmu_unmap(struct panfrost_gem_object *bo)
 {
        struct drm_gem_object *obj = &bo->base.base;
        struct panfrost_device *pfdev = to_panfrost_device(obj->dev);
-       struct io_pgtable_ops *ops = pfdev->mmu->pgtbl_ops;
+       struct io_pgtable_ops *ops = bo->mmu->pgtbl_ops;
        u64 iova = bo->node.start << PAGE_SHIFT;
        size_t len = bo->node.size << PAGE_SHIFT;
        size_t unmapped_len = 0;
@@ -230,13 +298,13 @@ void panfrost_mmu_unmap(struct panfrost_gem_object *bo)
        if (WARN_ON(!bo->is_mapped))
                return;
 
-       dev_dbg(pfdev->dev, "unmap: iova=%llx, len=%zx", iova, len);
+       dev_dbg(pfdev->dev, "unmap: as=%d, iova=%llx, len=%zx", bo->mmu->as, iova, len);
 
        ret = pm_runtime_get_sync(pfdev->dev);
        if (ret < 0)
                return;
 
-       mutex_lock(&pfdev->mmu->lock);
+       mutex_lock(&bo->mmu->lock);
 
        while (unmapped_len < len) {
                size_t unmapped_page;
@@ -250,10 +318,10 @@ void panfrost_mmu_unmap(struct panfrost_gem_object *bo)
                unmapped_len += pgsize;
        }
 
-       mmu_hw_do_operation(pfdev, 0, bo->node.start << PAGE_SHIFT,
+       mmu_hw_do_operation(pfdev, bo->mmu, bo->node.start << PAGE_SHIFT,
                            bo->node.size << PAGE_SHIFT, AS_COMMAND_FLUSH_PT);
 
-       mutex_unlock(&pfdev->mmu->lock);
+       mutex_unlock(&bo->mmu->lock);
 
        pm_runtime_mark_last_busy(pfdev->dev);
        pm_runtime_put_autosuspend(pfdev->dev);
@@ -262,9 +330,9 @@ void panfrost_mmu_unmap(struct panfrost_gem_object *bo)
 
 static void mmu_tlb_inv_context_s1(void *cookie)
 {
-       struct panfrost_device *pfdev = cookie;
+       struct panfrost_file_priv *priv = cookie;
 
-       mmu_hw_do_operation(pfdev, 0, 0, ~0UL, AS_COMMAND_FLUSH_MEM);
+       mmu_hw_do_operation(priv->pfdev, &priv->mmu, 0, ~0UL, AS_COMMAND_FLUSH_MEM);
 }
 
 static void mmu_tlb_inv_range_nosync(unsigned long iova, size_t size,
@@ -283,16 +351,69 @@ static const struct iommu_gather_ops mmu_tlb_ops = {
        .tlb_sync       = mmu_tlb_sync_context,
 };
 
+int panfrost_mmu_pgtable_alloc(struct panfrost_file_priv *priv)
+{
+       struct panfrost_mmu *mmu = &priv->mmu;
+       struct panfrost_device *pfdev = priv->pfdev;
+
+       mutex_init(&mmu->lock);
+       INIT_LIST_HEAD(&mmu->list);
+       mmu->as = -1;
+
+       mmu->pgtbl_cfg = (struct io_pgtable_cfg) {
+               .pgsize_bitmap  = SZ_4K | SZ_2M,
+               .ias            = FIELD_GET(0xff, pfdev->features.mmu_features),
+               .oas            = FIELD_GET(0xff00, pfdev->features.mmu_features),
+               .tlb            = &mmu_tlb_ops,
+               .iommu_dev      = pfdev->dev,
+       };
+
+       mmu->pgtbl_ops = alloc_io_pgtable_ops(ARM_MALI_LPAE, &mmu->pgtbl_cfg,
+                                             priv);
+       if (!mmu->pgtbl_ops)
+               return -EINVAL;
+
+       return 0;
+}
+
+void panfrost_mmu_pgtable_free(struct panfrost_file_priv *priv)
+{
+       struct panfrost_device *pfdev = priv->pfdev;
+       struct panfrost_mmu *mmu = &priv->mmu;
+
+       spin_lock(&pfdev->as_lock);
+       if (mmu->as >= 0) {
+               clear_bit(mmu->as, &pfdev->as_alloc_mask);
+               clear_bit(mmu->as, &pfdev->as_in_use_mask);
+               list_del(&mmu->list);
+       }
+       spin_unlock(&pfdev->as_lock);
+
+       free_io_pgtable_ops(mmu->pgtbl_ops);
+}
+
 static struct drm_mm_node *addr_to_drm_mm_node(struct panfrost_device *pfdev, int as, u64 addr)
 {
-       struct drm_mm_node *node;
+       struct drm_mm_node *node = NULL;
        u64 offset = addr >> PAGE_SHIFT;
+       struct panfrost_mmu *mmu;
 
-       drm_mm_for_each_node(node, &pfdev->mm) {
-               if (offset >= node->start && offset < (node->start + node->size))
-                       return node;
+       spin_lock(&pfdev->as_lock);
+       list_for_each_entry(mmu, &pfdev->as_lru_list, list) {
+               struct panfrost_file_priv *priv;
+               if (as != mmu->as)
+                       continue;
+
+               priv = container_of(mmu, struct panfrost_file_priv, mmu);
+               drm_mm_for_each_node(node, &priv->mm) {
+                       if (offset >= node->start && offset < (node->start + node->size))
+                               goto out;
+               }
        }
-       return NULL;
+
+out:
+       spin_unlock(&pfdev->as_lock);
+       return node;
 }
 
 #define NUM_FAULT_PAGES (SZ_2M / PAGE_SIZE)
@@ -317,6 +438,8 @@ int panfrost_mmu_map_fault_addr(struct panfrost_device *pfdev, int as, u64 addr)
                         node->start << PAGE_SHIFT);
                return -EINVAL;
        }
+       WARN_ON(bo->mmu->as != as);
+
        /* Assume 2MB alignment and size multiple */
        addr &= ~((u64)SZ_2M - 1);
        page_offset = addr >> PAGE_SHIFT;
@@ -370,11 +493,11 @@ int panfrost_mmu_map_fault_addr(struct panfrost_device *pfdev, int as, u64 addr)
                goto err_map;
        }
 
-       mmu_map_sg(pfdev, addr, IOMMU_WRITE | IOMMU_READ | IOMMU_NOEXEC, sgt);
+       mmu_map_sg(pfdev, bo->mmu, addr, IOMMU_WRITE | IOMMU_READ | IOMMU_NOEXEC, sgt);
 
        bo->is_mapped = true;
 
-       dev_dbg(pfdev->dev, "mapped page fault @ %llx", addr);
+       dev_dbg(pfdev->dev, "mapped page fault @ AS%d %llx", as, addr);
 
        return 0;
 
@@ -483,15 +606,8 @@ static irqreturn_t panfrost_mmu_irq_handler_thread(int irq, void *data)
 
 int panfrost_mmu_init(struct panfrost_device *pfdev)
 {
-       struct io_pgtable_ops *pgtbl_ops;
        int err, irq;
 
-       pfdev->mmu = devm_kzalloc(pfdev->dev, sizeof(*pfdev->mmu), GFP_KERNEL);
-       if (!pfdev->mmu)
-               return -ENOMEM;
-
-       mutex_init(&pfdev->mmu->lock);
-
        irq = platform_get_irq_byname(to_platform_device(pfdev->dev), "mmu");
        if (irq <= 0)
                return -ENODEV;
@@ -504,22 +620,6 @@ int panfrost_mmu_init(struct panfrost_device *pfdev)
                dev_err(pfdev->dev, "failed to request mmu irq");
                return err;
        }
-       pfdev->mmu->pgtbl_cfg = (struct io_pgtable_cfg) {
-               .pgsize_bitmap  = SZ_4K | SZ_2M,
-               .ias            = FIELD_GET(0xff, pfdev->features.mmu_features),
-               .oas            = FIELD_GET(0xff00, pfdev->features.mmu_features),
-               .tlb            = &mmu_tlb_ops,
-               .iommu_dev      = pfdev->dev,
-       };
-
-       pgtbl_ops = alloc_io_pgtable_ops(ARM_MALI_LPAE, &pfdev->mmu->pgtbl_cfg,
-                                        pfdev);
-       if (!pgtbl_ops)
-               return -ENOMEM;
-
-       pfdev->mmu->pgtbl_ops = pgtbl_ops;
-
-       panfrost_mmu_enable(pfdev, 0);
 
        return 0;
 }
@@ -528,6 +628,4 @@ void panfrost_mmu_fini(struct panfrost_device *pfdev)
 {
        mmu_write(pfdev, MMU_INT_MASK, 0);
        mmu_disable(pfdev, 0);
-
-       free_io_pgtable_ops(pfdev->mmu->pgtbl_ops);
 }
index d5f9b24537db36b09706426b51c5661c1cf36e9d..7c5b6775ae230cc77f576c978667bd82d011f199 100644 (file)
@@ -5,6 +5,8 @@
 #define __PANFROST_MMU_H__
 
 struct panfrost_gem_object;
+struct panfrost_file_priv;
+struct panfrost_mmu;
 
 int panfrost_mmu_map(struct panfrost_gem_object *bo);
 void panfrost_mmu_unmap(struct panfrost_gem_object *bo);
@@ -13,4 +15,10 @@ int panfrost_mmu_init(struct panfrost_device *pfdev);
 void panfrost_mmu_fini(struct panfrost_device *pfdev);
 void panfrost_mmu_reset(struct panfrost_device *pfdev);
 
+u32 panfrost_mmu_as_get(struct panfrost_device *pfdev, struct panfrost_mmu *mmu);
+void panfrost_mmu_as_put(struct panfrost_device *pfdev, struct panfrost_mmu *mmu);
+
+int panfrost_mmu_pgtable_alloc(struct panfrost_file_priv *priv);
+void panfrost_mmu_pgtable_free(struct panfrost_file_priv *priv);
+
 #endif