[mlgo] Use have_tflite instead of have_tf_api

[kazu]

Summary

We are in the process of retiring LLVM_HAVE_TF_API in favor of
LLVM_HAVE_TFLITE. This patch takes care of the transition in
llvm/test.

llvm/test/CMakeLists.txt

llvm_canonicalize_cmake_booleans(
   LLVM_ENABLE_LIBXML2
   LLVM_LINK_LLVM_DYLIB
   LLVM_TOOL_LTO_BUILD
   LLVM_USE_INTEL_JITEVENTS
   LLVM_BUILD_EXAMPLES
   LLVM_ENABLE_PLUGINS
   LLVM_BYE_LINK_INTO_TOOLS
   LLVM_HAVE_TF_AOT
-  LLVM_HAVE_TF_API
+  LLVM_HAVE_TFLITE
   LLVM_INLINER_MODEL_AUTOGENERATED
   LLVM_RAEVICT_MODEL_AUTOGENERATED
   LLVM_ENABLE_EXPENSIVE_CHECKS
   LLVM_INCLUDE_DXIL_TESTS
   LLVM_TOOL_LLVM_DRIVER_BUILD
   )
 
 configure_lit_site_cfg(

llvm/test/CodeGen/MLRegalloc/default-eviction-advisor.ll

 ; Check that, in the absence of dependencies, we emit an error message when
 ; trying to use ML-driven advisor.
 ; REQUIRES: !have_tf_aot
-; REQUIRES: !have_tf_api
+; REQUIRES: !have_tflite
 ; REQUIRES: default_triple
 ; RUN: not llc -O2 -regalloc-enable-advisor=development < %s 2>&1 | FileCheck %s
 ; RUN: not llc -O2 -regalloc-enable-advisor=release < %s 2>&1 | FileCheck %s
 ; RUN: llc -O2 -regalloc-enable-advisor=default < %s 2>&1 | FileCheck %s --check-prefix=DEFAULT
 
 ; regalloc-enable-advisor is not enabled for NVPTX
 ; UNSUPPORTED: target=nvptx{{.*}}
 
 define void @f2(i64 %lhs, i64 %rhs, i64* %addr) {
   %sum = add i64 %lhs, %rhs
   store i64 %sum, i64* %addr
   ret void
 }
 
 ; CHECK: Requested regalloc eviction advisor analysis could be created. Using default
 ; DEFAULT-NOT: Requested regalloc eviction advisor analysis could be created. Using default

llvm/test/CodeGen/MLRegalloc/default-priority-advisor.ll

 ; Check that, in the absence of dependencies, we emit an error message when
 ; trying to use ML-driven advisor.
 ; REQUIRES: !have_tf_aot
-; REQUIRES: !have_tf_api
+; REQUIRES: !have_tflite
 ; REQUIRES: default_triple
 ; RUN: not llc -O2 -regalloc-enable-priority-advisor=development < %s 2>&1 | FileCheck %s
 ; RUN: not llc -O2 -regalloc-enable-priority-advisor=release < %s 2>&1 | FileCheck %s
 ; RUN: llc -O2 -regalloc-enable-priority-advisor=default < %s 2>&1 | FileCheck %s --check-prefix=DEFAULT
 
 ; regalloc-enable-priority-advisor is not enabled for NVPTX
 ; UNSUPPORTED: target=nvptx{{.*}}
 
 define void @f2(i64 %lhs, i64 %rhs, i64* %addr) {
   %sum = add i64 %lhs, %rhs
   store i64 %sum, i64* %addr
   ret void
 }
 
 ; CHECK: Requested regalloc priority advisor analysis could be created. Using default
 ; DEFAULT-NOT: Requested regalloc priority advisor analysis could be created. Using default

llvm/test/CodeGen/MLRegalloc/dev-mode-extra-features-logging.ll

-; REQUIRES: have_tf_api
+; REQUIRES: have_tflite
 ; 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 -o /dev/null -mtriple=x86_64-linux-unknown -regalloc=greedy \
 ; RUN:   -regalloc-enable-advisor=development \
 ; RUN:   -regalloc-training-log=%t1 -tfutils-text-log \

llvm/test/CodeGen/MLRegalloc/dev-mode-log-2-fcts.ll

-; REQUIRES: have_tf_api
+; REQUIRES: have_tflite
 ; REQUIRES: x86_64-linux
 ;
 ; Check that we can log more than 1 function.
 ;
 ; RUN: llc -mtriple=x86_64-linux-unknown -regalloc=greedy -regalloc-enable-advisor=development \
 ; RUN:   -regalloc-training-log=%t1 -tfutils-text-log < %s
 ; RUN: sed -i 's/ \+/ /g' %t1
 ; RUN: sed -i 's/\\n key:/\n key:/g' %t1

llvm/test/CodeGen/MLRegalloc/dev-mode-logging.ll

-; REQUIRES: have_tf_api
+; REQUIRES: have_tflite
 ; REQUIRES: x86_64-linux
 ;
 ; Check that we log correctly, both with a learned policy, and the default policy
 ;
 ; RUN: llc -o /dev/null -mtriple=x86_64-linux-unknown -regalloc=greedy \
 ; RUN:   -regalloc-enable-advisor=development -regalloc-training-log=%t1 \
 ; RUN:   -tfutils-text-log < %S/Inputs/input.ll
 ; RUN: sed -i 's/ \+/ /g' %t1

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

-; REQUIRES: have_tf_api
+; REQUIRES: have_tflite
 ; REQUIRES: x86_64-linux
 ;
 ; Check that we log correctly, both with a learned policy, and the default policy
 ;
 ; RUN: llc -o /dev/null -mtriple=x86_64-linux-unknown -regalloc=greedy \
 ; RUN:   -regalloc-enable-priority-advisor=development \
 ; RUN:   -regalloc-priority-training-log=%t1 -tfutils-text-log \
 ; RUN:   < %S/Inputs/input.ll

llvm/test/CodeGen/MLRegalloc/dev-rel-equivalence.ll

-; REQUIRES: have_tf_api
+; REQUIRES: have_tflite
 ; REQUIRES: llvm_raevict_model_autogenerated
 ; REQUIRES: x86_64-linux
 ;
 ; Check that the same model (==the autogenerated one) produces the same output
 ; regardless of how it's evaluated, which is different from the default
 ;
 ; RUN: llc -mtriple=x86_64-linux-unknown -regalloc=greedy -regalloc-enable-advisor=default \
 ; RUN:   %S/Inputs/input.ll -o %t.default

llvm/test/Transforms/Inline/ML/bounds-checks-rewards.ll

 ; Test behavior when inlining policy grows size out of control.
 ; In all cases, the end result is the same: mandatory inlinings must happen.
 ; Also in all cases, we don't record the mandatory inlining (there's nothing to
 ; learn from it).
 ; However, when we discover we 'trip' over the artificially-low size increase 
 ; factor, we penalize the 'bad' decision.
-; REQUIRES: have_tf_api
+; REQUIRES: have_tflite
 ;
 ; Generate mock model
 ; RUN: rm -rf %t
 ; RUN: rm -rf %t_savedmodel
 ; 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
 ;
 ; When the bounds are very wide ("no bounds"), all inlinings happen.

llvm/test/Transforms/Inline/ML/dev-mode-log-deleted.ll

-; REQUIRES: have_tf_api
+; REQUIRES: have_tflite
 ; RUN: opt -enable-ml-inliner=development -passes=scc-oz-module-inliner \
 ; RUN:     -training-log=- -tfutils-text-log  -S < %s | FileCheck %s 
 
 define i32 @top() {
     %a = call i32 @to_be_deleted()
     %b = call i32 @externally_visible()
     %ret = add i32 %a, %b
     ret i32 %ret

 ; CHECK-NEXT:       }
 ; CHECK-NEXT:     }
 ; CHECK-NEXT:     feature {
 ; CHECK-NEXT:       int64_list {
 ; CHECK-NEXT:         value: 1
 ; CHECK-NEXT:       }
 ; CHECK-NEXT:     }
 ; CHECK-NEXT:   }
 No newline at end of file

llvm/test/Transforms/Inline/ML/development-training-log.ll

 ; Test that we can produce a log if we have or do not have a model, in development mode.
-; REQUIRES: have_tf_api
+; REQUIRES: have_tflite
 ; 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

llvm/test/Transforms/Inline/ML/ml-test-development-mode.ll

 ; The default inliner doesn't elide @adder, it believes it's too costly to inline 
 ; adder into switcher. The ML inliner carries out that inlining, resulting in
 ; a smaller result (part of it is that adder gets elided).
 ;
 ; This test uses Inputs/test-module.ll, as it shares it with a similar test
 ; for the 'release' mode.
 ;
-; REQUIRES: have_tf_api
+; REQUIRES: have_tflite
 ; RUN: rm -rf %t
 ; RUN: rm -rf %t_savedmodel
 ; 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 -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
 ; 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
 ; RUN: opt -passes=scc-oz-module-inliner -enable-ml-inliner=development -training-log=%t.log -tfutils-use-simplelogger -S < %S/Inputs/test-module.ll 2>&1
 ; RUN: %python %S/../../../../lib/Analysis/models/log_reader.py %t.log | FileCheck %s --check-prefix=CHECK-LOG
 
 CHECK-LOG: observation: 0
 CHECK-LOG-NEXT: {{^sroa_savings:}} 0
 CHECK-LOG: {{^cost_estimate:}} -30
 CHECK-LOG: {{^inlining_decision:}} 1
 CHECK-LOG-NEXT: observation: 1
 CHECK-LOG: observation: 6

llvm/test/Transforms/Inline/ML/size-estimator-default.ll

-; REQUIRES: !have_tf_api
+; REQUIRES: !have_tflite
 ; RUN: opt -passes='print<inliner-size-estimator>' -S < %S/Inputs/size-estimator.ll 2>&1 | FileCheck %s
 
 ; CHECK: [InlineSizeEstimatorAnalysis] size estimate for branches: None
 No newline at end of file

llvm/test/Transforms/Inline/ML/size-estimator-training.ll

-; REQUIRES: have_tf_api
+; REQUIRES: have_tflite
 ; RUN: opt -passes='print<inliner-size-estimator>' -S < %S/Inputs/size-estimator.ll 2>&1 | FileCheck %s --check-prefix=DEFAULT
 ; RUN: opt -passes='print<inliner-size-estimator>' -ml-inliner-ir2native-model=%S/../../../../unittests/Analysis/Inputs/ir2native_x86_64_model -S < %S/Inputs/size-estimator.ll 2>&1 | FileCheck %s
 
 ; DEFAULT: [InlineSizeEstimatorAnalysis] size estimate for branches: None
 ; CHECK: [InlineSizeEstimatorAnalysis] size estimate for branches: 28
 No newline at end of file

llvm/test/Transforms/Inline/ML/state-tracking-coro.ll

 ; Based on llvm/test/Transforms/Coroutines/coro-split-02.ll
 ; Corosplit will keep f1 and add 3 more functions.
 ; RUN: opt -passes='default<O1>,print<inline-advisor>' -training-log=/dev/null \
 ; RUN:   -S -enable-ml-inliner=development -keep-inline-advisor-for-printing < %s 2>&1 | FileCheck %s
-; REQUIRES: have_tf_api
+; REQUIRES: have_tflite
 ;
 ; CHECK: [MLInlineAdvisor] Nodes: 4 Edges: 0
 
 %"struct.std::coroutine_handle" = type { ptr }
 %"struct.std::coroutine_handle.0" = type { %"struct.std::coroutine_handle" }
 %"struct.lean_future<int>::Awaiter" = type { i32, %"struct.std::coroutine_handle.0" }
 
 declare ptr @malloc(i64)

llvm/test/Transforms/Inline/ML/state-tracking-scc-splits.ll

 ; Based on llvm/test/Other/cgscc-iterate-function-mutation.ll
 ; RUN: opt -passes='default<O3>,print<inline-advisor>' -training-log=/dev/null \
 ; RUN:   -S -enable-ml-inliner=development -keep-inline-advisor-for-printing < %s 2>&1 | FileCheck %s
-; REQUIRES: have_tf_api
+; REQUIRES: have_tflite
 ;
 ; CHECK: [MLInlineAdvisor] Nodes: 36 Edges: 0
 
 declare void @readnone() nofree nosync readnone
 declare void @unknown()
 declare void @reference_function_pointer(ptr) nofree nosync readnone
 
 ; The @test1_* set of functions checks that when we mutate functions with

llvm/test/Transforms/Inline/ML/state-tracking.ll

 ; RUN: opt -passes='default<O3>,print<inline-advisor>' -training-log=/dev/null \
 ; RUN:   -S -enable-ml-inliner=development -keep-inline-advisor-for-printing < %s 2>&1 | FileCheck %s
-; REQUIRES: have_tf_api
+; REQUIRES: have_tflite
 ;
 ; CHECK: [MLInlineAdvisor] Nodes: 3 Edges: 1
 
 declare i32 @f1()
 
 define i32 @f2() {
     ret i32 1
 }
 
 define i32 @f3() noinline {
     ret i32 2
 }
 
 define i32 @f4() {
     %a = call i32 @f1()
     %b = call i32 @f2()
     %c = call i32 @f3()
     %d = add i32 %a, %b
     %e = add i32 %d, %c
     ret i32 %e
 }
 No newline at end of file

llvm/test/Transforms/Inline/inlining-advisor-default.ll

 ; Check that, in the absence of dependencies, we emit an error message when
 ; trying to use ML-driven inlining.
 ; REQUIRES: !have_tf_aot
-; REQUIRES: !have_tf_api
+; REQUIRES: !have_tflite
 ; RUN: not opt -passes=scc-oz-module-inliner -enable-ml-inliner=development -S < %s 2>&1 | FileCheck %s
 ; RUN: not opt -passes=scc-oz-module-inliner -enable-ml-inliner=release -S < %s 2>&1 | FileCheck %s
 
 declare i64 @f1()
 
 ; CHECK: Could not setup Inlining Advisor for the requested mode and/or options
 No newline at end of file

llvm/test/lit.cfg.py

config.substitutions.append(
         ('%llvmdylib',
          '{}/libLLVM-{}{}'.format(config.llvm_shlib_dir,
                                   config.llvm_dylib_version,
                                   config.llvm_shlib_ext)))
 
 if config.have_tf_aot:
     config.available_features.add("have_tf_aot")
 
-if config.have_tf_api:
-    config.available_features.add("have_tf_api")
+if config.have_tflite:
+    config.available_features.add("have_tflite")
 
 if config.llvm_inliner_model_autogenerated:
     config.available_features.add("llvm_inliner_model_autogenerated")
 
 if config.llvm_raevict_model_autogenerated:
     config.available_features.add("llvm_raevict_model_autogenerated")
 
 def have_cxx_shared_library():

llvm/test/lit.site.cfg.py.in

 config.llvm_dylib_version = "@LLVM_VERSION_MAJOR@@LLVM_VERSION_SUFFIX@"
 config.llvm_host_triple = '@LLVM_HOST_TRIPLE@'
 config.host_arch = "@HOST_ARCH@"
 config.have_opt_viewer_modules = @LLVM_HAVE_OPT_VIEWER_MODULES@
 config.libcxx_used = @LLVM_LIBCXX_USED@
 config.has_plugins = @LLVM_ENABLE_PLUGINS@
 config.linked_bye_extension = @LLVM_BYE_LINK_INTO_TOOLS@
 config.have_tf_aot = @LLVM_HAVE_TF_AOT@
-config.have_tf_api = @LLVM_HAVE_TF_API@
+config.have_tflite = @LLVM_HAVE_TFLITE@
 config.llvm_inliner_model_autogenerated = @LLVM_INLINER_MODEL_AUTOGENERATED@
 config.llvm_raevict_model_autogenerated = @LLVM_RAEVICT_MODEL_AUTOGENERATED@
 config.expensive_checks = @LLVM_ENABLE_EXPENSIVE_CHECKS@
 config.dxil_tests = @LLVM_INCLUDE_DXIL_TESTS@
 config.have_llvm_driver = @LLVM_TOOL_LLVM_DRIVER_BUILD@
 
 import lit.llvm
 lit.llvm.initialize(lit_config, config)
 
 # Let the main config do the real work.
 lit_config.load_config(
     config, os.path.join(config.llvm_src_root, "test/lit.cfg.py"))