Skip to content

Commit 2e8e579

Browse files
authored
Fix remaining instances of for aten in (True, False): (pytorch#13529)
These should be using get_aten_mode_options because ATen mode is disabled in OSS.
1 parent baa41c6 commit 2e8e579

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

examples/models/llama/runner/targets.bzl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
1+
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "get_aten_mode_options", "runtime")
22

33
def _get_operator_lib(aten = False):
44
if aten:
@@ -18,7 +18,7 @@ def get_qnn_dependency():
1818
return []
1919

2020
def define_common_targets():
21-
for aten in (True, False):
21+
for aten in get_aten_mode_options():
2222
aten_suffix = "_aten" if aten else ""
2323
runtime.cxx_library(
2424
name = "runner" + aten_suffix,

examples/models/llama/targets.bzl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1-
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "get_oss_build_kwargs", "runtime")
1+
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "get_aten_mode_options", "get_oss_build_kwargs", "runtime")
22

33
def define_common_targets():
4-
for aten in (True, False):
4+
for aten in get_aten_mode_options():
55
if not runtime.is_oss or not aten:
66
aten_suffix = "_aten" if aten else ""
77

extension/llm/sampler/targets.bzl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1-
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
1+
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "get_aten_mode_options", "runtime")
22

33
def define_common_targets():
4-
for aten in (True, False):
4+
for aten in get_aten_mode_options():
55
aten_suffix = "_aten" if aten else ""
66

77
runtime.cxx_library(

0 commit comments

Comments
 (0)