#include "llvm/Analysis/ProfileSummaryInfo.h"
#include "llvm/IR/BasicBlock.h"
#include "llvm/IR/CFG.h"
+#include "llvm/IR/Constants.h"
#include "llvm/IR/DebugLoc.h"
#include "llvm/IR/Function.h"
#include "llvm/ProfileData/SampleProf.h"
#include "llvm/Support/CommandLine.h"
+#include "llvm/Transforms/Utils/ModuleUtils.h"
namespace llvm {
using namespace sampleprof;
/// Return true if the given callsite is hot wrt to hot cutoff threshold.
bool callsiteIsHot(const FunctionSamples *CallsiteFS, ProfileSummaryInfo *PSI,
bool ProfAccForSymsInList);
+
+/// Create a global variable to flag FSDiscriminators are used.
+void createFSDiscriminatorVariable(Module *M);
+
} // end of namespace sampleprofutil
} // end of namespace llvm
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/Debug.h"
#include "llvm/Support/raw_ostream.h"
+#include "llvm/Transforms/Utils/SampleProfileLoaderBaseUtil.h"
#include <unordered_map>
using namespace llvm;
using namespace sampleprof;
+using namespace sampleprofutil;
#define DEBUG_TYPE "mirfs-discriminators"
}
if (Changed) {
- Module *M = MF.getFunction().getParent();
- const char *FSDiscriminatorVar = "__llvm_fs_discriminator__";
- if (!M->getGlobalVariable(FSDiscriminatorVar)) {
- auto &Context = M->getContext();
- // Create a global variable to flag that FSDiscriminators are used.
- new GlobalVariable(*M, Type::getInt1Ty(Context), true,
- GlobalValue::WeakAnyLinkage,
- ConstantInt::getTrue(Context), FSDiscriminatorVar);
- }
-
+ createFSDiscriminatorVariable(MF.getFunction().getParent());
LLVM_DEBUG(dbgs() << "Num of FS Discriminators: " << NumNewD << "\n");
}
#include "llvm/Support/Debug.h"
#include "llvm/Support/raw_ostream.h"
#include "llvm/Transforms/Utils.h"
+#include "llvm/Transforms/Utils/SampleProfileLoaderBaseUtil.h"
#include <utility>
using namespace llvm;
+using namespace sampleprofutil;
#define DEBUG_TYPE "add-discriminators"
if (NoDiscriminators || !F.getSubprogram())
return false;
+ // Create FSDiscriminatorVariable if flow sensitive discriminators are used.
+ if (EnableFSDiscriminator)
+ createFSDiscriminatorVariable(F.getParent());
+
bool Changed = false;
using Location = std::pair<StringRef, unsigned>;
return Total > 0 ? Used * 100 / Total : 100;
}
+/// Create a global variable to flag FSDiscriminators are used.
+void createFSDiscriminatorVariable(Module *M) {
+ const char *FSDiscriminatorVar = "__llvm_fs_discriminator__";
+ if (M->getGlobalVariable(FSDiscriminatorVar))
+ return;
+
+ auto &Context = M->getContext();
+ // Place this variable to llvm.used so it won't be GC'ed.
+ appendToUsed(*M, {new GlobalVariable(*M, Type::getInt1Ty(Context), true,
+ GlobalValue::WeakODRLinkage,
+ ConstantInt::getTrue(Context),
+ FSDiscriminatorVar)});
+}
+
} // end of namespace sampleprofutil
} // end of namespace llvm