From 7a6747fab05148be18f037b9cc6c0ce069ccf390 Mon Sep 17 00:00:00 2001 From: Scott Wolchok Date: Tue, 19 Aug 2025 15:37:57 -0700 Subject: [PATCH] Update [ghstack-poisoned] --- .ci/scripts/unittest-buck2.sh | 6 +++--- extension/llm/runner/io_manager/targets.bzl | 4 ++-- extension/llm/runner/targets.bzl | 4 ++-- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/.ci/scripts/unittest-buck2.sh b/.ci/scripts/unittest-buck2.sh index 658fafdfcca..f748be62ac1 100755 --- a/.ci/scripts/unittest-buck2.sh +++ b/.ci/scripts/unittest-buck2.sh @@ -12,9 +12,9 @@ set -eux buck2 query "//backends/apple/... + //backends/example/... + \ //backends/mediatek/... + //backends/transforms/... + \ //backends/xnnpack/... + //configurations/... + //extension/flat_tensor: + \ -//kernels/aten/... + //kernels/optimized/... + //kernels/portable/... + \ -//kernels/quantized/... + //kernels/test/... + //runtime/... + //schema/... \ -+ //test/... + //util/..." +//extension/llm/runner: + //kernels/aten/... + //kernels/optimized/... + \ +//kernels/portable/... + //kernels/quantized/... + //kernels/test/... + \ +//runtime/... + //schema/... + //test/... + //util/..." # TODO: optimized ops are unbuildable because they now use ATen; put # them back after we can use PyTorch in OSS buck. diff --git a/extension/llm/runner/io_manager/targets.bzl b/extension/llm/runner/io_manager/targets.bzl index e538572c51b..ef93d541098 100644 --- a/extension/llm/runner/io_manager/targets.bzl +++ b/extension/llm/runner/io_manager/targets.bzl @@ -1,8 +1,8 @@ -load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime") +load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "get_aten_mode_options", "runtime") def define_common_targets(): - for aten in (True, False): + for aten in get_aten_mode_options(): aten_suffix = "_aten" if aten else "" # Interface for IOManager. No concrete impl from this dep. diff --git a/extension/llm/runner/targets.bzl b/extension/llm/runner/targets.bzl index a6c17f3037c..05f05ac6fad 100644 --- a/extension/llm/runner/targets.bzl +++ b/extension/llm/runner/targets.bzl @@ -1,4 +1,4 @@ -load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime") +load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "get_aten_mode_options", "runtime") def define_common_targets(): runtime.cxx_library( @@ -32,7 +32,7 @@ def define_common_targets(): ], ) - for aten in (True, False): + for aten in get_aten_mode_options(): aten_suffix = "_aten" if aten else "" runtime.cxx_library(