From d19fcbcde490a0b0069b386179c6d055ffa79adc Mon Sep 17 00:00:00 2001
From: Evgeny Mankov <evgeny.mankov@gmail.com>
Date: Mon, 22 Jul 2019 22:13:54 +0300
Subject: [PATCH] [HIPIFY][#1239] Fix an assert

---
 hipify-clang/src/HipifyAction.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/hipify-clang/src/HipifyAction.cpp b/hipify-clang/src/HipifyAction.cpp
index 241ca7ecae..d6fe9b88e5 100644
--- a/hipify-clang/src/HipifyAction.cpp
+++ b/hipify-clang/src/HipifyAction.cpp
@@ -267,11 +267,12 @@ void HipifyAction::PragmaDirective(clang::SourceLocation Loc, clang::PragmaIntro
     return;
   }
   clang::Preprocessor& PP = getCompilerInstance().getPreprocessor();
-  const clang::Token tok = PP.LookAhead(0);
+  clang::Token tok;
+  PP.Lex(tok);
   StringRef Text(SM.getCharacterData(tok.getLocation()), tok.getLength());
   if (Text == "once") {
     pragmaOnce = true;
-    pragmaOnceLoc = PP.LookAhead(1).getLocation();
+    pragmaOnceLoc = tok.getEndLoc();
   }
 }