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*da6d2128SBen Dooks #ifndef CONFIG_RISCV_BOOT_SPINWAIT 192ffc48fcSAtish Patra const struct cpu_operations cpu_ops_spinwait = { 202ffc48fcSAtish Patra .name = "", 212ffc48fcSAtish Patra .cpu_prepare = NULL, 222ffc48fcSAtish Patra .cpu_start = NULL, 232ffc48fcSAtish Patra }; 242ffc48fcSAtish Patra #endif 252875fe05SAtish Patra 262875fe05SAtish Patra void __init cpu_set_ops(int cpuid) 272875fe05SAtish Patra { 28cfafe260SAtish Patra #if IS_ENABLED(CONFIG_RISCV_SBI) 29cfafe260SAtish Patra if (sbi_probe_extension(SBI_EXT_HSM) > 0) { 30cfafe260SAtish Patra if (!cpuid) 31410bb20aSAtish Patra pr_info("SBI HSM extension detected\n"); 32cfafe260SAtish Patra cpu_ops[cpuid] = &cpu_ops_sbi; 33cfafe260SAtish Patra } else 34cfafe260SAtish Patra #endif 352875fe05SAtish Patra cpu_ops[cpuid] = &cpu_ops_spinwait; 362875fe05SAtish Patra } 37