xref: /llvm-project/llvm/test/Transforms/Inline/ML/ml-test-development-mode.ll (revision 5898be19e66d97662e9d87c526105c6ebd961cba)
1; The default inliner doesn't elide @adder, it believes it's too costly to inline
2; adder into switcher. The ML inliner carries out that inlining, resulting in
3; a smaller result (part of it is that adder gets elided).
4;
5; This test uses Inputs/test-module.ll, as it shares it with a similar test
6; for the 'release' mode.
7;
8; REQUIRES: have_tflite
9; RUN: rm -rf %t
10; RUN: rm -rf %t_savedmodel
11; RUN: %python %S/../../../../lib/Analysis/models/gen-inline-oz-test-model.py %t_savedmodel
12; RUN: %python %S/../../../../lib/Analysis/models/saved-model-to-tflite.py %t_savedmodel %t
13; RUN: opt -passes=scc-oz-module-inliner -enable-ml-inliner=default -S < %S/Inputs/test-module.ll 2>&1 | FileCheck %S/Inputs/test-module.ll --check-prefix=DEFAULT
14; RUN: opt -passes=scc-oz-module-inliner -enable-ml-inliner=development -ml-inliner-model-under-training=%t -S < %S/Inputs/test-module.ll 2>&1 | FileCheck %S/Inputs/test-module.ll --check-prefix=CHECK
15; RUN: opt -passes=scc-oz-module-inliner -enable-ml-inliner=development -training-log=%t.log -S < %S/Inputs/test-module.ll 2>&1
16; RUN: %python %S/../../../../lib/Analysis/models/log_reader.py %t.log | FileCheck %s --check-prefix=CHECK-LOG
17
18CHECK-LOG: observation: 0
19CHECK-LOG-NEXT: {{^sroa_savings:}} 0
20CHECK-LOG: {{^cost_estimate:}} -30
21CHECK-LOG: {{^inlining_decision:}} 1
22CHECK-LOG-NEXT: observation: 1
23CHECK-LOG: observation: 6
24