.bant-macros

# Copyright 2026 The OpenXLA Authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ==============================================================================
#
# Bant macro expansions for XLA custom rules.
#
# Syntax: https://github.com/hzeller/bant?tab=readme-ov-file#macros.
#
# See developer guide on how to use Bant tool for making sure that your targets
# list all required dependencies:
# https://openxla.org/xla/developer_environment#remove_unused_dependencies_from_build_files.

# XLA tests
xla_test = bant_forward_args(cc_test())
xla_cc_test = bant_forward_args(cc_test())

# tf_proto_library() wraps proto_library() and cc_proto_library().
# It generates a proto_library with the given name and a cc_proto_library
# with a "_cc" suffix that provides .pb.h headers.
tf_proto_library = (
    proto_library(
        name = name,
        srcs = srcs,
        deps = deps,
    ),
    cc_proto_library(
        name = name + "_cc",
        deps = [":" + name],
    ),
)