mirror of https://gitee.com/openkylin/linux.git
47 lines
1.2 KiB
C
47 lines
1.2 KiB
C
// 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;
|
|
}
|