diff --git a/llvm/tools/llvm-mca/Scheduler.cpp b/llvm/tools/llvm-mca/Scheduler.cpp index 4ba48a28b8ed..cc904c84b484 100644 --- a/llvm/tools/llvm-mca/Scheduler.cpp +++ b/llvm/tools/llvm-mca/Scheduler.cpp @@ -24,6 +24,11 @@ using namespace llvm; ResourceStrategy::~ResourceStrategy() = default; +void Scheduler::initializeStrategy(std::unique_ptr S) { + // Ensure we have a valid (non-null) strategy object. + Strategy = S ? std::move(S) : llvm::make_unique(); +} + void DefaultResourceStrategy::skipMask(uint64_t Mask) { NextInSequenceMask &= (~Mask); if (!NextInSequenceMask) { @@ -116,7 +121,7 @@ void ResourceManager::setCustomStrategyImpl(std::unique_ptr S, unsigned Index = getResourceStateIndex(ResourceID); assert(Index < Resources.size() && "Invalid processor resource index!"); assert(S && "Unexpected null strategy in input!"); - Strategies[Index].reset(S.get()); + Strategies[Index] = std::move(S); } unsigned ResourceManager::resolveResourceMask(uint64_t Mask) const { diff --git a/llvm/tools/llvm-mca/Scheduler.h b/llvm/tools/llvm-mca/Scheduler.h index 37ebdb453e23..ed138fd71804 100644 --- a/llvm/tools/llvm-mca/Scheduler.h +++ b/llvm/tools/llvm-mca/Scheduler.h @@ -430,6 +430,11 @@ class Scheduler : public HardwareUnit { std::vector ReadySet; std::vector IssuedSet; + /// Verify the given selection strategy and set the Strategy member + /// accordingly. If no strategy is provided, the DefaultSchedulerStrategy is + /// used. + void initializeStrategy(std::unique_ptr S); + /// Issue an instruction without updating the ready queue. void issueInstructionImpl( InstRef &IR, @@ -446,16 +451,19 @@ class Scheduler : public HardwareUnit { public: Scheduler(const llvm::MCSchedModel &Model, LSUnit *Lsu) - : LSU(Lsu), Strategy(llvm::make_unique()), - Resources(llvm::make_unique(Model)) {} + : LSU(Lsu), Resources(llvm::make_unique(Model)) { + initializeStrategy(nullptr); + } Scheduler(const llvm::MCSchedModel &Model, LSUnit *Lsu, std::unique_ptr SelectStrategy) - : LSU(Lsu), Strategy(std::move(SelectStrategy)), - Resources(llvm::make_unique(Model)) {} + : LSU(Lsu), Resources(llvm::make_unique(Model)) { + initializeStrategy(std::move(SelectStrategy)); + } Scheduler(std::unique_ptr RM, LSUnit *Lsu, std::unique_ptr SelectStrategy) - : LSU(Lsu), Strategy(std::move(SelectStrategy)), - Resources(std::move(RM)) {} + : LSU(Lsu), Resources(std::move(RM)) { + initializeStrategy(std::move(SelectStrategy)); + } // Stalls generated by the scheduler. enum Status {