llvm/test/CodeGen/MLRegAlloc/dev-mode-prio-logging.ll

   1 ; REQUIRES: have_tflite
   2 ; REQUIRES: x86_64-linux
   3 ;
   4 ; Check that we log correctly, both with a learned policy, and the default policy
   5 ;
   6 ; RUN: llc -o /dev/null -mtriple=x86_64-linux-unknown -regalloc=greedy \
   7 ; RUN:   -regalloc-enable-priority-advisor=development \
   8 ; RUN:   -regalloc-priority-training-log=%t1 \
   9 ; RUN:   < %S/Inputs/input.ll
  10 ; RUN: %python %S/../../../lib/Analysis/models/log_reader.py %t1 > %t1.readable
  11 ; RUN: FileCheck --input-file %t1.readable %s --check-prefixes=CHECK,NOML
  12 ; RUN: diff %t1.readable %S/Inputs/reference-prio-log-noml.txt
  13
  14 ; RUN: rm -rf %t && mkdir %t
  15 ; RUN: %python %S/../../../lib/Analysis/models/gen-regalloc-priority-test-model.py %t_savedmodel
  16 ; RUN: %python %S/../../../lib/Analysis/models/saved-model-to-tflite.py %t_savedmodel %t
  17 ; RUN: llc -o /dev/null -mtriple=x86_64-linux-unknown -regalloc=greedy \
  18 ; RUN:   -regalloc-enable-priority-advisor=development \
  19 ; RUN:   -regalloc-priority-training-log=%t2 \
  20 ; RUN:   -regalloc-priority-model=%t < %S/Inputs/input.ll
  21 ; RUN: %python %S/../../../lib/Analysis/models/log_reader.py %t2 > %t2.readable
  22 ; RUN: FileCheck --input-file %t2.readable %s --check-prefixes=CHECK,ML
  23
  24 ; CHECK-NOT: nan
  25 ; CHECK-LABEL: priority:
  26 ; NOML-SAME: 2684358144.0
  27 ; ML-SAME: 3535
  28 ; CHECK-LABEL: reward: