1658456ccf
This patch implements a custom printer/parser for the immediate operand of s_wait_alu that prints/parses the decoded counter values. Format: ``` .<counter1>_<value1>_<counter2>_<value2> ``` Example: `s_wait_alu .VaVdst_1_VmVsrc_1` ; Which is equivalent to this: `s_wait_alu 8167` Features: - If a counter is at its maximum value it won't get printed. - The parser will error out if a counter is greater or equal to its max value. - If all counters are disabled we can use 'AllOff'. - For now we also accept numeric values for backwards compatibility with older MIR. Note: This is similar to https://github.com/llvm/llvm-project/pull/96004 but for `s_wait_alu`.
73 lines
2.7 KiB
C++
73 lines
2.7 KiB
C++
//===-- llvm/Target/AMDGPU/AMDGPUMIRFormatter.h -----------------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
/// \file
|
|
/// AMDGPU specific overrides of MIRFormatter.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_LIB_TARGET_AMDGPUMIRFORMATTER_H
|
|
#define LLVM_LIB_TARGET_AMDGPUMIRFORMATTER_H
|
|
|
|
#include "Utils/AMDGPUBaseInfo.h"
|
|
#include "llvm/CodeGen/MIRFormatter.h"
|
|
|
|
namespace llvm {
|
|
|
|
class MachineFunction;
|
|
struct PerFunctionMIParsingState;
|
|
|
|
class AMDGPUMIRFormatter final : public MIRFormatter {
|
|
public:
|
|
explicit AMDGPUMIRFormatter(const MCSubtargetInfo &STI) : STI(STI) {}
|
|
~AMDGPUMIRFormatter() override = default;
|
|
|
|
/// Implement target specific printing for machine operand immediate value, so
|
|
/// that we can have more meaningful mnemonic than a 64-bit integer. Passing
|
|
/// None to OpIdx means the index is unknown.
|
|
void printImm(raw_ostream &OS, const MachineInstr &MI,
|
|
std::optional<unsigned> OpIdx, int64_t Imm) const override;
|
|
|
|
/// Implement target specific parsing of immediate mnemonics. The mnemonic is
|
|
/// a string with a leading dot.
|
|
bool parseImmMnemonic(const unsigned OpCode, const unsigned OpIdx,
|
|
StringRef Src, int64_t &Imm,
|
|
ErrorCallbackType ErrorCallback) const override;
|
|
|
|
/// Implement target specific parsing of target custom pseudo source value.
|
|
bool
|
|
parseCustomPseudoSourceValue(StringRef Src, MachineFunction &MF,
|
|
PerFunctionMIParsingState &PFS,
|
|
const PseudoSourceValue *&PSV,
|
|
ErrorCallbackType ErrorCallback) const override;
|
|
|
|
private:
|
|
const MCSubtargetInfo &STI;
|
|
/// Prints the string to represent s_wait_alu immediate value.
|
|
void printSWaitAluImm(uint64_t Imm, raw_ostream &OS) const;
|
|
/// Print the string to represent s_delay_alu immediate value
|
|
void printSDelayAluImm(int64_t Imm, llvm::raw_ostream &OS) const;
|
|
|
|
/// Parse the immediate pseudo literal for s_wait_alu
|
|
bool parseSWaitAluImmMnemonic(
|
|
const unsigned int OpIdx, int64_t &Imm, StringRef &Src,
|
|
MIRFormatter::ErrorCallbackType &ErrorCallback) const;
|
|
|
|
/// Parse the immediate pseudo literal for s_delay_alu
|
|
bool parseSDelayAluImmMnemonic(
|
|
const unsigned int OpIdx, int64_t &Imm, llvm::StringRef &Src,
|
|
llvm::MIRFormatter::ErrorCallbackType &ErrorCallback) const;
|
|
|
|
};
|
|
|
|
} // end namespace llvm
|
|
|
|
#endif
|