diff options
Diffstat (limited to 'lib/sbi/sbi_scratch.c')
-rw-r--r-- | lib/sbi/sbi_scratch.c | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/lib/sbi/sbi_scratch.c b/lib/sbi/sbi_scratch.c index 8b63570..96bae5b 100644 --- a/lib/sbi/sbi_scratch.c +++ b/lib/sbi/sbi_scratch.c @@ -14,6 +14,7 @@ #include <sbi/sbi_scratch.h> #include <sbi/sbi_string.h> +u32 last_hartid_having_scratch = SBI_HARTMASK_MAX_BITS; struct sbi_scratch *hartid_to_scratch_table[SBI_HARTMASK_MAX_BITS] = { 0 }; static spinlock_t extra_lock = SPIN_LOCK_INITIALIZER; @@ -32,6 +33,8 @@ int sbi_scratch_init(struct sbi_scratch *scratch) hartid_to_scratch_table[i] = ((hartid2scratch)scratch->hartid_to_scratch)(i, sbi_platform_hart_index(plat, i)); + if (hartid_to_scratch_table[i]) + last_hartid_having_scratch = i; } return 0; @@ -71,7 +74,7 @@ done: spin_unlock(&extra_lock); if (ret) { - for (i = 0; i < SBI_HARTMASK_MAX_BITS; i++) { + for (i = 0; i < sbi_scratch_last_hartid(); i++) { rscratch = sbi_hartid_to_scratch(i); if (!rscratch) continue; |