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(