correctly parse extension for spike-disasm

diff --git a/spike_main/spike-dasm.cc b/spike_main/spike-dasm.cc
index 7f15db0bb168be15967d1b14572cba92f75087bb..aecaa0a05ec3c3c13cf174572fe8be3c38949414 100644 (file)
--- a/spike_main/spike-dasm.cc
+++ b/spike_main/spike-dasm.cc
@@ -22,6 +22,12 @@ int main(int argc, char** argv)
   std::function<extension_t*()> extension;
   option_parser_t parser;
   parser.option(0, "extension", 1, [&](const char* s){extension = find_extension(s);});
+  parser.parse(argv);
+
+  if (extension) {
+    for (auto disasm_insn : extension()->get_disasms())
+      d.add_insn(disasm_insn);
+  }
 
   while (getline(cin, s))
   {