[CI] Fix build (#3194)

veblush · web-flow · commit ba753cef3e6e · 2025-09-19T20:48:22.000Z
- Upgraded the Bazel BuildTool to the latest version (8.2.1).
- Updated the `tflm-ci` Docker image and pushed the new tag (0.6.1) to the `ghcr.io/tflm-bot/tflm-ci registry`.
- Updated the WORKSPACE to load `rules_cc` and `rules_shell` explicitly
- Ran `buildifier` to ensure all BUILD files to have all the fixes.

BUG=Clean-up
diff --git a/BUILD b/BUILD
@@ -1,3 +1,4 @@
+load("@bazel_skylib//rules:common_settings.bzl", "bool_flag")
 load("@hedron_compile_commands//:refresh_compile_commands.bzl", "refresh_compile_commands")
 
 # `bazel run` this target to generate compile_commands.json, which can be used
@@ -8,8 +9,6 @@ refresh_compile_commands(
     targets = ["//..."],
 )
 
-load("@bazel_skylib//rules:common_settings.bzl", "bool_flag")
-
 bool_flag(
     name = "with_compression",
     build_setting_default = False,
diff --git a/WORKSPACE b/WORKSPACE
@@ -19,6 +19,13 @@ load("//tensorflow:workspace.bzl", "workspace")
 
 workspace()
 
+# init for rules_shell
+load("@rules_shell//shell:repositories.bzl", "rules_shell_dependencies", "rules_shell_toolchains")
+
+rules_shell_dependencies()
+
+rules_shell_toolchains()
+
 load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 
 # compile_commands.json generator
diff --git a/ci/Dockerfile.micro b/ci/Dockerfile.micro
@@ -28,7 +28,7 @@ COPY ci/install_qemu.sh ./
 RUN ./install_qemu.sh /tmp/qemu_install
 
 # This stage is the final CI container.
-FROM python:3.10-bullseye as tflm-ci
+FROM python:3.10-bullseye AS tflm-ci
 
 RUN apt-get update
 
diff --git a/ci/install_buildifier.sh b/ci/install_buildifier.sh
@@ -16,13 +16,13 @@
 
 set -e
 # Download buildifier.
-wget https://github.com/bazelbuild/buildtools/releases/download/4.2.3/buildifier-linux-amd64
+wget https://github.com/bazelbuild/buildtools/releases/download/v8.2.1/buildifier-linux-amd64
 mv buildifier-linux-amd64 buildifier
 chmod +x buildifier
 sudo mv buildifier /usr/local/bin/.
 
 # Download buildozer.
-wget https://github.com/bazelbuild/buildtools/releases/download/4.2.3/buildozer-linux-amd64
+wget https://github.com/bazelbuild/buildtools/releases/download/v8.2.1/buildozer-linux-amd64
 mv buildozer-linux-amd64 buildozer
 chmod +x buildozer
 sudo mv buildozer /usr/local/bin/.
diff --git a/codegen/examples/hello_world/BUILD b/codegen/examples/hello_world/BUILD
@@ -1,3 +1,4 @@
+load("@rules_cc//cc:cc_binary.bzl", "cc_binary")
 load("//codegen:build_def.bzl", "tflm_inference_library")
 
 package(default_visibility = ["//visibility:public"])
diff --git a/python/tests/BUILD b/python/tests/BUILD
@@ -1,3 +1,5 @@
+load("@rules_cc//cc:cc_test.bzl", "cc_test")
+
 cc_test(
     name = "cc_deps_link_test",
     size = "small",
diff --git a/python/tflite_micro/BUILD b/python/tflite_micro/BUILD
@@ -1,18 +1,19 @@
 load("@bazel_skylib//rules:common_settings.bzl", "string_flag")
-load("@rules_python//python:defs.bzl", "py_library", "py_test")
 load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
-load("//python:py_namespace.bzl", "py_namespace")
-load("//tools:expand_stamp_vars.bzl", "expand_stamp_vars")
+load("@rules_python//python:defs.bzl", "py_library", "py_test")
 load("@rules_python//python:packaging.bzl", "py_package", "py_wheel")
+load("@rules_shell//shell:sh_test.bzl", "sh_test")
 load("@tflm_pip_deps//:requirements.bzl", "requirement")
-load(
-    "//tensorflow/lite/micro:build_def.bzl",
-    "tflm_cc_library",
-)
+load("//python:py_namespace.bzl", "py_namespace")
 load(
     "//tensorflow:extra_rules.bzl",
     "tflm_python_op_resolver_friends",
 )
+load(
+    "//tensorflow/lite/micro:build_def.bzl",
+    "tflm_cc_library",
+)
+load("//tools:expand_stamp_vars.bzl", "expand_stamp_vars")
 
 package(
     features = ["-layering_check"],
diff --git a/python/tflite_micro/signal/BUILD b/python/tflite_micro/signal/BUILD
@@ -1,7 +1,8 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
 load("@rules_python//python:defs.bzl", "py_library", "py_test")
+load("@tflm_pip_deps//:requirements.bzl", "requirement")
 load("//python/tflite_micro/signal:tflm_signal.bzl", "py_tflm_signal_library")
 load("//tensorflow:extra_rules.bzl", "tflm_signal_friends")
-load("@tflm_pip_deps//:requirements.bzl", "requirement")
 
 package(
     default_visibility = [":__subpackages__"],
diff --git a/python/tflite_micro/signal/tflm_signal.bzl b/python/tflite_micro/signal/tflm_signal.bzl
@@ -1,5 +1,7 @@
 """Build rule for wrapping a custom TF OP from .cc to python."""
 
+load("@rules_cc//cc:cc_binary.bzl", "cc_binary")
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
 load("@rules_python//python:defs.bzl", "py_library")
 
 # TODO(b/286890280): refactor to be more generic build target for any custom OP
@@ -37,7 +39,7 @@ def py_tflm_signal_library(
     if cc_op_defs:
         binary_name = "ops/_" + name + ".so"
         library_name = name + "_cc"
-        native.cc_library(
+        cc_library(
             name = library_name,
             copts = select({
                 "//conditions:default": ["-pthread"],
@@ -49,8 +51,7 @@ def py_tflm_signal_library(
                 ["@tensorflow_cc_deps//:cc_library"] +
                 select({"//conditions:default": []}),
         )
-
-        native.cc_binary(
+        cc_binary(
             name = binary_name,
             copts = select({
                 "//conditions:default": ["-pthread"],
@@ -80,7 +81,7 @@ def tflm_signal_kernel_library(
         deps = [],
         copts = [],
         alwayslink = 1):
-    native.cc_library(
+    cc_library(
         name = name,
         srcs = srcs,
         hdrs = hdrs,
diff --git a/python/tflite_micro/signal/utils/BUILD b/python/tflite_micro/signal/utils/BUILD
@@ -1,7 +1,8 @@
+load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
+
 # Signal python utilities.
 load("@rules_python//python:defs.bzl", "py_library", "py_test")
 load("@tflm_pip_deps//:requirements.bzl", "requirement")
-load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
 
 package(
     default_visibility = [
diff --git a/signal/micro/kernels/BUILD b/signal/micro/kernels/BUILD
@@ -1,3 +1,5 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
+load("@rules_cc//cc:cc_test.bzl", "cc_test")
 load(
     "//tensorflow/lite/micro:build_def.bzl",
     "tflm_cc_library",
diff --git a/signal/src/BUILD b/signal/src/BUILD
@@ -1,3 +1,5 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
+
 package(
     default_visibility = ["//signal:__subpackages__"],
     licenses = ["notice"],
diff --git a/signal/src/kiss_fft_wrappers/BUILD b/signal/src/kiss_fft_wrappers/BUILD
@@ -1,3 +1,5 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
+
 package(
     default_visibility = ["//signal/src:__subpackages__"],
     licenses = ["notice"],
diff --git a/signal/testdata/BUILD b/signal/testdata/BUILD
@@ -1,5 +1,7 @@
 # Test data for signal tests. This is used to store large arrays which would make tests less readable.
 
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
+
 package(
     default_visibility = ["//signal/micro/kernels:__subpackages__"],
     licenses = ["notice"],
diff --git a/tensorflow/compiler/mlir/lite/core/c/BUILD b/tensorflow/compiler/mlir/lite/core/c/BUILD
@@ -1,3 +1,4 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
 load("//tensorflow/lite:build_def.bzl", "tflite_copts")
 
 package(
diff --git a/tensorflow/compiler/mlir/lite/kernels/internal/BUILD b/tensorflow/compiler/mlir/lite/kernels/internal/BUILD
@@ -1,3 +1,5 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
+
 package(
     default_visibility = ["//visibility:public"],
     licenses = ["notice"],
diff --git a/tensorflow/compiler/mlir/lite/schema/BUILD b/tensorflow/compiler/mlir/lite/schema/BUILD
@@ -1,4 +1,5 @@
 load("@flatbuffers//:build_defs.bzl", "flatbuffer_cc_library")
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
 
 package(
     default_visibility = [
diff --git a/tensorflow/lite/BUILD b/tensorflow/lite/BUILD
@@ -1,3 +1,5 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
+
 package(
     default_visibility = ["//visibility:public"],
     licenses = ["notice"],
diff --git a/tensorflow/lite/c/BUILD b/tensorflow/lite/c/BUILD
@@ -1,3 +1,4 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
 load(
     "//tensorflow/lite:build_def.bzl",
     "tflite_copts",
diff --git a/tensorflow/lite/core/BUILD b/tensorflow/lite/core/BUILD
@@ -1,3 +1,5 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
+
 cc_library(
     name = "macros",
     hdrs = ["macros.h"],
diff --git a/tensorflow/lite/core/c/BUILD b/tensorflow/lite/core/c/BUILD
@@ -1,3 +1,4 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
 load(
     "//tensorflow/lite:build_def.bzl",
     "tflite_copts",
diff --git a/tensorflow/lite/experimental/microfrontend/lib/BUILD b/tensorflow/lite/experimental/microfrontend/lib/BUILD
@@ -1,3 +1,6 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
+load("@rules_cc//cc:cc_test.bzl", "cc_test")
+
 # Library for generating feature vectors from audio data
 package(
     default_visibility = ["//visibility:public"],
diff --git a/tensorflow/lite/kernels/BUILD b/tensorflow/lite/kernels/BUILD
@@ -1,3 +1,4 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
 load("//tensorflow/lite:build_def.bzl", "tflite_copts")
 load(
     "//tensorflow/lite/micro:build_def.bzl",
diff --git a/tensorflow/lite/kernels/internal/BUILD b/tensorflow/lite/kernels/internal/BUILD
@@ -1,3 +1,4 @@
+load("@rules_cc//cc:cc_library.bzl", "cc_library")
 load("//tensorflow/lite:build_def.bzl", "tflite_copts")
 load(
     "//tensorflow/lite/micro:build_def.bzl",
diff --git a/tensorflow/lite/micro/build_def.bzl b/tensorflow/lite/micro/build_def.bzl
@@ -1,5 +1,9 @@
 """TfLite Micro BUILD options."""
 
+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")
+
 def tflm_copts():
     """Returns the default copts for targets in TFLM.
 
@@ -45,21 +49,21 @@ def tflm_defines():
     return defines
 
 def tflm_cc_binary(copts = tflm_copts(), defines = tflm_defines(), **kwargs):
-    native.cc_binary(
+    cc_binary(
         copts = copts,
         defines = defines,
         **kwargs
     )
 
 def tflm_cc_library(copts = tflm_copts(), defines = tflm_defines(), **kwargs):
-    native.cc_library(
+    cc_library(
         copts = copts,
         defines = defines,
         **kwargs
     )
 
 def tflm_cc_test(copts = tflm_copts(), defines = tflm_defines(), **kwargs):
-    native.cc_test(
+    cc_test(
         copts = copts,
         defines = defines,
         **kwargs
diff --git a/tensorflow/lite/micro/compression/BUILD b/tensorflow/lite/micro/compression/BUILD
@@ -1,15 +1,16 @@
-load(
-    "//tensorflow/lite/micro:build_def.bzl",
-    "tflm_cc_library",
-    "tflm_cc_test",
-)
 load(
     "@flatbuffers//:build_defs.bzl",
     "flatbuffer_cc_library",
     "flatbuffer_py_library",
 )
 load("@rules_python//python:defs.bzl", "py_binary", "py_library", "py_test")
+load("@rules_shell//shell:sh_test.bzl", "sh_test")
 load("@tflm_pip_deps//:requirements.bzl", "requirement")
+load(
+    "//tensorflow/lite/micro:build_def.bzl",
+    "tflm_cc_library",
+    "tflm_cc_test",
+)
 load("view_tests.bzl", "generate_view_tests")
 
 package(
diff --git a/tensorflow/lite/micro/compression/view_tests.bzl b/tensorflow/lite/micro/compression/view_tests.bzl
@@ -1,3 +1,5 @@
+load("@rules_python//python:defs.bzl", "py_test")
+
 def generate_view_tests(targets):
     """Generates py_test targets for each target's path and a test_suite to
     group them.
@@ -10,8 +12,7 @@ def generate_view_tests(targets):
         # Create a test name from the last component of the target name
         short_name = target.split(":")[-1] if ":" in target else target.split("/")[-1]
         test_name = "view_test_{}".format(short_name.replace(".", "_"))
-
-        native.py_test(
+        py_test(
             name = test_name,
             srcs = ["view_test.py"],
             args = ["$(location {})".format(target)],
diff --git a/tensorflow/lite/micro/examples/memory_footprint/BUILD b/tensorflow/lite/micro/examples/memory_footprint/BUILD
@@ -1,3 +1,4 @@
+load("@rules_cc//cc:cc_binary.bzl", "cc_binary")
 load(
     "//tensorflow/lite/micro:build_def.bzl",
     "generate_cc_arrays",
diff --git a/tensorflow/lite/micro/examples/person_detection/utils/BUILD b/tensorflow/lite/micro/examples/person_detection/utils/BUILD
@@ -1,5 +1,5 @@
-load("@tflm_pip_deps//:requirements.bzl", "requirement")
 load("@rules_python//python:defs.bzl", "py_binary", "py_library", "py_test")
+load("@tflm_pip_deps//:requirements.bzl", "requirement")
 
 package(
     features = ["-layering_check"],
diff --git a/tensorflow/lite/micro/kernels/BUILD b/tensorflow/lite/micro/kernels/BUILD
@@ -1,11 +1,4 @@
 load("@bazel_skylib//rules:common_settings.bzl", "string_flag")
-load(
-    "//tensorflow/lite/micro:build_def.bzl",
-    "tflm_cc_library",
-    "tflm_cc_test",
-    "tflm_copts",
-    "tflm_kernel_cc_library",
-)
 load(
     "//tensorflow:extra_rules.bzl",
     "tflm_kernel_friends",
@@ -15,6 +8,13 @@ load(
     "xtensa_hifi_5_config",
     "xtensa_vision_p6_config",
 )
+load(
+    "//tensorflow/lite/micro:build_def.bzl",
+    "tflm_cc_library",
+    "tflm_cc_test",
+    "tflm_copts",
+    "tflm_kernel_cc_library",
+)
 
 package(
     features = [
diff --git a/tensorflow/lite/micro/kernels/test_data_generation/BUILD b/tensorflow/lite/micro/kernels/test_data_generation/BUILD
@@ -1,3 +1,5 @@
+load("@rules_cc//cc:cc_binary.bzl", "cc_binary")
+
 package(
     # Disabling layering_check because of http://b/177257332
     features = ["-layering_check"],
diff --git a/tensorflow/lite/micro/memory_planner/BUILD b/tensorflow/lite/micro/memory_planner/BUILD
@@ -1,3 +1,4 @@
+load("@rules_cc//cc:cc_test.bzl", "cc_test")
 load(
     "//tensorflow/lite/micro:build_def.bzl",
     "tflm_cc_library",
diff --git a/tensorflow/lite/micro/python/interpreter/src/BUILD b/tensorflow/lite/micro/python/interpreter/src/BUILD
@@ -1,11 +1,11 @@
-load(
-    "//tensorflow/lite/micro:build_def.bzl",
-    "tflm_cc_library",
-)
 load(
     "//tensorflow:extra_rules.bzl",
     "tflm_python_op_resolver_friends",
 )
+load(
+    "//tensorflow/lite/micro:build_def.bzl",
+    "tflm_cc_library",
+)
 
 package(
     features = ["-layering_check"],
diff --git a/tensorflow/lite/micro/python/tflite_size/src/BUILD b/tensorflow/lite/micro/python/tflite_size/src/BUILD
@@ -1,5 +1,5 @@
-load("@rules_python//python:defs.bzl", "py_binary", "py_library")
 load("@pybind11_bazel//:build_defs.bzl", "pybind_extension", "pybind_library")
+load("@rules_python//python:defs.bzl", "py_binary", "py_library")
 
 package(
     default_visibility = ["//visibility:public"],
diff --git a/tensorflow/lite/micro/testing/BUILD b/tensorflow/lite/micro/testing/BUILD
@@ -1,14 +1,14 @@
 load("@rules_python//python:defs.bzl", "py_binary", "py_library")
 load("@tflm_pip_deps//:requirements.bzl", "requirement")
+load(
+    "//tensorflow:extra_rules.bzl",
+    "tflm_kernel_friends",
+)
 load(
     "//tensorflow/lite/micro:build_def.bzl",
     "tflm_cc_library",
     "tflm_cc_test",
 )
-load(
-    "//tensorflow:extra_rules.bzl",
-    "tflm_kernel_friends",
-)
 
 package(
     # Disabling layering_check because of http://b/177257332
diff --git a/tensorflow/lite/micro/tools/BUILD b/tensorflow/lite/micro/tools/BUILD
diff --git a/tensorflow/lite/python/BUILD b/tensorflow/lite/python/BUILD
diff --git a/tensorflow/lite/schema/BUILD b/tensorflow/lite/schema/BUILD
diff --git a/tensorflow/lite/tools/BUILD b/tensorflow/lite/tools/BUILD
diff --git a/tensorflow/workspace.bzl b/tensorflow/workspace.bzl
diff --git a/third_party/flatbuffers/BUILD.oss b/third_party/flatbuffers/BUILD.oss
diff --git a/third_party/flatbuffers/build_defs.bzl b/third_party/flatbuffers/build_defs.bzl
diff --git a/third_party/kissfft/BUILD.bazel b/third_party/kissfft/BUILD.bazel

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
	`1`	`+load("@rules_cc//cc:cc_binary.bzl", "cc_binary")`
`1`	`2`	`load("//codegen:build_def.bzl", "tflm_inference_library")`
`2`	`3`
`3`	`4`	`package(default_visibility = ["//visibility:public"])`
Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,5 @@`
	`1`	`+load("@rules_cc//cc:cc_test.bzl", "cc_test")`
	`2`	`+`
`1`	`3`	`cc_test(`
`2`	`4`	`name = "cc_deps_link_test",`
`3`	`5`	`size = "small",`