forked from OSchip/llvm-project
Use isFromMainFile instead of comparing FileIDs directly.
llvm-svn: 49687
This commit is contained in:
parent
f6aad13242
commit
d61ed3b6cb
|
@ -384,7 +384,7 @@ void RewriteTest::HandleTopLevelDecl(Decl *D) {
|
|||
RewriteForwardProtocolDecl(FP);
|
||||
}
|
||||
// If we have a decl in the main file, see if we should rewrite it.
|
||||
if (SM->getDecomposedFileLoc(Loc).first == MainFileID)
|
||||
if (SM->isFromMainFile(Loc))
|
||||
return HandleDeclInMainFile(D);
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue