42 lines
1.4 KiB
Plaintext
42 lines
1.4 KiB
Plaintext
# 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],
|
|
),
|
|
)
|