2 files changed, 5 insertions, 1 deletions
diff --git a/tools/llvm-mc/AsmLexer.h b/tools/llvm-mc/AsmLexer.h
index 5d59d0a..32bbb50 100644
--- a/tools/llvm-mc/AsmLexer.h
+++ b/tools/llvm-mc/AsmLexer.h
@@ -14,6 +14,7 @@
 #ifndef ASMLEXER_H
 #define ASMLEXER_H
 
+#include "llvm/MC/MCAsmLexer.h"
 #include "llvm/Support/DataTypes.h"
 #include <string>
 #include <cassert>
@@ -52,7 +53,7 @@ namespace asmtok {
 }
 
 /// AsmLexer - Lexer class for assembly files.
-class AsmLexer {
+class AsmLexer : public MCAsmLexer {
   SourceMgr &SrcMgr;
   
   const char *CurPtr;
diff --git a/tools/llvm-mc/AsmParser.h b/tools/llvm-mc/AsmParser.h
index b996758..d9f4b4c 100644
--- a/tools/llvm-mc/AsmParser.h
+++ b/tools/llvm-mc/AsmParser.h
@@ -24,6 +24,7 @@ class MCContext;
 class MCInst;
 class MCStreamer;
 class MCValue;
+class TargetAsmParser;
 
 class AsmParser : MCAsmParser {
 public:
@@ -46,6 +47,8 @@ public:
 public:
   TargetAsmParser &getTargetParser() const { return TargetParser; }
 
+  virtual MCAsmLexer &getLexer() { return Lexer; }
+
 private:
   bool ParseStatement();