From 80686fa0afd469b19d144f30152c783b67fc0a20 Mon Sep 17 00:00:00 2001 From: Scott Wolchok Date: Thu, 27 Mar 2025 14:37:40 -0700 Subject: [PATCH] Update [ghstack-poisoned] --- extension/parallel/targets.bzl | 2 +- kernels/optimized/cpu/targets.bzl | 2 +- kernels/optimized/lib_defs.bzl | 2 +- kernels/portable/cpu/util/targets.bzl | 8 ++++---- runtime/kernel/targets.bzl | 4 ++-- 5 files changed, 9 insertions(+), 9 deletions(-) diff --git a/extension/parallel/targets.bzl b/extension/parallel/targets.bzl index 3a2d6f354ef..dbfb3ff160c 100644 --- a/extension/parallel/targets.bzl +++ b/extension/parallel/targets.bzl @@ -17,6 +17,6 @@ def define_common_targets(): "@EXECUTORCH_CLIENTS", ], deps = [ - "//executorch/extension/threadpool:threadpool", + "//executorch/runtime/kernel:thread_parallel_interface", ], ) diff --git a/kernels/optimized/cpu/targets.bzl b/kernels/optimized/cpu/targets.bzl index 8dadd5a49e6..017dff8a127 100644 --- a/kernels/optimized/cpu/targets.bzl +++ b/kernels/optimized/cpu/targets.bzl @@ -99,8 +99,8 @@ _OPTIMIZED_ATEN_OPS = ( op_target( name = "op_where", deps = [ - "//executorch/extension/threadpool:threadpool", "//executorch/kernels/portable/cpu/util:elementwise_util", + "//executorch/runtime/kernel:thread_parallel_interface", ], ), ) diff --git a/kernels/optimized/lib_defs.bzl b/kernels/optimized/lib_defs.bzl index 85886365a01..6e884457e35 100644 --- a/kernels/optimized/lib_defs.bzl +++ b/kernels/optimized/lib_defs.bzl @@ -232,9 +232,9 @@ def define_libs(is_fbcode=False): "DEFAULT": [], }) + LIBBLAS_DEPS, exported_deps = [ - "//executorch/extension/threadpool:threadpool", "//executorch/kernels/optimized:libutils", "//executorch/runtime/core/exec_aten:lib", + "//executorch/runtime/kernel:thread_parallel_interface", ], **get_apple_framework_deps_kwargs(is_fbcode), ) diff --git a/kernels/portable/cpu/util/targets.bzl b/kernels/portable/cpu/util/targets.bzl index e756a9bf282..a623b9d4d7a 100644 --- a/kernels/portable/cpu/util/targets.bzl +++ b/kernels/portable/cpu/util/targets.bzl @@ -12,7 +12,6 @@ def define_common_targets(): runtime.cxx_library( name = "all_deps", deps = [ - "//executorch/extension/threadpool:threadpool", "//executorch/kernels/portable/cpu/util:functional_util", "//executorch/kernels/portable/cpu/util:broadcast_util", "//executorch/kernels/portable/cpu/util:kernel_ops_util", @@ -33,6 +32,7 @@ def define_common_targets(): "//executorch/kernels/portable/cpu/util:slice_util", "//executorch/kernels/portable/cpu/util:elementwise_util", "//executorch/kernels/portable/cpu/util:upsample_util", + "//executorch/runtime/kernel:thread_parallel_interface", ], visibility = ["//executorch/...", "@EXECUTORCH_CLIENTS"], ) @@ -111,7 +111,7 @@ def define_common_targets(): ":broadcast_util", ":dtype_util", "//executorch/runtime/kernel:kernel_runtime_context", - "//executorch/extension/threadpool:threadpool", + "//executorch/runtime/kernel:thread_parallel_interface", ], deps = [ "//executorch/kernels/portable/cpu:scalar_utils", @@ -245,7 +245,7 @@ def define_common_targets(): srcs = [], exported_headers = ["functional_util.h"], exported_deps = [ - "//executorch/extension/threadpool:threadpool", + "//executorch/runtime/kernel:thread_parallel_interface", ], deps = [ "//executorch/runtime/kernel:kernel_includes", @@ -319,7 +319,7 @@ def define_common_targets(): "//executorch/runtime/core/exec_aten/util:tensor_util{}".format(suffix), ], exported_deps = [ - "//executorch/extension/threadpool:threadpool", + "//executorch/runtime/kernel:thread_parallel_interface", ], exported_preprocessor_flags = ["-DUSE_ATEN_LIB"] if aten_mode else [], visibility = [ diff --git a/runtime/kernel/targets.bzl b/runtime/kernel/targets.bzl index b6aa9d7a95e..0726752d3dd 100644 --- a/runtime/kernel/targets.bzl +++ b/runtime/kernel/targets.bzl @@ -59,9 +59,9 @@ def define_common_targets(): "//executorch/runtime/core/portable_type/c10/c10:c10", "//executorch/runtime/platform:platform", ], - # Don't depend on this target, depend on //executorch/extension/threadpool:threadpool. visibility = [ - "//executorch/extension/threadpool/...", + "//executorch/...", + "@EXECUTORCH_CLIENTS", ], )