12345678910111213141516171819202122232425262728293031323334353637383940414243444546 |
- // SPDX-License-Identifier: GPL-2.0-only
- /*
- * Copyright (c) 2020 Western Digital Corporation or its affiliates.
- */
- #include <linux/errno.h>
- #include <linux/mm.h>
- #include <linux/of.h>
- #include <linux/string.h>
- #include <linux/sched.h>
- #include <linux/sched/task_stack.h>
- #include <asm/cpu_ops.h>
- #include <asm/sbi.h>
- #include <asm/smp.h>
- const struct cpu_operations *cpu_ops[NR_CPUS] __ro_after_init;
- void *__cpu_up_stack_pointer[NR_CPUS] __section(".data");
- void *__cpu_up_task_pointer[NR_CPUS] __section(".data");
- extern const struct cpu_operations cpu_ops_sbi;
- extern const struct cpu_operations cpu_ops_spinwait;
- void cpu_update_secondary_bootdata(unsigned int cpuid,
- struct task_struct *tidle)
- {
- int hartid = cpuid_to_hartid_map(cpuid);
- /* Make sure tidle is updated */
- smp_mb();
- WRITE_ONCE(__cpu_up_stack_pointer[hartid],
- task_stack_page(tidle) + THREAD_SIZE);
- WRITE_ONCE(__cpu_up_task_pointer[hartid], tidle);
- }
- void __init cpu_set_ops(int cpuid)
- {
- #if IS_ENABLED(CONFIG_RISCV_SBI)
- if (sbi_probe_extension(SBI_EXT_HSM) > 0) {
- if (!cpuid)
- pr_info("SBI v0.2 HSM extension detected\n");
- cpu_ops[cpuid] = &cpu_ops_sbi;
- } else
- #endif
- cpu_ops[cpuid] = &cpu_ops_spinwait;
- }
|