12875fe05SAtish Patra // SPDX-License-Identifier: GPL-2.0-only 22875fe05SAtish Patra /* 32875fe05SAtish Patra * Copyright (c) 2020 Western Digital Corporation or its affiliates. 42875fe05SAtish Patra */ 52875fe05SAtish Patra 62875fe05SAtish Patra #include <linux/errno.h> 72875fe05SAtish Patra #include <linux/mm.h> 82875fe05SAtish Patra #include <linux/of.h> 92875fe05SAtish Patra #include <linux/string.h> 102875fe05SAtish Patra #include <linux/sched.h> 112875fe05SAtish Patra #include <asm/cpu_ops.h> 122875fe05SAtish Patra #include <asm/sbi.h> 132875fe05SAtish Patra #include <asm/smp.h> 142875fe05SAtish Patra 152875fe05SAtish Patra const struct cpu_operations *cpu_ops[NR_CPUS] __ro_after_init; 162875fe05SAtish Patra 17cfafe260SAtish Patra extern const struct cpu_operations cpu_ops_sbi; 18*2ffc48fcSAtish Patra #ifdef CONFIG_RISCV_BOOT_SPINWAIT 192875fe05SAtish Patra extern const struct cpu_operations cpu_ops_spinwait; 20*2ffc48fcSAtish Patra #else 21*2ffc48fcSAtish Patra const struct cpu_operations cpu_ops_spinwait = { 22*2ffc48fcSAtish Patra .name = "", 23*2ffc48fcSAtish Patra .cpu_prepare = NULL, 24*2ffc48fcSAtish Patra .cpu_start = NULL, 25*2ffc48fcSAtish Patra }; 26*2ffc48fcSAtish Patra #endif 272875fe05SAtish Patra 282875fe05SAtish Patra void __init cpu_set_ops(int cpuid) 292875fe05SAtish Patra { 30cfafe260SAtish Patra #if IS_ENABLED(CONFIG_RISCV_SBI) 31cfafe260SAtish Patra if (sbi_probe_extension(SBI_EXT_HSM) > 0) { 32cfafe260SAtish Patra if (!cpuid) 33410bb20aSAtish Patra pr_info("SBI HSM extension detected\n"); 34cfafe260SAtish Patra cpu_ops[cpuid] = &cpu_ops_sbi; 35cfafe260SAtish Patra } else 36cfafe260SAtish Patra #endif 372875fe05SAtish Patra cpu_ops[cpuid] = &cpu_ops_spinwait; 382875fe05SAtish Patra } 39