arch/riscv/cpu/cpu.c

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * Copyright (C) 2018, Bin Meng <bmeng.cn@gmail.com>
   4  */
   5
   6 #include <common.h>
   7 #include <cpu.h>
   8 #include <dm.h>
   9 #include <init.h>
  10 #include <log.h>
  11 #include <asm/encoding.h>
  12 #include <dm/uclass-internal.h>
  13 #include <linux/bitops.h>
  14
  15 /*
  16  * The variables here must be stored in the data section since they are used
  17  * before the bss section is available.
  18  */
  19 #ifdef CONFIG_OF_PRIOR_STAGE
  20 phys_addr_t prior_stage_fdt_address __attribute__((section(".data")));
  21 #endif
  22 #ifndef CONFIG_XIP
  23 u32 hart_lottery __attribute__((section(".data"))) = 0;
  24
  25 /*
  26  * The main hart running U-Boot has acquired available_harts_lock until it has
  27  * finished initialization of global data.
  28  */
  29 u32 available_harts_lock = 1;
  30 #endif
  31
  32 static inline bool supports_extension(char ext)
  33 {
  34 #ifdef CONFIG_CPU
  35         struct udevice *dev;
  36         char desc[32];
  37
  38         uclass_find_first_device(UCLASS_CPU, &dev);
  39         if (!dev) {
  40                 debug("unable to find the RISC-V cpu device\n");
  41                 return false;
  42         }
  43         if (!cpu_get_desc(dev, desc, sizeof(desc))) {
  44                 /* skip the first 4 characters (rv32|rv64) */
  45                 if (strchr(desc + 4, ext))
  46                         return true;
  47         }
  48
  49         return false;
  50 #else  /* !CONFIG_CPU */
  51 #if CONFIG_IS_ENABLED(RISCV_MMODE)
  52         return csr_read(CSR_MISA) & (1 << (ext - 'a'));
  53 #else  /* !CONFIG_IS_ENABLED(RISCV_MMODE) */
  54 #warning "There is no way to determine the available extensions in S-mode."
  55 #warning "Please convert your board to use the RISC-V CPU driver."
  56         return false;
  57 #endif /* CONFIG_IS_ENABLED(RISCV_MMODE) */
  58 #endif /* CONFIG_CPU */
  59 }
  60
  61 static int riscv_cpu_probe(void)
  62 {
  63 #ifdef CONFIG_CPU
  64         int ret;
  65
  66         /* probe cpus so that RISC-V timer can be bound */
  67         ret = cpu_probe_all();
  68         if (ret)
  69                 return log_msg_ret("RISC-V cpus probe failed\n", ret);
  70 #endif
  71
  72         return 0;
  73 }
  74
  75 int arch_cpu_init_dm(void)
  76 {
  77         int ret;
  78
  79         ret = riscv_cpu_probe();
  80         if (ret)
  81                 return ret;
  82
  83         /* Enable FPU */
  84         if (supports_extension('d') || supports_extension('f')) {
  85                 csr_set(MODE_PREFIX(status), MSTATUS_FS);
  86                 csr_write(CSR_FCSR, 0);
  87         }
  88
  89         if (CONFIG_IS_ENABLED(RISCV_MMODE)) {
  90                 /*
  91                  * Enable perf counters for cycle, time,
  92                  * and instret counters only
  93                  */
  94                 csr_write(CSR_MCOUNTEREN, GENMASK(2, 0));
  95
  96                 /* Disable paging */
  97                 if (supports_extension('s'))
  98                         csr_write(CSR_SATP, 0);
  99         }
 100
 101         return 0;
 102 }
 103
 104 int arch_early_init_r(void)
 105 {
 106         return riscv_cpu_probe();
 107 }