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> 12*87df2b5cSConor Dooley #include <asm/cpu_ops_sbi.h> 132875fe05SAtish Patra #include <asm/sbi.h> 142875fe05SAtish Patra #include <asm/smp.h> 152875fe05SAtish Patra 162875fe05SAtish Patra const struct cpu_operations *cpu_ops[NR_CPUS] __ro_after_init; 172875fe05SAtish Patra 18cfafe260SAtish Patra extern const struct cpu_operations cpu_ops_sbi; 19da6d2128SBen Dooks #ifndef CONFIG_RISCV_BOOT_SPINWAIT 202ffc48fcSAtish Patra const struct cpu_operations cpu_ops_spinwait = { 212ffc48fcSAtish Patra .name = "", 222ffc48fcSAtish Patra .cpu_prepare = NULL, 232ffc48fcSAtish Patra .cpu_start = NULL, 242ffc48fcSAtish Patra }; 252ffc48fcSAtish Patra #endif 262875fe05SAtish Patra 272875fe05SAtish Patra void __init cpu_set_ops(int cpuid) 282875fe05SAtish Patra { 29cfafe260SAtish Patra #if IS_ENABLED(CONFIG_RISCV_SBI) 30cfafe260SAtish Patra if (sbi_probe_extension(SBI_EXT_HSM) > 0) { 31cfafe260SAtish Patra if (!cpuid) 32410bb20aSAtish Patra pr_info("SBI HSM extension detected\n"); 33cfafe260SAtish Patra cpu_ops[cpuid] = &cpu_ops_sbi; 34cfafe260SAtish Patra } else 35cfafe260SAtish Patra #endif 362875fe05SAtish Patra cpu_ops[cpuid] = &cpu_ops_spinwait; 372875fe05SAtish Patra } 38