e5e3dccd07
This patch adds in instruction based features to the regalloc advisor gated behind a flag so a user can decide at runtime whether or not they want to enable the feature. The features are only enabled when LLVM is compiled in MLGO develpment mode (LLVM_HAVE_TF_API) is set to true. To extract the instruction features, I'm taking a list of segments from each LiveInterval and noting the start and end SlotIndices. This list is then sorted based on the start SlotIndex and I iterate through each SlotIndex to grab instructions, making sure to check for overlaps. This results in a vector of opcodes and binary mapping matrix that maps live ranges to the opcodes of the instructions within that LR. Reviewed By: mtrofin Differential Revision: https://reviews.llvm.org/D131930
53 lines
2.2 KiB
LLVM
53 lines
2.2 KiB
LLVM
; REQUIRES: have_tf_api
|
|
; REQUIRES: x86_64-linux
|
|
;
|
|
; Check that we log the currently in development features correctly with both the default
|
|
; case and with a learned policy.
|
|
;
|
|
; RUN: llc -mtriple=x86_64-linux-unknown -regalloc=greedy -regalloc-enable-advisor=development \
|
|
; RUN: -regalloc-training-log=%t1 -tfutils-text-log \
|
|
; RUN: -regalloc-enable-development-features < %S/Inputs/input.ll
|
|
; RUN: sed -i 's/ \+/ /g' %t1
|
|
; RUN: sed -i 's/\\n key:/\n key:/g' %t1
|
|
; RUN: sed -i 's/\\n feature/\n feature/g' %t1
|
|
; RUN: sed -i 's/\\n/ /g' %t1
|
|
; RUN: FileCheck --input-file %t1 %s
|
|
|
|
; RUN: rm -rf %t && mkdir %t
|
|
; RUN: %python %S/../../../lib/Analysis/models/gen-regalloc-eviction-test-model.py %t_savedmodel
|
|
; RUN: %python %S/../../../lib/Analysis/models/saved-model-to-tflite.py %t_savedmodel %t
|
|
; RUN: llc -mtriple=x86_64-linux-unknown -regalloc=greedy -regalloc-enable-advisor=development \
|
|
; RUN: -regalloc-training-log=%t2 -tfutils-text-log -regalloc-model=%t \
|
|
; RUN: -regalloc-enable-development-features < %S/Inputs/input.ll
|
|
; RUN: sed -i 's/ \+/ /g' %t2
|
|
; RUN: sed -i 's/\\n key:/\n key:/g' %t2
|
|
; RUN: sed -i 's/\\n feature/\n feature/g' %t2
|
|
; RUN: sed -i 's/\\n/ /g' %t2
|
|
; RUN: FileCheck --input-file %t2 %s
|
|
|
|
; CHECK-NOT: nan
|
|
; CHECK-LABEL: key: \"instructions\"
|
|
; Check the first five opcodes in the first eviction problem
|
|
; CHECK-NEXT: value: 19
|
|
; CHECK-SAME: value: 19
|
|
; CHECK-SAME: value: 3031
|
|
; CHECK-SAME: value: 1245
|
|
; CHECK-SAME: value: 1264
|
|
; The first eviction problem is significantly less than 300 instructions. Check
|
|
; that there is a zero value
|
|
; CHECK-SAME: value: 0
|
|
; Only the candidate virtreg and the 10th LR are included in this problem. Make
|
|
; sure the other LRs have values of zero.
|
|
; CHECK-LABEL: key: \"instructions_mapping\"
|
|
; CHECK-COUNT-2700: value: 0
|
|
; CHECK-SAME: value: 1
|
|
; Indexing 300 back from where the candidate vr actual resides due to the fact
|
|
; that not all the values between the 10th LR and the candidate are zero.
|
|
; CHECK-COUNT-6600: value: 0
|
|
; CHECK-SAME: value: 1
|
|
; Ensure that we can still go through the mapping matrices for the rest of the
|
|
; eviction problems to make sure we haven't hit the end of the matrix above.
|
|
; There are a total of 23 eviction problems with this test.
|
|
; CHECK-COUNT-22: int64_list
|
|
; CHECK: key: \"is_free\"
|