mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Mosaic] Rename dep name.

Browse Source 
PiperOrigin-RevId: 732985217
This commit is contained in:
Tzu-Wei Sung 2025-03-03 11:00:42 -08:00 committed by jax authors
parent 1a57fdf704
commit 5179642eb5
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jaxlib/jax.bzl
View File

@ -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 = []

4
jaxlib/mosaic/BUILD
View File

@ -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(