Commits

Branch selector

 main

User selector

pschuh

Datepicker

All time

Commit History

Commits on Jun 7, 2022

• Update testReducerNoDtype and others to not run duplicate tests.

  pschuh

  committed
  b8b5c83
  More actions

Commits on May 27, 2022

• Bump XLA version to include fix to libtpu.

  

  pschuh

  authored and

  jax authors

  committed
  c9ba08b
  More actions

Commits on May 4, 2022

• If optimization_barrier is not available, disable the jax2tf tests.

  pschuh

  committed
  12743e7
  More actions

Commits on Apr 15, 2022

• Merge branch 'main' into opt-barrier-gpu

  pschuh

  authored
  58ef6cb
  More actions

Commits on Apr 14, 2022

• Switch gpu, cpu and jax2tf to use the new OptimizationBarrier op.

  pschuh

  committed
  b0c41b9
  More actions

Commits on Apr 1, 2022

• Fix race condition for weakref destructor by catching rare exceptions.

  pschuh

  committed
  df1c478
  More actions

Commits on Mar 23, 2022

• Update util.py

  pschuh

  authored
  c1bb767
  More actions

Commits on Mar 21, 2022

• Add weakreaf_lru_cache to prevent caches from pinning jaxprs.

  pschuh

  committed
  d0e0da0
  More actions

Commits on Feb 15, 2022

• Merge branch 'main' into opt-barrier

  pschuh

  authored
  662c441
  More actions

Commits on Feb 14, 2022

• Add translation rule for optimization barrier.

  pschuh

  committed
  7ce911b
  More actions

Commits on Jan 26, 2022

• Update get_compile_options to accept Device objects.

  pschuh

  committed
  c06933e
  More actions

Commits on Nov 30, 2021

• Update device_get docs to mention parrallelism.

  pschuh

  committed
  46a1033
  More actions

Commits on Jul 12, 2021

• Add partitioning support to outfeed.

  pschuh

  committed
  541b31d
  More actions

Commits on Jun 11, 2021

• Add multi-backend support to parallel primatives.

  pschuh

  committed
  6f5abf0
  More actions

Commits on Jun 10, 2021

• Jaxpr transform was losing name.

  pschuh

  committed
  10c0b8d
  More actions

Commits on May 7, 2021

• Revert "Use convert_element_type instead of device_put_raw."

  pschuh

  authored
  9224601
  More actions

Commits on May 6, 2021

• Merge branch 'master' into convert_element

  pschuh

  authored
  9d3e535
  More actions

Commits on Apr 25, 2021

• jnp.asarray -> _device_put_raw for call_tf.py

  pschuh

  committed
  2c35b1a
  More actions

Commits on Apr 22, 2021

• Use convert_element_type insteadof device_put_raw.

  

  pschuh

  authored and

  mattjj

  committed
  99b755f
  More actions

Commits on Apr 21, 2021

• Change wrap_execute_replicated -> compile_replicated.

  pschuh

  committed
  bb96ae7
  More actions

Commits on Mar 2, 2021

• Add backend specific path for execute_replicated.

  pschuh

  committed
  211d64d
  More actions

Commits on Feb 25, 2021

• Decouple ShardedDeviceArray from _DeviceArray

  pschuh

  committed
  1028939
  More actions

Commits on Feb 5, 2021

• Convert execute_on_local_devices to execute_sharded_on_local_devices for jit(pmap).

  pschuh

  committed
  71a00bb
  More actions

Commits on Feb 4, 2021

• Add support for execute_sharded_on_local_devices.

  pschuh

  committed
  ed5d1fa
  More actions

Commits on Feb 1, 2021

• Bugfix to allow pxla.replicate to generate Replicated buffers.

  pschuh

  committed
  bb0f50f
  More actions

Commits on Jan 21, 2021

• Refactor how the result processing code wraps the indexing information.

  pschuh

  committed
  df273c0
  More actions