diff mbox series

[v4,09/16] lib: sbi: Optimize sbi_hsm_hart_started_mask() implementation

Message ID 20201020093930.2551352-10-anup.patel@wdc.com
State Accepted
Headers show
Series OpenSBI domain support | expand

Commit Message

Anup Patel Oct. 20, 2020, 9:39 a.m. UTC
Instead of calling sbi_hsm_hart_get_state() in a loop, we can simply
call a new inline __sbi_hsm_hart_get_state() which only takes "hartid"
and enforce domain checks using sbi_domain_assigned_hartmask().

This patch optimizes sbi_hsm_hart_started_mask() as-per above.

Signed-off-by: Anup Patel <anup.patel@wdc.com>
Reviewed-by: Atish Patra <atish.patra@wdc.com>
---
 lib/sbi/sbi_hsm.c | 34 ++++++++++++++++++++++------------
 1 file changed, 22 insertions(+), 12 deletions(-)

Comments

Anup Patel Oct. 21, 2020, 7:06 a.m. UTC | #1
> -----Original Message-----
> From: Anup Patel <Anup.Patel@wdc.com>
> Sent: 20 October 2020 15:09
> To: Atish Patra <Atish.Patra@wdc.com>; Alistair Francis
> <Alistair.Francis@wdc.com>
> Cc: Anup Patel <anup@brainfault.org>; opensbi@lists.infradead.org; Anup
> Patel <Anup.Patel@wdc.com>
> Subject: [PATCH v4 09/16] lib: sbi: Optimize sbi_hsm_hart_started_mask()
> implementation
> 
> Instead of calling sbi_hsm_hart_get_state() in a loop, we can simply call a
> new inline __sbi_hsm_hart_get_state() which only takes "hartid"
> and enforce domain checks using sbi_domain_assigned_hartmask().
> 
> This patch optimizes sbi_hsm_hart_started_mask() as-per above.
> 
> Signed-off-by: Anup Patel <anup.patel@wdc.com>
> Reviewed-by: Atish Patra <atish.patra@wdc.com>
> ---
>  lib/sbi/sbi_hsm.c | 34 ++++++++++++++++++++++------------
>  1 file changed, 22 insertions(+), 12 deletions(-)
> 
> diff --git a/lib/sbi/sbi_hsm.c b/lib/sbi/sbi_hsm.c index 8121efb..e1b2b2c
> 100644
> --- a/lib/sbi/sbi_hsm.c
> +++ b/lib/sbi/sbi_hsm.c
> @@ -57,20 +57,27 @@ int sbi_hsm_hart_state_to_status(int state)
>  	return ret;
>  }
> 
> -int sbi_hsm_hart_get_state(const struct sbi_domain *dom, u32 hartid)
> +static inline int __sbi_hsm_hart_get_state(u32 hartid)
>  {
>  	struct sbi_hsm_data *hdata;
>  	struct sbi_scratch *scratch;
> 
>  	scratch = sbi_hartid_to_scratch(hartid);
> -	if (!scratch || !sbi_domain_is_assigned_hart(dom, hartid))
> +	if (!scratch)
>  		return SBI_HART_UNKNOWN;
> 
>  	hdata = sbi_scratch_offset_ptr(scratch, hart_data_offset);
> -
>  	return atomic_read(&hdata->state);
>  }
> 
> +int sbi_hsm_hart_get_state(const struct sbi_domain *dom, u32 hartid) {
> +	if (!sbi_domain_is_assigned_hart(dom, hartid))
> +		return SBI_HART_UNKNOWN;
> +
> +	return __sbi_hsm_hart_get_state(hartid); }
> +
>  static bool sbi_hsm_hart_started(const struct sbi_domain *dom, u32 hartid)
> {
>  	if (sbi_hsm_hart_get_state(dom, hartid) == SBI_HART_STARTED)
> @@ -90,18 +97,21 @@ static bool sbi_hsm_hart_started(const struct
> sbi_domain *dom, u32 hartid)  int sbi_hsm_hart_started_mask(const struct
> sbi_domain *dom,
>  			      ulong hbase, ulong *out_hmask)  {
> -	ulong i;
> -	ulong hcount = sbi_scratch_last_hartid() + 1;
> +	ulong i, hmask, dmask;
> +	ulong hend = sbi_scratch_last_hartid() + 1;
> 
>  	*out_hmask = 0;
> -	if (hcount <= hbase)
> +	if (hend <= hbase)
>  		return SBI_EINVAL;
> -	if (BITS_PER_LONG < (hcount - hbase))
> -		hcount = BITS_PER_LONG;
> -
> -	for (i = hbase; i < hcount; i++) {
> -		if (sbi_hsm_hart_get_state(dom, i) == SBI_HART_STARTED)
> -			*out_hmask |= 1UL << (i - hbase);
> +	if (BITS_PER_LONG < (hend - hbase))
> +		hend = hbase + BITS_PER_LONG;
> +
> +	dmask = sbi_domain_get_assigned_hartmask(dom, hbase);
> +	for (i = hbase; i < hend; i++) {
> +		hmask = 1UL << (i - hbase);
> +		if ((dmask & hmask) &&
> +		    (__sbi_hsm_hart_get_state(i) == SBI_HART_STARTED))
> +			*out_hmask |= hmask;
>  	}
> 
>  	return 0;
> --
> 2.25.1

Applied this patch to the riscv/opensbi repo

Regards,
Anup
diff mbox series

Patch

diff --git a/lib/sbi/sbi_hsm.c b/lib/sbi/sbi_hsm.c
index 8121efb..e1b2b2c 100644
--- a/lib/sbi/sbi_hsm.c
+++ b/lib/sbi/sbi_hsm.c
@@ -57,20 +57,27 @@  int sbi_hsm_hart_state_to_status(int state)
 	return ret;
 }
 
-int sbi_hsm_hart_get_state(const struct sbi_domain *dom, u32 hartid)
+static inline int __sbi_hsm_hart_get_state(u32 hartid)
 {
 	struct sbi_hsm_data *hdata;
 	struct sbi_scratch *scratch;
 
 	scratch = sbi_hartid_to_scratch(hartid);
-	if (!scratch || !sbi_domain_is_assigned_hart(dom, hartid))
+	if (!scratch)
 		return SBI_HART_UNKNOWN;
 
 	hdata = sbi_scratch_offset_ptr(scratch, hart_data_offset);
-
 	return atomic_read(&hdata->state);
 }
 
+int sbi_hsm_hart_get_state(const struct sbi_domain *dom, u32 hartid)
+{
+	if (!sbi_domain_is_assigned_hart(dom, hartid))
+		return SBI_HART_UNKNOWN;
+
+	return __sbi_hsm_hart_get_state(hartid);
+}
+
 static bool sbi_hsm_hart_started(const struct sbi_domain *dom, u32 hartid)
 {
 	if (sbi_hsm_hart_get_state(dom, hartid) == SBI_HART_STARTED)
@@ -90,18 +97,21 @@  static bool sbi_hsm_hart_started(const struct sbi_domain *dom, u32 hartid)
 int sbi_hsm_hart_started_mask(const struct sbi_domain *dom,
 			      ulong hbase, ulong *out_hmask)
 {
-	ulong i;
-	ulong hcount = sbi_scratch_last_hartid() + 1;
+	ulong i, hmask, dmask;
+	ulong hend = sbi_scratch_last_hartid() + 1;
 
 	*out_hmask = 0;
-	if (hcount <= hbase)
+	if (hend <= hbase)
 		return SBI_EINVAL;
-	if (BITS_PER_LONG < (hcount - hbase))
-		hcount = BITS_PER_LONG;
-
-	for (i = hbase; i < hcount; i++) {
-		if (sbi_hsm_hart_get_state(dom, i) == SBI_HART_STARTED)
-			*out_hmask |= 1UL << (i - hbase);
+	if (BITS_PER_LONG < (hend - hbase))
+		hend = hbase + BITS_PER_LONG;
+
+	dmask = sbi_domain_get_assigned_hartmask(dom, hbase);
+	for (i = hbase; i < hend; i++) {
+		hmask = 1UL << (i - hbase);
+		if ((dmask & hmask) &&
+		    (__sbi_hsm_hart_get_state(i) == SBI_HART_STARTED))
+			*out_hmask |= hmask;
 	}
 
 	return 0;