Path fixes for OSS litert device scripts.

LiteRT-PiperOrigin-RevId: 826070554

Author: LukeBoyer

litert/ats/BUILD

@@ -15,7 +15,7 @@
 load("@rules_cc//cc:cc_library.bzl", "cc_library")
 load("@rules_cc//cc:cc_test.bzl", "cc_test")
 load("//litert/ats:ats.bzl", "litert_define_ats")
-load("//litert/build_common:litert_build_defs.bzl", "litert_test")
+load("//litert/build_common:litert_build_defs.bzl", "commandline_flag_copts", "litert_test")
 load("//litert/integration_test:litert_device.bzl", "litert_device_test")
 load("//litert/integration_test:litert_device_script.bzl", "make_download_model_provider")
 
@@ -31,7 +31,7 @@ package(
 litert_test(
     name = "ats",
     srcs = ["ats.cc"],
-    copts = ["-DGOOGLE_COMMANDLINEFLAGS_FULL_API=1"],
+    copts = commandline_flag_copts(),
     linkstatic = True,
     no_main = True,
     tags = [
@@ -62,6 +62,7 @@ cc_library(
     testonly = True,
     srcs = ["configure.cc"],
     hdrs = ["configure.h"],
+    copts = commandline_flag_copts(),
     deps = [
         ":common",
         "//litert/c:litert_common",
@@ -199,6 +200,7 @@ litert_device_test(
 cc_test(
     name = "check_ats",
     srcs = ["check_ats.cc"],
+    copts = commandline_flag_copts(),
     data = [
         "//litert/test:testdata/simple_model.tflite",
         "//litert/vendors/examples:example_dispatch_so",
@@ -346,9 +348,8 @@ litert_define_ats(
     backend = "example",
     compile_only_suffix = "_aot",
     do_register = [
-        # "sub.*f32",
-        # "mul.*f32",
-        ".*ExtraModel.*",
+        "sub.*f32",
+        "mul.*f32",
     ],
     extra_flags = ["--limit=1"],
     jit_suffix = "",

litert/ats/ats_aot.sh

@@ -14,7 +14,7 @@
 
 #!/bin/bash
 
-source "${0%.*}_lib.sh" || exit 1
+source "${0%.sh}_lib.sh" || exit 1
 
 # TODO: Unify workdirs with other scripts.
 readonly models_out="/tmp/litert_extras/ats"

litert/build_common/litert_build_defs.bzl

@@ -116,6 +116,12 @@ def _valid_so_name(name):
 def _make_target_ref(name):
     return ":{}".format(name)
 
+def commandline_flag_copts():
+    return select({
+        "@org_tensorflow//tensorflow:android": ["-DGOOGLE_COMMANDLINEFLAGS_FULL_API=1"] + if_oss(["-DABSL_FLAGS_STRIP_NAMES=0"]),
+        "//conditions:default": [],
+    })
+
 ####################################################################################################
 # Explicitly Link System Libraries ("ungrte")
 

litert/integration_test/BUILD

@@ -16,6 +16,7 @@
 load("@rules_cc//cc:cc_binary.bzl", "cc_binary")
 load("@rules_cc//cc:cc_library.bzl", "cc_library")
 load("@rules_cc//cc:cc_test.bzl", "cc_test")
+load("//litert/build_common:litert_build_defs.bzl", "if_oss")
 load("//litert/build_common:special_rule.bzl", "litert_android_linkopts")
 load("//litert/integration_test:litert_device.bzl", "litert_device_exec", "litert_device_test", "litert_integration_test")
 load("//litert/integration_test:litert_device_common.bzl", "device_rlocation", "get_libs")
@@ -416,7 +417,7 @@ platform(
 cc_binary(
     name = "dummy_binary",
     srcs = ["dummy_binary.cc"],
-    copts = ["-DGOOGLE_COMMANDLINEFLAGS_FULL_API=1"],
+    copts = ["-DGOOGLE_COMMANDLINEFLAGS_FULL_API=1"] + if_oss(["-DABSL_FLAGS_STRIP_NAMES=0"]),
     deps = [
         "//litert/core:filesystem",
         "@com_google_absl//absl/container:flat_hash_set",
@@ -510,7 +511,7 @@ litert_device_test(
     name = "tools_test",
     srcs = ["smoketests/tools_test.cc"],
     backend_id = "cpu",
-    copts = ["-DGOOGLE_COMMANDLINEFLAGS_FULL_API=1"],
+    copts = ["-DGOOGLE_COMMANDLINEFLAGS_FULL_API=1"] + if_oss(["-DABSL_FLAGS_STRIP_NAMES=0"]),
     data = ["//litert/integration_test/models:single_op"],
     exec_args = ["--models_dir={}".format(device_rlocation("//litert/integration_test/models:single_op"))],
     deps = [

litert/integration_test/device_script_template.sh

@@ -17,7 +17,7 @@
 # Shell library for working with data and executable files from bzl between host
 # and device. Meant to be templated via litert_device_script.bzl.
 
-source third_party/odml/litert/litert/integration_test/device_script_common.sh || exit 1
+source litert/integration_test/device_script_common.sh || exit 1
 
 # Root of runfiles on the device.
 device_runfiles_root="/data/local/tmp/runfiles"

litert/integration_test/device_script_test.sh

@@ -14,7 +14,7 @@
 
 #!/bin/bash
 
-source ${0#.*}_lib.sh || exit 1
+source ${0%.sh}_lib.sh || exit 1
 
 check_host=""
 check_device=""

litert/integration_test/download_model_provider.sh

@@ -14,7 +14,7 @@
 
 #!/bin/bash
 
-source third_party/odml/litert/litert/integration_test/device_script_common.sh || exit 1
+source litert/integration_test/device_script_common.sh || exit 1
 
 # TODO: Unify workdirs with other scripts.
 readonly work_dir="/tmp/litert_extras"

litert/integration_test/mobile_install.sh

@@ -14,7 +14,7 @@
 
 #!/bin/bash
 
-source "${0%.*}_lib.sh" || exit 1
+source "${0%.sh}_lib.sh" || exit 1
 
 extra_args=("${@:1}")
 d_bin=$(device_bin)