[RFC PATCH v2 58/58] iommu/arm-smmu-v3-kvm: Support command queue batching
Mostafa Saleh
smostafa at google.com
Thu Dec 12 10:04:22 PST 2024
Similar to the kernel driver, we can batch commands at EL2 to avoid
writing to MMIO space, this is quite noticable if the SMMU doesn't
support range invalidation so it has to invalidate page per page.
Signed-off-by: Mostafa Saleh <smostafa at google.com>
---
arch/arm64/include/asm/arm-smmu-v3-common.h | 16 ++++
arch/arm64/kvm/hyp/nvhe/iommu/arm-smmu-v3.c | 95 ++++++++++++++++-----
drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3.h | 15 ----
3 files changed, 88 insertions(+), 38 deletions(-)
diff --git a/arch/arm64/include/asm/arm-smmu-v3-common.h b/arch/arm64/include/asm/arm-smmu-v3-common.h
index f2fbd286f674..2578c8e9202e 100644
--- a/arch/arm64/include/asm/arm-smmu-v3-common.h
+++ b/arch/arm64/include/asm/arm-smmu-v3-common.h
@@ -573,4 +573,20 @@ struct arm_smmu_cmdq_ent {
};
};
+#define Q_OVERFLOW_FLAG (1U << 31)
+#define Q_OVF(p) ((p) & Q_OVERFLOW_FLAG)
+
+/*
+ * This is used to size the command queue and therefore must be at least
+ * BITS_PER_LONG so that the valid_map works correctly (it relies on the
+ * total number of queue entries being a multiple of BITS_PER_LONG).
+ */
+#define CMDQ_BATCH_ENTRIES BITS_PER_LONG
+
+struct arm_smmu_cmdq_batch {
+ u64 cmds[CMDQ_BATCH_ENTRIES * CMDQ_ENT_DWORDS];
+ struct arm_smmu_cmdq *cmdq;
+ int num;
+};
+
#endif /* _ARM_SMMU_V3_COMMON_H */
diff --git a/arch/arm64/kvm/hyp/nvhe/iommu/arm-smmu-v3.c b/arch/arm64/kvm/hyp/nvhe/iommu/arm-smmu-v3.c
index 60f0760f49eb..62760136c6fb 100644
--- a/arch/arm64/kvm/hyp/nvhe/iommu/arm-smmu-v3.c
+++ b/arch/arm64/kvm/hyp/nvhe/iommu/arm-smmu-v3.c
@@ -96,12 +96,20 @@ static void smmu_reclaim_pages(u64 phys, size_t size)
#define Q_WRAP(smmu, reg) ((reg) & (1 << (smmu)->cmdq_log2size))
#define Q_IDX(smmu, reg) ((reg) & ((1 << (smmu)->cmdq_log2size) - 1))
-static bool smmu_cmdq_full(struct hyp_arm_smmu_v3_device *smmu)
+static bool smmu_cmdq_has_space(struct hyp_arm_smmu_v3_device *smmu, u32 n)
{
- u64 cons = readl_relaxed(smmu->base + ARM_SMMU_CMDQ_CONS);
+ u64 smmu_cons = readl_relaxed(smmu->base + ARM_SMMU_CMDQ_CONS);
+ u32 space, prod, cons;
- return Q_IDX(smmu, smmu->cmdq_prod) == Q_IDX(smmu, cons) &&
- Q_WRAP(smmu, smmu->cmdq_prod) != Q_WRAP(smmu, cons);
+ prod = Q_IDX(smmu, smmu->cmdq_prod);
+ cons = Q_IDX(smmu, smmu_cons);
+
+ if (Q_WRAP(smmu, smmu->cmdq_prod) == Q_WRAP(smmu, smmu_cons))
+ space = (1 << smmu->cmdq_log2size) - (prod - cons);
+ else
+ space = cons - prod;
+
+ return space >= n;
}
static bool smmu_cmdq_empty(struct hyp_arm_smmu_v3_device *smmu)
@@ -112,22 +120,8 @@ static bool smmu_cmdq_empty(struct hyp_arm_smmu_v3_device *smmu)
Q_WRAP(smmu, smmu->cmdq_prod) == Q_WRAP(smmu, cons);
}
-static int smmu_add_cmd(struct hyp_arm_smmu_v3_device *smmu,
- struct arm_smmu_cmdq_ent *ent)
+static int smmu_build_cmd(u64 *cmd, struct arm_smmu_cmdq_ent *ent)
{
- int i;
- int ret;
- u64 cmd[CMDQ_ENT_DWORDS] = {};
- int idx = Q_IDX(smmu, smmu->cmdq_prod);
- u64 *slot = smmu->cmdq_base + idx * CMDQ_ENT_DWORDS;
-
- if (smmu->iommu.power_is_off)
- return -EPIPE;
-
- ret = smmu_wait_event(smmu, !smmu_cmdq_full(smmu));
- if (ret)
- return ret;
-
cmd[0] |= FIELD_PREP(CMDQ_0_OP, ent->opcode);
switch (ent->opcode) {
@@ -175,15 +169,49 @@ static int smmu_add_cmd(struct hyp_arm_smmu_v3_device *smmu,
return -EINVAL;
}
- for (i = 0; i < CMDQ_ENT_DWORDS; i++)
- slot[i] = cpu_to_le64(cmd[i]);
+ return 0;
+}
+
+static int smmu_issue_cmds(struct hyp_arm_smmu_v3_device *smmu,
+ u64 *cmds, int n)
+{
+ int idx = Q_IDX(smmu, smmu->cmdq_prod);
+ u64 *slot = smmu->cmdq_base + idx * CMDQ_ENT_DWORDS;
+ int i;
+ int ret;
+ u32 prod;
+
+ if (smmu->iommu.power_is_off)
+ return -EPIPE;
+
+ ret = smmu_wait_event(smmu, smmu_cmdq_has_space(smmu, n));
+ if (ret)
+ return ret;
+
+ for (i = 0; i < CMDQ_ENT_DWORDS * n; i++)
+ slot[i] = cpu_to_le64(cmds[i]);
+
+ prod = (Q_WRAP(smmu, smmu->cmdq_prod) | Q_IDX(smmu, smmu->cmdq_prod)) + n;
+ smmu->cmdq_prod = Q_OVF(smmu->cmdq_prod) | Q_WRAP(smmu, prod) | Q_IDX(smmu, prod);
- smmu->cmdq_prod++;
writel(Q_IDX(smmu, smmu->cmdq_prod) | Q_WRAP(smmu, smmu->cmdq_prod),
smmu->base + ARM_SMMU_CMDQ_PROD);
return 0;
}
+static int smmu_add_cmd(struct hyp_arm_smmu_v3_device *smmu,
+ struct arm_smmu_cmdq_ent *ent)
+{
+ u64 cmd[CMDQ_ENT_DWORDS] = {};
+ int ret;
+
+ ret = smmu_build_cmd(cmd, ent);
+ if (ret)
+ return ret;
+
+ return smmu_issue_cmds(smmu, cmd, 1);
+}
+
static int smmu_sync_cmd(struct hyp_arm_smmu_v3_device *smmu)
{
int ret;
@@ -685,6 +713,23 @@ static void smmu_tlb_flush_all(void *cookie)
kvm_iommu_unlock(&smmu->iommu);
}
+static void smmu_cmdq_batch_add(struct hyp_arm_smmu_v3_device *smmu,
+ struct arm_smmu_cmdq_batch *cmds,
+ struct arm_smmu_cmdq_ent *cmd)
+{
+ int index;
+
+ if (cmds->num == CMDQ_BATCH_ENTRIES) {
+ smmu_issue_cmds(smmu, cmds->cmds, cmds->num);
+ cmds->num = 0;
+ }
+
+ index = cmds->num * CMDQ_ENT_DWORDS;
+ smmu_build_cmd(&cmds->cmds[index], cmd);
+
+ cmds->num++;
+}
+
static int smmu_tlb_inv_range_smmu(struct hyp_arm_smmu_v3_device *smmu,
struct kvm_hyp_iommu_domain *domain,
struct arm_smmu_cmdq_ent *cmd,
@@ -694,6 +739,7 @@ static int smmu_tlb_inv_range_smmu(struct hyp_arm_smmu_v3_device *smmu,
unsigned long end = iova + size, num_pages = 0, tg = 0;
size_t inv_range = granule;
struct hyp_arm_smmu_v3_domain *smmu_domain = domain->priv;
+ struct arm_smmu_cmdq_batch cmds;
kvm_iommu_lock(&smmu->iommu);
if (smmu->iommu.power_is_off)
@@ -723,6 +769,8 @@ static int smmu_tlb_inv_range_smmu(struct hyp_arm_smmu_v3_device *smmu,
num_pages++;
}
+ cmds.num = 0;
+
while (iova < end) {
if (smmu->features & ARM_SMMU_FEAT_RANGE_INV) {
/*
@@ -749,11 +797,12 @@ static int smmu_tlb_inv_range_smmu(struct hyp_arm_smmu_v3_device *smmu,
num_pages -= num << scale;
}
cmd->tlbi.addr = iova;
- WARN_ON(smmu_add_cmd(smmu, cmd));
+ smmu_cmdq_batch_add(smmu, &cmds, cmd);
BUG_ON(iova + inv_range < iova);
iova += inv_range;
}
+ WARN_ON(smmu_issue_cmds(smmu, cmds.cmds, cmds.num));
ret = smmu_sync_cmd(smmu);
out_ret:
kvm_iommu_unlock(&smmu->iommu);
diff --git a/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3.h b/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3.h
index d91dfe55835d..18f878bb7f98 100644
--- a/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3.h
+++ b/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3.h
@@ -20,8 +20,6 @@ struct arm_smmu_device;
#define Q_IDX(llq, p) ((p) & ((1 << (llq)->max_n_shift) - 1))
#define Q_WRP(llq, p) ((p) & (1 << (llq)->max_n_shift))
-#define Q_OVERFLOW_FLAG (1U << 31)
-#define Q_OVF(p) ((p) & Q_OVERFLOW_FLAG)
#define Q_ENT(q, p) ((q)->base + \
Q_IDX(&((q)->llq), p) * \
(q)->ent_dwords)
@@ -35,13 +33,6 @@ struct arm_smmu_device;
#define CMDQ_PROD_OWNED_FLAG Q_OVERFLOW_FLAG
-/*
- * This is used to size the command queue and therefore must be at least
- * BITS_PER_LONG so that the valid_map works correctly (it relies on the
- * total number of queue entries being a multiple of BITS_PER_LONG).
- */
-#define CMDQ_BATCH_ENTRIES BITS_PER_LONG
-
/* High-level queue structures */
#define ARM_SMMU_POLL_TIMEOUT_US 1000000 /* 1s! */
#define ARM_SMMU_POLL_SPIN_COUNT 10
@@ -100,12 +91,6 @@ static inline bool arm_smmu_cmdq_supports_cmd(struct arm_smmu_cmdq *cmdq,
return cmdq->supports_cmd ? cmdq->supports_cmd(ent) : true;
}
-struct arm_smmu_cmdq_batch {
- u64 cmds[CMDQ_BATCH_ENTRIES * CMDQ_ENT_DWORDS];
- struct arm_smmu_cmdq *cmdq;
- int num;
-};
-
struct arm_smmu_evtq {
struct arm_smmu_queue q;
struct iopf_queue *iopf;
--
2.47.0.338.g60cca15819-goog
More information about the linux-arm-kernel
mailing list