derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
268 lines
12 KiB
TableGen
268 lines
12 KiB
TableGen
// RUN: llvm-tblgen -I %p/../../../include -gen-global-isel-combiner \
|
|
// RUN: -combiners=MyCombiner %s | \
|
|
// RUN: FileCheck %s
|
|
|
|
include "llvm/Target/Target.td"
|
|
include "llvm/Target/GlobalISel/Combine.td"
|
|
|
|
def MyTargetISA : InstrInfo;
|
|
def MyTarget : Target { let InstructionSet = MyTargetISA; }
|
|
|
|
def dummy;
|
|
|
|
def HasAnswerToEverything : Predicate<"Subtarget->getAnswerToUniverse() == 42 && Subtarget->getAnswerToLife() == 42">;
|
|
|
|
def WipOpcodeTest0 : GICombineRule<
|
|
(defs root:$d),
|
|
(match (wip_match_opcode G_TRUNC):$d),
|
|
(apply [{ APPLY }])>;
|
|
|
|
def WipOpcodeTest1 : GICombineRule<
|
|
(defs root:$d),
|
|
(match (wip_match_opcode G_TRUNC, G_SEXT):$d),
|
|
(apply [{ APPLY }])>;
|
|
|
|
// Note: also checks that spaces in the type name are removed.
|
|
def reg_matchinfo : GIDefMatchData<"Register ">;
|
|
def InstTest0 : GICombineRule<
|
|
(defs root:$d, reg_matchinfo:$r0, reg_matchinfo:$r1),
|
|
(match (COPY $a, $b):$d),
|
|
(apply [{ APPLY ${r0}, ${r1} }])>;
|
|
|
|
let Predicates = [HasAnswerToEverything] in
|
|
def InstTest1 : GICombineRule<
|
|
(defs root:$d, reg_matchinfo:$r0),
|
|
(match (COPY $a, $b):$d,
|
|
(G_ZEXT $b, $c),
|
|
[{ return CHECK ${a}, ${b}, ${c}, ${d} }]),
|
|
(apply [{ APPLY }])>;
|
|
|
|
def InOutInstTest0 : GICombineRule<
|
|
(defs root:$root),
|
|
(match (G_ZEXT $tmp, $ext),
|
|
(G_STORE $tmp, $ptr):$root),
|
|
(apply (G_STORE $ext, $ptr):$root)>;
|
|
|
|
// Imm operand of G_CONSTANT should match a literal int, while the second
|
|
// should match a constant.
|
|
def InOutInstTest1 : GICombineRule<
|
|
(defs root:$dst),
|
|
(match
|
|
(G_CONSTANT $x, -42:$z),
|
|
(G_AND $dst, $x, (i32 43))),
|
|
(apply (G_TRUNC $dst, $z))>;
|
|
|
|
def MatchICst: GICombinePatFrag<
|
|
(outs),
|
|
(ins gi_mo:$foo, gi_imm:$cst),
|
|
[(pattern "return matchIConstant(${foo}, ${cst});")]>;
|
|
|
|
def PatFragTest0 : GICombineRule<
|
|
(defs root:$dst),
|
|
(match (G_ZEXT $dst, $cst), (MatchICst $cst, (i32 0))),
|
|
(apply (COPY $dst, (i32 0)))>;
|
|
|
|
// TODO: add test with temp reg use
|
|
|
|
def MyCombiner: GICombiner<"GenMyCombiner", [
|
|
WipOpcodeTest0,
|
|
WipOpcodeTest1,
|
|
InstTest0,
|
|
InstTest1,
|
|
InOutInstTest0,
|
|
InOutInstTest1,
|
|
PatFragTest0
|
|
]>;
|
|
|
|
// We have at most 2 registers used by one rule at a time, so we should only have 2 registers MDInfos.
|
|
|
|
// Check predicates
|
|
// CHECK: switch (PredicateID) {
|
|
// CHECK-NEXT: case GICXXPred_MI_Predicate_GICombiner0: {
|
|
// CHECK-NEXT: return matchIConstant(State.MIs[0]->getOperand(1), 0);
|
|
// CHECK-NEXT: }
|
|
|
|
// Verify we gate on opcodes with generated combines and reset MatchData on each
|
|
// tryCombineAll.
|
|
// CHECK: static bool GenMyCombiner_canMatchOpcode(unsigned Opc) {
|
|
// CHECK-NEXT: switch (Opc) {
|
|
// CHECK-DAG: case TargetOpcode::COPY:
|
|
// CHECK-DAG: case TargetOpcode::G_AND:
|
|
// CHECK-DAG: case TargetOpcode::G_STORE:
|
|
// CHECK-DAG: case TargetOpcode::G_TRUNC:
|
|
// CHECK-DAG: case TargetOpcode::G_SEXT:
|
|
// CHECK-DAG: case TargetOpcode::G_ZEXT:
|
|
// CHECK: return true;
|
|
// CHECK-NEXT: default:
|
|
// CHECK-NEXT: return false;
|
|
// CHECK-NEXT: }
|
|
// CHECK-NEXT: }
|
|
// CHECK-EMPTY:
|
|
// CHECK: bool GenMyCombiner::tryCombineAll(MachineInstr &I) const {
|
|
// CHECK-NEXT: if (!GenMyCombiner_canMatchOpcode(I.getOpcode()))
|
|
// CHECK-NEXT: return false;
|
|
// CHECK-NEXT: const PredicateBitset AvailableFeatures = getAvailableFeatures();
|
|
// CHECK-NEXT: State.MIs.clear();
|
|
// CHECK-NEXT: State.MIs.push_back(&I);
|
|
// CHECK-NEXT: if (executeMatchTable(*this, State, ExecInfo, B, getMatchTable(), Helper.getTII(), MRI, Helper.getTRI(), Helper.getRBI(), AvailableFeatures, /*CoverageInfo*/ nullptr))
|
|
// CHECK-NEXT: return true;
|
|
// CHECK-NEXT: }
|
|
// CHECK-EMPTY:
|
|
// CHECK-NEXT: return false;
|
|
// CHECK-NEXT: }
|
|
|
|
// Check apply
|
|
// CHECK: enum {
|
|
// CHECK-NEXT: GICXXCustomAction_GICombiner0 = GICXXCustomAction_Invalid + 1,
|
|
// CHECK-NEXT: GICXXCustomAction_GICombiner1,
|
|
// CHECK-NEXT: GICXXCustomAction_GICombiner2,
|
|
// CHECK-NEXT: };
|
|
|
|
// CHECK: bool GenMyCombiner::runCustomAction(unsigned ApplyID, const MatcherState &State, NewMIVector &OutMIs) const {
|
|
// CHECK-NEXT: Helper.getBuilder().setInstrAndDebugLoc(*State.MIs[0]);
|
|
// CHECK-NEXT: switch(ApplyID) {
|
|
// CHECK-NEXT: case GICXXCustomAction_GICombiner0:{
|
|
// CHECK-NEXT: // Apply Patterns
|
|
// CHECK-NEXT: APPLY
|
|
// CHECK-NEXT: return true;
|
|
// CHECK-NEXT: }
|
|
// CHECK-NEXT: case GICXXCustomAction_GICombiner1:{
|
|
// CHECK-NEXT: Register GIMatchData_r0;
|
|
// CHECK-NEXT: Register GIMatchData_r1;
|
|
// CHECK-NEXT: // Apply Patterns
|
|
// CHECK-NEXT: APPLY GIMatchData_r0, GIMatchData_r1
|
|
// CHECK-NEXT: return true;
|
|
// CHECK-NEXT: }
|
|
// CHECK-NEXT: case GICXXCustomAction_GICombiner2:{
|
|
// CHECK-NEXT: Register GIMatchData_r0;
|
|
// CHECK-NEXT: // Match Patterns
|
|
// CHECK-NEXT: if(![&](){return CHECK State.MIs[0]->getOperand(0), State.MIs[0]->getOperand(1), State.MIs[1]->getOperand(1), State.MIs[0]}()) {
|
|
// CHECK-NEXT: return false;
|
|
// CHECK-NEXT: }
|
|
// CHECK-NEXT: // Apply Patterns
|
|
// CHECK-NEXT: APPLY
|
|
// CHECK-NEXT: return true;
|
|
// CHECK-NEXT: }
|
|
// CHECK-NEXT: }
|
|
// CHECK-NEXT: llvm_unreachable("Unknown Apply Action");
|
|
// CHECK-NEXT: }
|
|
|
|
// Verify match table.
|
|
// CHECK: const uint8_t *GenMyCombiner::getMatchTable() const {
|
|
// CHECK-NEXT: constexpr static uint8_t MatchTable0[] = {
|
|
// CHECK-NEXT: GIM_SwitchOpcode, /*MI*/0, /*[*/GIMT_Encode2([[#LOWER:]]), GIMT_Encode2([[#UPPER:]]), /*)*//*default:*//*Label 6*/ GIMT_Encode4([[#DEFAULT:]]),
|
|
// CHECK-NEXT: /*TargetOpcode::COPY*//*Label 0*/ GIMT_Encode4([[#%u, mul(UPPER-LOWER, 4) + 10]]),
|
|
// CHECK-COUNT-40: GIMT_Encode4(0),
|
|
// CHECK-NEXT: /*TargetOpcode::G_AND*//*Label 1*/ GIMT_Encode4([[L1:[0-9]+]]),
|
|
// CHECK-COUNT-34: GIMT_Encode4(0),
|
|
// CHECK-NEXT: /*TargetOpcode::G_STORE*//*Label 2*/ GIMT_Encode4([[L2:[0-9]+]]),
|
|
// CHECK-COUNT-30: GIMT_Encode4(0),
|
|
// CHECK-NEXT: /*TargetOpcode::G_TRUNC*//*Label 3*/ GIMT_Encode4([[L3:[0-9]+]]),
|
|
// CHECK-COUNT-4: GIMT_Encode4(0),
|
|
// CHECK-NEXT: /*TargetOpcode::G_SEXT*//*Label 4*/ GIMT_Encode4([[L4:[0-9]+]]), GIMT_Encode4(0),
|
|
// CHECK-NEXT: /*TargetOpcode::G_ZEXT*//*Label 5*/ GIMT_Encode4([[L5:[0-9]+]]),
|
|
// CHECK-NEXT: // Label 0: @[[#%u, mul(UPPER-LOWER, 4) + 10]]
|
|
// CHECK-NEXT: GIM_Try_CheckFeatures, /*On fail goto*//*Label 7*/ GIMT_Encode4([[L7:[0-9]+]]), GIMT_Encode2(GIFBS_HasAnswerToEverything), // Rule ID 4 //
|
|
// CHECK-NEXT: GIM_CheckSimplePredicate, GIMT_Encode2(GICXXPred_Simple_IsRule3Enabled),
|
|
// CHECK-NEXT: // MIs[0] a
|
|
// CHECK-NEXT: // No operand predicates
|
|
// CHECK-NEXT: // MIs[0] b
|
|
// CHECK-NEXT: GIM_RecordInsnIgnoreCopies, /*DefineMI*/1, /*MI*/0, /*OpIdx*/1, // MIs[1]
|
|
// CHECK-NEXT: GIM_CheckOpcode, /*MI*/1, GIMT_Encode2(TargetOpcode::G_ZEXT),
|
|
// CHECK-NEXT: // MIs[1] c
|
|
// CHECK-NEXT: // No operand predicates
|
|
// CHECK-NEXT: GIM_CheckIsSafeToFold, /*NumInsns*/1,
|
|
// CHECK-NEXT: // Combiner Rule #3: InstTest1
|
|
// CHECK-NEXT: GIR_DoneWithCustomAction, /*Fn*/GIMT_Encode2(GICXXCustomAction_GICombiner2),
|
|
// CHECK-NEXT: // Label 7: @[[L7]]
|
|
// CHECK-NEXT: GIM_Try, /*On fail goto*//*Label 8*/ GIMT_Encode4([[L8:[0-9]+]]), // Rule ID 3 //
|
|
// CHECK-NEXT: GIM_CheckSimplePredicate, GIMT_Encode2(GICXXPred_Simple_IsRule2Enabled),
|
|
// CHECK-NEXT: // MIs[0] a
|
|
// CHECK-NEXT: // No operand predicates
|
|
// CHECK-NEXT: // MIs[0] b
|
|
// CHECK-NEXT: // No operand predicates
|
|
// CHECK-NEXT: // Combiner Rule #2: InstTest0
|
|
// CHECK-NEXT: GIR_DoneWithCustomAction, /*Fn*/GIMT_Encode2(GICXXCustomAction_GICombiner1),
|
|
// CHECK-NEXT: // Label 8: @[[L8]]
|
|
// CHECK-NEXT: GIM_Reject,
|
|
// CHECK-NEXT: // Label 1: @[[L1]]
|
|
// CHECK-NEXT: GIM_Try, /*On fail goto*//*Label 9*/ GIMT_Encode4([[L9:[0-9]+]]), // Rule ID 6 //
|
|
// CHECK-NEXT: GIM_CheckSimplePredicate, GIMT_Encode2(GICXXPred_Simple_IsRule5Enabled),
|
|
// CHECK-NEXT: GIM_RootCheckType, /*Op*/2, /*Type*/GILLT_s32,
|
|
// CHECK-NEXT: // MIs[0] dst
|
|
// CHECK-NEXT: // No operand predicates
|
|
// CHECK-NEXT: // MIs[0] x
|
|
// CHECK-NEXT: GIM_RecordInsnIgnoreCopies, /*DefineMI*/1, /*MI*/0, /*OpIdx*/1, // MIs[1]
|
|
// CHECK-NEXT: GIM_CheckOpcode, /*MI*/1, GIMT_Encode2(TargetOpcode::G_CONSTANT),
|
|
// CHECK-NEXT: // MIs[1] z
|
|
// CHECK-NEXT: GIM_CheckLiteralInt, /*MI*/1, /*Op*/1, GIMT_Encode8(18446744073709551574u),
|
|
// CHECK-NEXT: GIM_CheckConstantInt8, /*MI*/0, /*Op*/2, 43,
|
|
// CHECK-NEXT: GIM_CheckIsSafeToFold, /*NumInsns*/1,
|
|
// CHECK-NEXT: // Combiner Rule #5: InOutInstTest1
|
|
// CHECK-NEXT: GIR_BuildRootMI, /*Opcode*/GIMT_Encode2(TargetOpcode::G_TRUNC),
|
|
// CHECK-NEXT: GIR_RootToRootCopy, /*OpIdx*/0, // dst
|
|
// CHECK-NEXT: GIR_Copy, /*NewInsnID*/0, /*OldInsnID*/1, /*OpIdx*/1, // z
|
|
// CHECK-NEXT: GIR_EraseRootFromParent_Done,
|
|
// CHECK-NEXT: // Label 9: @[[L9]]
|
|
// CHECK-NEXT: GIM_Reject,
|
|
// CHECK-NEXT: // Label 2: @[[L2]]
|
|
// CHECK-NEXT: GIM_Try, /*On fail goto*//*Label 10*/ GIMT_Encode4([[L10:[0-9]+]]), // Rule ID 5 //
|
|
// CHECK-NEXT: GIM_CheckSimplePredicate, GIMT_Encode2(GICXXPred_Simple_IsRule4Enabled),
|
|
// CHECK-NEXT: // MIs[0] tmp
|
|
// CHECK-NEXT: GIM_RecordInsnIgnoreCopies, /*DefineMI*/1, /*MI*/0, /*OpIdx*/0, // MIs[1]
|
|
// CHECK-NEXT: GIM_CheckOpcode, /*MI*/1, GIMT_Encode2(TargetOpcode::G_ZEXT),
|
|
// CHECK-NEXT: // MIs[1] ext
|
|
// CHECK-NEXT: // No operand predicates
|
|
// CHECK-NEXT: // MIs[0] ptr
|
|
// CHECK-NEXT: // No operand predicates
|
|
// CHECK-NEXT: GIM_CheckIsSafeToFold, /*NumInsns*/1,
|
|
// CHECK-NEXT: // Combiner Rule #4: InOutInstTest0
|
|
// CHECK-NEXT: GIR_BuildRootMI, /*Opcode*/GIMT_Encode2(TargetOpcode::G_STORE),
|
|
// CHECK-NEXT: GIR_Copy, /*NewInsnID*/0, /*OldInsnID*/1, /*OpIdx*/1, // ext
|
|
// CHECK-NEXT: GIR_RootToRootCopy, /*OpIdx*/1, // ptr
|
|
// CHECK-NEXT: GIR_MergeMemOperands, /*InsnID*/0, /*NumInsns*/2, /*MergeInsnID's*/0, 1,
|
|
// CHECK-NEXT: GIR_EraseRootFromParent_Done,
|
|
// CHECK-NEXT: // Label 10: @[[L10]]
|
|
// CHECK-NEXT: GIM_Reject,
|
|
// CHECK-NEXT: // Label 3: @[[L3]]
|
|
// CHECK-NEXT: GIM_Try, /*On fail goto*//*Label 11*/ GIMT_Encode4([[L11:[0-9]+]]), // Rule ID 0 //
|
|
// CHECK-NEXT: GIM_CheckSimplePredicate, GIMT_Encode2(GICXXPred_Simple_IsRule0Enabled),
|
|
// CHECK-NEXT: // Combiner Rule #0: WipOpcodeTest0; wip_match_opcode 'G_TRUNC'
|
|
// CHECK-NEXT: GIR_DoneWithCustomAction, /*Fn*/GIMT_Encode2(GICXXCustomAction_GICombiner0),
|
|
// CHECK-NEXT: // Label 11: @[[L11]]
|
|
// CHECK-NEXT: GIM_Try, /*On fail goto*//*Label 12*/ GIMT_Encode4([[L12:[0-9]+]]), // Rule ID 1 //
|
|
// CHECK-NEXT: GIM_CheckSimplePredicate, GIMT_Encode2(GICXXPred_Simple_IsRule1Enabled),
|
|
// CHECK-NEXT: // Combiner Rule #1: WipOpcodeTest1; wip_match_opcode 'G_TRUNC'
|
|
// CHECK-NEXT: GIR_DoneWithCustomAction, /*Fn*/GIMT_Encode2(GICXXCustomAction_GICombiner0),
|
|
// CHECK-NEXT: // Label 12: @[[L12]]
|
|
// CHECK-NEXT: GIM_Reject,
|
|
// CHECK-NEXT: // Label 4: @[[L4]]
|
|
// CHECK-NEXT: GIM_Try, /*On fail goto*//*Label 13*/ GIMT_Encode4([[L13:[0-9]+]]), // Rule ID 2 //
|
|
// CHECK-NEXT: GIM_CheckSimplePredicate, GIMT_Encode2(GICXXPred_Simple_IsRule1Enabled),
|
|
// CHECK-NEXT: // Combiner Rule #1: WipOpcodeTest1; wip_match_opcode 'G_SEXT'
|
|
// CHECK-NEXT: GIR_DoneWithCustomAction, /*Fn*/GIMT_Encode2(GICXXCustomAction_GICombiner0),
|
|
// CHECK-NEXT: // Label 13: @[[L13]]
|
|
// CHECK-NEXT: GIM_Reject,
|
|
// CHECK-NEXT: // Label 5: @[[L5]]
|
|
// CHECK-NEXT: GIM_Try, /*On fail goto*//*Label 14*/ GIMT_Encode4([[L14:[0-9]+]]), // Rule ID 7 //
|
|
// CHECK-NEXT: GIM_CheckSimplePredicate, GIMT_Encode2(GICXXPred_Simple_IsRule6Enabled),
|
|
// CHECK-NEXT: // MIs[0] dst
|
|
// CHECK-NEXT: // No operand predicates
|
|
// CHECK-NEXT: // MIs[0] cst
|
|
// CHECK-NEXT: // No operand predicates
|
|
// CHECK-NEXT: GIM_CheckCxxInsnPredicate, /*MI*/0, /*FnId*/GIMT_Encode2(GICXXPred_MI_Predicate_GICombiner0),
|
|
// CHECK-NEXT: GIR_MakeTempReg, /*TempRegID*/0, /*TypeID*/GILLT_s32,
|
|
// CHECK-NEXT: GIR_BuildConstant, /*TempRegID*/0, /*Val*/GIMT_Encode8(0),
|
|
// CHECK-NEXT: // Combiner Rule #6: PatFragTest0 @ [__PatFragTest0_match_1[0]]
|
|
// CHECK-NEXT: GIR_BuildRootMI, /*Opcode*/GIMT_Encode2(TargetOpcode::COPY),
|
|
// CHECK-NEXT: GIR_RootToRootCopy, /*OpIdx*/0, // dst
|
|
// CHECK-NEXT: GIR_AddSimpleTempRegister, /*InsnID*/0, /*TempRegID*/0,
|
|
// CHECK-NEXT: GIR_EraseRootFromParent_Done,
|
|
// CHECK-NEXT: // Label 14: @[[L14]]
|
|
// CHECK-NEXT: GIM_Reject,
|
|
// CHECK-NEXT: // Label 6: @[[#%u, DEFAULT]]
|
|
// CHECK-NEXT: GIM_Reject,
|
|
// CHECK-NEXT: }; // Size: [[#%u, DEFAULT + 1]] bytes
|
|
// CHECK-NEXT: return MatchTable0;
|
|
// CHECK-NEXT: }
|