CI: 02/04/25 upstream sync #221
github-actions
merged
37 commits into
rocm-main
from
ci-upstream-sync-106_1
Add docs
cb2ab409
Add jax.random.multinomial.
32411a43
[better_errors] Merge the JaxprDebugInfo and TracingDebugInfo into co…
c70de6de
Add TSAN numpy step
476b2398
Handle effects in lax.custom_linear_solve.
d42e3650
Merge pull request #25474 from jaro-sevcik:compilation-cache-mock-doc
12c76cde
Merge pull request #26261 from vfdev-5:use-tsan-numpy-in-tsan-ci-2
aa64372b
Merge pull request #26262 from gnecula:debug_info_one
7e353913
Format sourcemap directory names to work on windows
6d7b0357
jax.lax: improve docs for floor, ceil, round.
49c4020f
[mosaic_gpu] `LayoutTest` now correctly resets the value of MOSAIC_GP…
2d7e4ab2
[pallas:triton] Fallback lowering rules for math functions now use ge…
bf6489ff
Merge pull request #26246 from jakevdp:lax-docs
a5d2d89b
Merge pull request #26277 from justinjfu:sourcemap_windows_fix
40d35b42
Merge pull request #25688 from carlosgmartin:random_multinomial
95535df1
Simplify implementation of random.orthogonal.
c478f44e
Merge pull request #26275 from dfm:effects-in-custom-linear-solve
17d0b86c
Merge pull request #25655 from carlosgmartin:simplify_random_orthogonal
2c10a65b
[pallas:triton] The lowering now uses PTX instead of Triton IR
7929cd84
Remove portpicker for free threaded python 3.13t in test-requirements…
59a3552a
[pallas:triton] Fixed dispatch tablee for `lax.pow_p`
f58207a2
[JAX] disable flaky parameter permutations for `sparse_bcoo_bcsr` test.
0abd9538
documentation: fix function name
72b6704d
[direct-linearize] fix name stack tests
8f967c51
`try_matching_out_with_in_spec_for_all_auto` should only work for Nam…
5e7e6911
Merge pull request #26290 from mattjj:linearize-name-stack-fixes-2
363f1e69
Propagate source ranges in location.
60d3836f
[sharding_in_types] Add a canonicalize_value step before dispatching …
bc1a7066
Merge pull request #26289 from reikdas:reikdas/docfix
6281b860
Strip all info except for shape and dtype while matching input and ou…
654a2f6e
[Pallas] Support promise_in_bounds mode in jnp.take_along_axis.
124e1239
[pallas_mgpu] Add a test for emit_pipeline with wgmma.
c7d535d3
Update XLA dependency to use revision
6738986c
Remove old backward compatibility mode for old PRGN custom call on GPU
9f797990
fix binomial sample test on TPU
58a202ad
Merge branch 'rocm-main' into ci-upstream-sync-106_1
c3e27f86
Reduce pytest threads
c20bb5b4
charleshofer
approved these changes
on 2025-02-07
Login to write a write a comment.
Login via GitHub