spike/spike.cc

   1 // See LICENSE for license details.
   2
   3 #include "sim.h"
   4 #include "htif.h"
   5 #include "cachesim.h"
   6 #include "extension.h"
   7 #include <dlfcn.h>
   8 #include <fesvr/option_parser.h>
   9 #include <stdio.h>
  10 #include <stdlib.h>
  11 #include <getopt.h>
  12 #include <vector>
  13 #include <string>
  14 #include <memory>
  15
  16 static void help()
  17 {
  18   fprintf(stderr, "usage: spike [host options] <target program> [target options]\n");
  19   fprintf(stderr, "Host Options:\n");
  20   fprintf(stderr, "  -p <n>             Simulate <n> processors\n");
  21   fprintf(stderr, "  -m <n>             Provide <n> MB of target memory\n");
  22   fprintf(stderr, "  -d                 Interactive debug mode\n");
  23   fprintf(stderr, "  -h                 Print this help message\n");
  24   fprintf(stderr, "  --ic=<S>:<W>:<B>   Instantiate a cache model with S sets,\n");
  25   fprintf(stderr, "  --dc=<S>:<W>:<B>     W ways, and B-byte blocks (with S and\n");
  26   fprintf(stderr, "  --l2=<S>:<W>:<B>     B both powers of 2).\n");
  27   fprintf(stderr, "  --extension=<name> Specify RoCC Extension\n");
  28   fprintf(stderr, "  --extlib=<name>    Shared library to load\n");
  29   exit(1);
  30 }
  31
  32 int main(int argc, char** argv)
  33 {
  34   bool debug = false;
  35   size_t nprocs = 1;
  36   size_t mem_mb = 0;
  37   std::unique_ptr<icache_sim_t> ic;
  38   std::unique_ptr<dcache_sim_t> dc;
  39   std::unique_ptr<cache_sim_t> l2;
  40   std::function<extension_t*()> extension;
  41
  42   option_parser_t parser;
  43   parser.help(&help);
  44   parser.option('h', 0, 0, [&](const char* s){help();});
  45   parser.option('d', 0, 0, [&](const char* s){debug = true;});
  46   parser.option('p', 0, 1, [&](const char* s){nprocs = atoi(s);});
  47   parser.option('m', 0, 1, [&](const char* s){mem_mb = atoi(s);});
  48   parser.option(0, "ic", 1, [&](const char* s){ic.reset(new icache_sim_t(s));});
  49   parser.option(0, "dc", 1, [&](const char* s){dc.reset(new dcache_sim_t(s));});
  50   parser.option(0, "l2", 1, [&](const char* s){l2.reset(cache_sim_t::construct(s, "L2$"));});
  51   parser.option(0, "extension", 1, [&](const char* s){
  52     if (!extensions().count(s))
  53       fprintf(stderr, "unknown extension %s!\n", s), exit(-1);
  54     extension = extensions()[s];
  55   });
  56   parser.option(0, "extlib", 1, [&](const char *s){
  57     void *lib = dlopen(s, RTLD_NOW | RTLD_GLOBAL);
  58     if (lib == NULL) {
  59       fprintf(stderr, "Unable to load extlib '%s': %s\n", s, dlerror());
  60       exit(-1);
  61     }
  62   });
  63
  64   auto argv1 = parser.parse(argv);
  65   if (!*argv1)
  66     help();
  67   std::vector<std::string> htif_args(argv1, (const char*const*)argv + argc);
  68   sim_t s(nprocs, mem_mb, htif_args);
  69
  70   if (ic && l2) ic->set_miss_handler(&*l2);
  71   if (dc && l2) dc->set_miss_handler(&*l2);
  72   for (size_t i = 0; i < nprocs; i++)
  73   {
  74     if (ic) s.get_core(i)->get_mmu()->register_memtracer(&*ic);
  75     if (dc) s.get_core(i)->get_mmu()->register_memtracer(&*dc);
  76     if (extension) s.get_core(i)->register_extension(extension());
  77   }
  78
  79   s.set_debug(debug);
  80   return s.run();
  81 }