From 5179642eb5572b8fbec01dca5e03e2a636e513c2 Mon Sep 17 00:00:00 2001 From: Tzu-Wei Sung Date: Mon, 3 Mar 2025 11:00:42 -0800 Subject: [PATCH] [Mosaic] Rename dep name. PiperOrigin-RevId: 732985217 --- jaxlib/jax.bzl | 2 +- jaxlib/mosaic/BUILD | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/jaxlib/jax.bzl b/jaxlib/jax.bzl index eb016da80..6b6d7e99e 100644 --- a/jaxlib/jax.bzl +++ b/jaxlib/jax.bzl @@ -46,7 +46,7 @@ mosaic_gpu_internal_users = [] mosaic_internal_users = [] pallas_gpu_internal_users = [] pallas_tpu_internal_users = [] -pallas_extension_deps = [] +mosaic_extension_deps = [] jax_internal_export_back_compat_test_util_visibility = [] jax_internal_test_harnesses_visibility = [] diff --git a/jaxlib/mosaic/BUILD b/jaxlib/mosaic/BUILD index 8179ed904..4cc2530dd 100644 --- a/jaxlib/mosaic/BUILD +++ b/jaxlib/mosaic/BUILD @@ -14,7 +14,7 @@ load("@rules_python//python:defs.bzl", "py_library") # 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. -load("//jaxlib:jax.bzl", "pallas_extension_deps") +load("//jaxlib:jax.bzl", "mosaic_extension_deps") licenses(["notice"]) @@ -95,7 +95,7 @@ cc_library( "@xla//xla:shape_util", "@xla//xla:util", "@xla//xla/tsl/platform:errors", - ] + pallas_extension_deps, + ] + mosaic_extension_deps, ) gentbl_cc_library(