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

Remove some dead code from gpu_prng.py

Browse Source
This commit is contained in:
Peter Hawkins 2024-10-29 09:29:56 -04:00
parent 5ccfc8d716
commit bee2bc443a
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
jaxlib/gpu_prng.py
View File

@ -13,12 +13,9 @@
# limitations under the License.
from __future__ import annotations
import functools
from functools import partial
import importlib
import itertools
import operator
import jaxlib.mlir.ir as ir
@ -61,8 +58,6 @@ if _hip_prng:
xla_client.register_custom_call_target(_name, _value, platform="ROCM",
api_version=api_version)
_prod = lambda xs: functools.reduce(operator.mul, xs, 1)
def _threefry2x32_lowering(prng, platform: str, keys, data,
length: int | ir.Value | None = None,