[go: nahoru, domu]
Skip to content

Commit

Permalink
Ported threefry2x32 for GPU to the typed XLA FFI
Browse files Browse the repository at this point in the history 
This allows lowering of threefry2x32 for GPU even on a machine without GPUs.

For the next 3 weeks, we only use the new custom call implementation if
we are not in "export" mode, and if we use a new jaxlib.

PiperOrigin-RevId: 629379237
  • Loading branch information
@gnecula @tensorflower-gardener
gnecula authored and tensorflower-gardener committed Jun 18, 2024
1 parent e23a719 commit 8bf2f19
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion third_party/xla/xla/python/xla_client.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@

# Just an internal arbitrary increasing number to help with backward-compatible
# changes. In JAX, reference this via jax._src.lib.xla_extension_version.
_version = 271
_version = 272

# Version number for MLIR:Python components.
mlir_api_version = 57
Expand Down

0 comments on commit 8bf2f19

Please sign in to comment.