forked from OSchip/llvm-project
58 lines
3.0 KiB
LLVM
58 lines
3.0 KiB
LLVM
; Test that we can produce a log if we have or do not have a model, in development mode.
|
|
; REQUIRES: have_tf_api
|
|
; Generate mock model
|
|
; RUN: rm -rf %t_savedmodel %t
|
|
; RUN: %python %S/../../../../lib/Analysis/models/gen-inline-oz-test-model.py %t_savedmodel
|
|
; RUN: %python %S/../../../../lib/Analysis/models/saved-model-to-tflite.py %t_savedmodel %t
|
|
;
|
|
; RUN: opt -enable-ml-inliner=development -passes=scc-oz-module-inliner -training-log=- -tfutils-text-log -ml-inliner-model-under-training=%t -ml-inliner-ir2native-model=%S/../../../../unittests/Analysis/Inputs/ir2native_x86_64_model -S < %s | FileCheck %s
|
|
; RUN: opt -enable-ml-inliner=development -passes=scc-oz-module-inliner -training-log=- -tfutils-text-log -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 | FileCheck %s --check-prefixes=EXTRA-OUTPUTS,CHECK
|
|
; RUN: opt -enable-ml-inliner=development -passes=scc-oz-module-inliner -training-log=- -tfutils-text-log -ml-inliner-ir2native-model=%S/../../../../unittests/Analysis/Inputs/ir2native_x86_64_model -S < %s | FileCheck %s
|
|
; RUN: opt -enable-ml-inliner=development -passes=scc-oz-module-inliner -training-log=- -tfutils-text-log -ml-inliner-model-under-training=%t -S < %s | FileCheck %s --check-prefix=NOREWARD
|
|
; RUN: opt -enable-ml-inliner=development -passes=scc-oz-module-inliner -training-log=- -tfutils-text-log -S < %s | FileCheck %s --check-prefix=NOREWARD
|
|
target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
|
|
target triple = "x86_64-pc-linux-gnu"
|
|
declare i32 @f1(i32)
|
|
declare i32 @f2(i32)
|
|
define dso_local i32 @branches(i32) {
|
|
%cond = icmp slt i32 %0, 3
|
|
br i1 %cond, label %then, label %else
|
|
then:
|
|
%ret.1 = call i32 @f1(i32 %0)
|
|
br label %last.block
|
|
else:
|
|
%ret.2 = call i32 @f2(i32 %0)
|
|
br label %last.block
|
|
last.block:
|
|
%ret = phi i32 [%ret.1, %then], [%ret.2, %else]
|
|
ret i32 %ret
|
|
}
|
|
define dso_local i32 @top() {
|
|
%1 = call i32 @branches(i32 2)
|
|
ret i32 %1
|
|
}
|
|
!llvm.module.flags = !{!0}
|
|
!llvm.ident = !{!1}
|
|
!0 = !{i32 1, !"wchar_size", i32 4}
|
|
!1 = !{!"clang version 7.0.0-6 (tags/RELEASE_700/final)"}
|
|
; Check we produce a protobuf that has inlining decisions and rewards.
|
|
; CHECK: key: "delta_size"
|
|
; CHECK-NEXT: value {
|
|
; CHECK-NEXT: feature {
|
|
; CHECK-NEXT: int64_list {
|
|
; CHECK-NEXT: value: 0
|
|
; CHECK-NEXT: }
|
|
; CHECK-NEXT: }
|
|
; CHECK-NOT: fake_extra_output
|
|
; EXTRA-OUTPUTS: key: "fake_extra_output"
|
|
; EXTRA-OUTPUTS-NEXT: value {
|
|
; EXTRA-OUTPUTS-NEXT: feature {
|
|
; EXTRA-OUTPUTS-NEXT: int64_list {
|
|
; EXTRA-OUTPUTS-NEXT: value: {{[0-9]+}}
|
|
; CHECK: key: "inlining_decision"
|
|
; CHECK-NEXT: value {
|
|
; CHECK-NEXT: feature {
|
|
; CHECK-NEXT: int64_list {
|
|
; CHECK-NEXT: value: 1
|
|
; NOREWARD-NOT: key: "delta_size"
|