xref: /llvm-project/llvm/test/Transforms/Inline/ML/development-training-log.ll (revision 5898be19e66d97662e9d87c526105c6ebd961cba)
1; Test that we can produce a log if we have or do not have a model, in development mode.
2; REQUIRES: have_tflite
3; Generate mock model
4; RUN: rm -rf %t_savedmodel %t
5; RUN: %python %S/../../../../lib/Analysis/models/gen-inline-oz-test-model.py %t_savedmodel
6; RUN: %python %S/../../../../lib/Analysis/models/saved-model-to-tflite.py %t_savedmodel %t
7;
8; RUN: opt -enable-ml-inliner=development -passes=scc-oz-module-inliner -training-log=%t1 -ml-inliner-model-under-training=%t -ml-inliner-ir2native-model=%S/../../../../unittests/Analysis/Inputs/ir2native_x86_64_model -S < %s
9; RUN: %python %S/../../../../lib/Analysis/models/log_reader.py %t1 | FileCheck %s
10; RUN: opt -enable-ml-inliner=development -passes=scc-oz-module-inliner -training-log=%t2 -ml-inliner-model-under-training=%t -ml-inliner-ir2native-model=%S/../../../../unittests/Analysis/Inputs/ir2native_x86_64_model -ml-inliner-output-spec-override=%S/Inputs/test_output_spec.json -S < %s
11; RUN: %python %S/../../../../lib/Analysis/models/log_reader.py %t2 | FileCheck %s --check-prefixes=EXTRA-OUTPUTS,CHECK
12; RUN: opt -enable-ml-inliner=development -passes=scc-oz-module-inliner -training-log=%t3 -ml-inliner-ir2native-model=%S/../../../../unittests/Analysis/Inputs/ir2native_x86_64_model -S < %s
13; RUN: %python %S/../../../../lib/Analysis/models/log_reader.py %t3 | FileCheck %s
14; RUN: opt -enable-ml-inliner=development -passes=scc-oz-module-inliner -training-log=%t4 -ml-inliner-model-under-training=%t -S < %s
15; RUN: %python %S/../../../../lib/Analysis/models/log_reader.py %t4 | FileCheck %s --check-prefix=NOREWARD
16; RUN: opt -enable-ml-inliner=development -passes=scc-oz-module-inliner -training-log=%t5 -S < %s
17; RUN: %python %S/../../../../lib/Analysis/models/log_reader.py %t5| FileCheck %s --check-prefix=NOREWARD
18target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
19target triple = "x86_64-pc-linux-gnu"
20declare i32 @f1(i32)
21declare i32 @f2(i32)
22define dso_local i32 @branches(i32) {
23  %cond = icmp slt i32 %0, 3
24  br i1 %cond, label %then, label %else
25then:
26  %ret.1 = call i32 @f1(i32 %0)
27  br label %last.block
28else:
29  %ret.2 = call i32 @f2(i32 %0)
30  br label %last.block
31last.block:
32  %ret = phi i32 [%ret.1, %then], [%ret.2, %else]
33  ret i32 %ret
34}
35define dso_local i32 @top() {
36  %1 = call i32 @branches(i32 2)
37  ret i32 %1
38}
39!llvm.module.flags = !{!0}
40!llvm.ident = !{!1}
41!0 = !{i32 1, !"wchar_size", i32 4}
42!1 = !{!"clang version 7.0.0-6 (tags/RELEASE_700/final)"}
43; Check we produce a log that has inlining decisions and rewards.
44; CHECK-NOT:        fake_extra_output:
45; EXTRA-OUTPUTS:    fake_extra_output: {{[0-9]+}}
46; CHECK:            inlining_decision: 1
47; CHECK:            delta_size: 0
48; NOREWARD-NOT:     delta_size:
49