1 //===-------------------- RISCVCustomBehaviour.h -----------------*-C++ -*-===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 /// \file 9 /// 10 /// This file defines the RISCVCustomBehaviour class which inherits from 11 /// CustomBehaviour. This class is used by the tool llvm-mca to enforce 12 /// target specific behaviour that is not expressed well enough in the 13 /// scheduling model for mca to enforce it automatically. 14 /// 15 //===----------------------------------------------------------------------===// 16 17 #ifndef LLVM_LIB_TARGET_RISCV_MCA_RISCVCUSTOMBEHAVIOUR_H 18 #define LLVM_LIB_TARGET_RISCV_MCA_RISCVCUSTOMBEHAVIOUR_H 19 20 #include "llvm/ADT/SmallVector.h" 21 #include "llvm/MC/MCInst.h" 22 #include "llvm/MC/MCInstrDesc.h" 23 #include "llvm/MC/MCInstrInfo.h" 24 #include "llvm/MCA/CustomBehaviour.h" 25 26 namespace llvm { 27 namespace mca { 28 29 class RISCVLMULInstrument : public Instrument { 30 public: 31 static const StringRef DESC_NAME; 32 static bool isDataValid(StringRef Data); 33 34 explicit RISCVLMULInstrument(StringRef Data) : Instrument(DESC_NAME, Data) {} 35 36 ~RISCVLMULInstrument() = default; 37 38 uint8_t getLMUL() const; 39 }; 40 41 class RISCVSEWInstrument : public Instrument { 42 public: 43 static const StringRef DESC_NAME; 44 static bool isDataValid(StringRef Data); 45 46 explicit RISCVSEWInstrument(StringRef Data) : Instrument(DESC_NAME, Data) {} 47 48 ~RISCVSEWInstrument() = default; 49 50 uint8_t getSEW() const; 51 }; 52 53 class RISCVInstrumentManager : public InstrumentManager { 54 public: 55 RISCVInstrumentManager(const MCSubtargetInfo &STI, const MCInstrInfo &MCII) 56 : InstrumentManager(STI, MCII) {} 57 58 bool shouldIgnoreInstruments() const override { return false; } 59 bool supportsInstrumentType(StringRef Type) const override; 60 61 /// Create a Instrument for RISC-V target 62 UniqueInstrument createInstrument(StringRef Desc, StringRef Data) override; 63 64 SmallVector<UniqueInstrument> createInstruments(const MCInst &Inst) override; 65 66 /// Using the Instrument, returns a SchedClassID to use instead of 67 /// the SchedClassID that belongs to the MCI or the original SchedClassID. 68 unsigned 69 getSchedClassID(const MCInstrInfo &MCII, const MCInst &MCI, 70 const SmallVector<Instrument *> &IVec) const override; 71 }; 72 73 } // namespace mca 74 } // namespace llvm 75 76 #endif 77