From bdbd83b33d2395fe917ccb9f0769f3e73cbd6690 Mon Sep 17 00:00:00 2001 From: Scott Wolchok Date: Tue, 19 Aug 2025 15:45:08 -0700 Subject: [PATCH] Update [ghstack-poisoned] --- examples/models/llama/runner/targets.bzl | 4 ++-- examples/models/llama/targets.bzl | 4 ++-- extension/llm/sampler/targets.bzl | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/models/llama/runner/targets.bzl b/examples/models/llama/runner/targets.bzl index e0a96af85bb..fd298ee628e 100644 --- a/examples/models/llama/runner/targets.bzl +++ b/examples/models/llama/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 _get_operator_lib(aten = False): if aten: @@ -18,7 +18,7 @@ def get_qnn_dependency(): return [] 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( name = "runner" + aten_suffix, diff --git a/examples/models/llama/targets.bzl b/examples/models/llama/targets.bzl index 40b26d69808..66c5dacc8e9 100644 --- a/examples/models/llama/targets.bzl +++ b/examples/models/llama/targets.bzl @@ -1,7 +1,7 @@ -load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "get_oss_build_kwargs", "runtime") +load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "get_aten_mode_options", "get_oss_build_kwargs", "runtime") def define_common_targets(): - for aten in (True, False): + for aten in get_aten_mode_options(): if not runtime.is_oss or not aten: aten_suffix = "_aten" if aten else "" diff --git a/extension/llm/sampler/targets.bzl b/extension/llm/sampler/targets.bzl index 77cac2e7d96..9b7751c19e7 100644 --- a/extension/llm/sampler/targets.bzl +++ b/extension/llm/sampler/targets.bzl @@ -1,7 +1,7 @@ -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 "" runtime.cxx_library(