Commits

Branch selector

 regex

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 6, 2023

• Fix tests that mistakenly used assertRaises(..., msg=...) to match a message.

  hawkinsp

  committed
  5311746

  Copy full SHA for 5311746

Commits on Oct 5, 2023

• [Mosaic] apply_vector_layout C++ rewrite (17): for.op (from cl/568376871)

  Show description for f4bb1c0
  

  tlongeri

  authored and

  jax authors

  committed
  f4bb1c0

  Copy full SHA for f4bb1c0

• random_test: Split into two so that each target is small enough to fit within a medium timeout.

  Show description for 8f911e1
  

  jax authors

  committed
  8f911e1

  Copy full SHA for 8f911e1

• [JAX] Keep CPU host callbacks alive via IFRT, rather than by attaching them to the Python object.

  Show description for 1512650
  

  hawkinsp

  authored and

  jax authors

  committed
  1512650

  Copy full SHA for 1512650

• [Mosaic] apply_vector_layout C++ rewrite: Have each rule use its own builder instead of passing a builder in the context

  Show description for 7c353c4
  

  tlongeri

  authored and

  jax authors

  committed
  7c353c4

  Copy full SHA for 7c353c4

• Merge pull request #17965 from tttttanya:Asynchronous-dispatch-jaxArray-update

  Show description for 70abf9a
  

  jax authors

  committed
  70abf9a

  Copy full SHA for 70abf9a

• Asynchronous dispatch doc update regarding jax.Array migration

  tttttanya

  committed
  9e50b9d

  Copy full SHA for 9e50b9d

• [Mosaic] infer_memref_layout C++ rewrite

  Show description for 61bd34c
  

  tlongeri

  authored and

  jax authors

  committed
  61bd34c

  Copy full SHA for 61bd34c

• Bump shard_count for shard_map_test to fix timeouts.

  Show description for a2b70e3
  

  jax authors

  committed
  a2b70e3

  Copy full SHA for a2b70e3

• Re-add JIT to jax.numpy.bitwise_count

  Show description for 4fd6bf4
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  4fd6bf4

  Copy full SHA for 4fd6bf4

• Expose mlir.ShapePolyLoweringState

  Show description for 797f577
  

  jax authors

  committed
  797f577

  Copy full SHA for 797f577

• [Mosaic] apply_vector_layout C++ rewrite: various bug fixes

  Show description for ab4a8e3
  

  tlongeri

  authored and

  jax authors

  committed
  ab4a8e3

  Copy full SHA for ab4a8e3

• Update XLA dependency to use revision

  Show description for 295cecd
  

  hawkinsp

  authored and

  jax authors

  committed
  295cecd

  Copy full SHA for 295cecd

• [Mosaic] apply_vector_layout: Use shape in generalizes check (in Python)

  Show description for 68c84a6
  

  tlongeri

  authored and

  jax authors

  committed
  68c84a6

  Copy full SHA for 68c84a6

• [Mosaic] Fix a buggy vector.broadcast rule in apply_vector_layout

  Show description for 633f68a
  

  apaszke

  authored and

  jax authors

  committed
  633f68a

  Copy full SHA for 633f68a

• [Mosaic] Handle a larger class of broadcasts with 1-sized trailing dimensions

  Show description for d8a81ba
  

  apaszke

  authored and

  jax authors

  committed
  d8a81ba

  Copy full SHA for d8a81ba

• [Mosaic] Fix a bug when elementwise op can incorrectly propagate a replicated layout

  Show description for 2e3a5d6
  

  apaszke

  authored and

  jax authors

  committed
  2e3a5d6

  Copy full SHA for 2e3a5d6

• [Mosaic] apply_vector_layout C++ rewrite (16): vector.multi_reduction

  Show description for dd2fcf5
  

  tlongeri

  authored and

  jax authors

  committed
  dd2fcf5

  Copy full SHA for dd2fcf5

• Merge pull request #17921 from gnecula:harness_random

  Show description for 201001b
  

  jax authors

  committed
  201001b

  Copy full SHA for 201001b

• [Mosaic] apply_vector_layout C++ rewrite (15): vector.shape_cast

  Show description for 07e4077
  

  tlongeri

  authored and

  jax authors

  committed
  07e4077

  Copy full SHA for 07e4077

• [XLA:Mosaic] Fix layout attribute string parse in python.

  Show description for 98aae41
  

  bythew3i

  authored and

  jax authors

  committed
  98aae41

  Copy full SHA for 98aae41

• sparse_test: Split into two so that each target is small enough to fit within a medium timeout.

  Show description for 1c37f50
  

  jax authors

  committed
  1c37f50

  Copy full SHA for 1c37f50

• [pallas] Fix `allow_tf32` value in Triton `dot_general` lowering.

  Show description for 465eb21
  

  chr1sj0nes

  authored and

  jax authors

  committed
  465eb21

  Copy full SHA for 465eb21

• [Mosaic] apply_vector_layout C++ rewrite (14): tpu.matmul and vector.contract

  Show description for 862d676
  

  tlongeri

  authored and

  jax authors

  committed
  862d676

  Copy full SHA for 862d676

Commits on Oct 4, 2023

• [Mosaic] apply_vector_layout C++ rewrite (13): scf.if, scf.yield

  Show description for 991e6ef
  

  tlongeri

  authored and

  jax authors

  committed
  991e6ef

  Copy full SHA for 991e6ef

• Merge pull request #17929 from hawkinsp:torchloader

  Show description for 9e3d64a
  

  jax authors

  committed
  9e3d64a

  Copy full SHA for 9e3d64a

• Merge pull request #17938 from jakevdp:bitwise-count

  Show description for 6065464
  

  jax authors

  committed
  6065464

  Copy full SHA for 6065464

• Merge pull request #17933 from cgarciae:condition-numpy-version

  Show description for 7be8df9
  

  jax authors

  committed
  7be8df9

  Copy full SHA for 7be8df9

• jnp.bitwise_count: call into lax.population_count

  jakevdp

  committed
  7df2957

  Copy full SHA for 7df2957

• condition numpy version based on python version

  cgarciae

  committed
  7498ffe

  Copy full SHA for 7498ffe

• Allow duration event listeners to take extra keyword arguments.

  Show description for be7f210
  

  jax authors

  committed
  be7f210

  Copy full SHA for be7f210

• random_test: reduce num_generated_cases to avoid timeouts

  Show description for 305efe0
  

  jax authors

  committed
  305efe0

  Copy full SHA for 305efe0

• [Mosaic] apply_vector_layout C++ rewrite: generalization for relayouts that reduce sublanes

  Show description for 933d353
  

  tlongeri

  authored and

  jax authors

  committed
  933d353

  Copy full SHA for 933d353

• Merge pull request #17925 from jakevdp:rand-loggamma

  Show description for 389334d
  

  jax authors

  committed
  389334d

  Copy full SHA for 389334d

• Simplify the torch data loader collate function using tree_map.

  Show description for d8a0227
  hawkinsp

  committed
  d8a0227

  Copy full SHA for d8a0227

Pagination

Previous

Next