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

Raise rather than return error

Browse Source
This commit is contained in:
Jake VanderPlas 2024-12-17 11:20:55 -08:00
parent 0fa541972e
commit 2518c6233e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/_src/core.py
View File

@ -1669,7 +1669,7 @@ def get_sharding(sharding, ndim):
context_mesh = mesh_lib.get_abstract_mesh()
if not context_mesh:
return RuntimeError("Please set the mesh via `jax.set_mesh` API.")
raise RuntimeError("Please set the mesh via `jax.set_mesh` API.")
assert sharding is None
return NamedSharding(context_mesh, P(*[None] * ndim))