local_only.bzl•2.47 kB
# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# This source code is licensed under both the MIT license found in the
# LICENSE-MIT file in the root directory of this source tree and the Apache
# License, Version 2.0 found in the LICENSE-APACHE file in the root directory
# of this source tree.
load("@prelude//cxx:cxx_context.bzl", "get_cxx_toolchain_info")
load(
"@prelude//cxx:cxx_toolchain_types.bzl",
"CxxToolchainInfo", # @unused Used as type
)
load("@prelude//linking:execution_preference.bzl", "LinkExecutionPreference", "get_link_execution_preference")
load("@prelude//python:toolchain.bzl", "PythonToolchainInfo")
def link_cxx_binary_locally(ctx: AnalysisContext, cxx_toolchain: [CxxToolchainInfo, None] = None) -> bool:
# Core tools are linked on RE because they are
# a) small enough to do so and
# b) don't get build stamping so they do cache correctly.
if _is_core_tool(ctx):
return False
return _cxx_toolchain_sets_link_binaries_locally(ctx, cxx_toolchain)
def get_resolved_cxx_binary_link_execution_preference(
ctx: AnalysisContext,
links: list[Label],
force_full_hybrid_if_capable: bool,
cxx_toolchain: [CxxToolchainInfo, None] = None) -> LinkExecutionPreference:
if force_full_hybrid_if_capable:
return LinkExecutionPreference("full_hybrid")
# Core tools can be linked on RE because they are
# a) small enough to do so and
# b) don't get build stamping so they do cache correctly.
if _is_core_tool(ctx):
return LinkExecutionPreference("any")
# Check if the toolchain has a preference.
if _cxx_toolchain_sets_link_binaries_locally(ctx, cxx_toolchain):
return LinkExecutionPreference("local")
# Else use the preference on the target.
return get_link_execution_preference(ctx, links)
def package_python_locally(ctx: AnalysisContext, python_toolchain: PythonToolchainInfo) -> bool:
if _is_core_tool(ctx) or getattr(ctx.attrs, "_package_remotely", False):
return False
return python_toolchain.build_standalone_binaries_locally
def _is_core_tool(ctx: AnalysisContext) -> bool:
return "is_core_tool" in getattr(ctx.attrs, "labels", [])
def _cxx_toolchain_sets_link_binaries_locally(ctx: AnalysisContext, cxx_toolchain: [CxxToolchainInfo, None]) -> bool:
if not cxx_toolchain:
cxx_toolchain = get_cxx_toolchain_info(ctx)
return cxx_toolchain.linker_info.link_binaries_locally