Commits

Branch selector

 rocm-main

User selector

All users

Datepicker

All time

Commit History

Commits on Feb 7, 2025

• Merge pull request #221 from ROCm/ci-upstream-sync-106_1

  Show description for 9fbc1c1
  github-actions[bot]

  authored
  9fbc1c1

  Copy full SHA for 9fbc1c1

• Reduce pytest threads

  charleshofer

  committed
  c20bb5b

  Copy full SHA for c20bb5b

Commits on Feb 4, 2025

• Merge branch 'rocm-main' into ci-upstream-sync-106_1

  charleshofer

  committed
  c3e27f8

  Copy full SHA for c3e27f8

• fix binomial sample test on TPU

  Show description for 58a202a
  

  froystig

  authored and

  Google-ML-Automation

  committed
  58a202a

  Copy full SHA for 58a202a

• Remove old backward compatibility mode for old PRGN custom call on GPU

  Show description for 9f79799
  

  gnecula

  authored and

  Google-ML-Automation

  committed
  9f79799

  Copy full SHA for 9f79799

• Update XLA dependency to use revision

  Show description for 6738986
  Google-ML-Automation

  committed
  6738986

  Copy full SHA for 6738986

• [pallas_mgpu] Add a test for emit_pipeline with wgmma.

  Show description for c7d535d
  

  petebu

  authored and

  Google-ML-Automation

  committed
  c7d535d

  Copy full SHA for c7d535d

• [Pallas] Support promise_in_bounds mode in jnp.take_along_axis.

  Show description for 124e123
  

  bythew3i

  authored and

  Google-ML-Automation

  committed
  124e123

  Copy full SHA for 124e123

• Strip all info except for shape and dtype while matching input and output avals for donation. For avals that are sharded, we match on shard_shape.

  Show description for 654a2f6
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  654a2f6

  Copy full SHA for 654a2f6

• Merge pull request jax-ml#26289 from reikdas:reikdas/docfix

  Show description for 6281b86
  Google-ML-Automation

  committed
  6281b86

  Copy full SHA for 6281b86

• [sharding_in_types] Add a canonicalize_value step before dispatching bind so that we can insert mesh_casts under the following conditions:

  Show description for bc1a706
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  bc1a706

  Copy full SHA for bc1a706

• Propagate source ranges in location.

  Show description for 60d3836
  

  jpienaar

  authored and

  Google-ML-Automation

  committed
  60d3836

  Copy full SHA for 60d3836

• Merge pull request jax-ml#26290 from mattjj:linearize-name-stack-fixes-2

  Show description for 363f1e6
  Google-ML-Automation

  committed
  363f1e6

  Copy full SHA for 363f1e6

• try_matching_out_with_in_spec_for_all_auto should only work for NamedSharding

  Show description for 5e7e691
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  5e7e691

  Copy full SHA for 5e7e691

• [direct-linearize] fix name stack tests

  Show description for 8f967c5
  

  mattjj

  and

  sharadmv

  committed
  8f967c5

  Copy full SHA for 8f967c5

• documentation: fix function name

  reikdas

  committed
  72b6704

  Copy full SHA for 72b6704

• [JAX] disable flaky parameter permutations for sparse_bcoo_bcsr test.

  Show description for 0abd953
  

  Varcho

  authored and

  Google-ML-Automation

  committed
  0abd953

  Copy full SHA for 0abd953

Commits on Feb 3, 2025

• [pallas:triton] Fixed dispatch tablee for lax.pow_p

  Show description for f58207a
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  f58207a

  Copy full SHA for f58207a

• Remove portpicker for free threaded python 3.13t in test-requirements.txt

  Show description for 59a3552
  

  kanglant

  authored and

  Google-ML-Automation

  committed
  59a3552

  Copy full SHA for 59a3552

• [pallas:triton] The lowering now uses PTX instead of Triton IR

  Show description for 7929cd8
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  7929cd8

  Copy full SHA for 7929cd8

• Merge pull request jax-ml#25655 from carlosgmartin:simplify_random_orthogonal

  Show description for 2c10a65
  Google-ML-Automation

  committed
  2c10a65

  Copy full SHA for 2c10a65

• Merge pull request jax-ml#26275 from dfm:effects-in-custom-linear-solve

  Show description for 17d0b86
  Google-ML-Automation

  committed
  17d0b86

  Copy full SHA for 17d0b86

• Simplify implementation of random.orthogonal.

  carlosgmartin

  committed
  c478f44

  Copy full SHA for c478f44

• Merge pull request jax-ml#25688 from carlosgmartin:random_multinomial

  Show description for 95535df
  Google-ML-Automation

  committed
  95535df

  Copy full SHA for 95535df

• Merge pull request jax-ml#26277 from justinjfu:sourcemap_windows_fix

  Show description for 40d35b4
  Google-ML-Automation

  committed
  40d35b4

  Copy full SHA for 40d35b4

• Merge pull request jax-ml#26246 from jakevdp:lax-docs

  Show description for a5d2d89
  Google-ML-Automation

  committed
  a5d2d89

  Copy full SHA for a5d2d89

• [pallas:triton] Fallback lowering rules for math functions now use general dtypes

  Show description for bf6489f
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  bf6489f

  Copy full SHA for bf6489f

• [mosaic_gpu] LayoutTest now correctly resets the value of MOSAIC_GPU_DUMP_SASS

  Show description for 2d7e4ab
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  2d7e4ab

  Copy full SHA for 2d7e4ab

• jax.lax: improve docs for floor, ceil, round.

  jakevdp

  committed
  49c4020

  Copy full SHA for 49c4020

• Format sourcemap directory names to work on windows

  justinjfu

  committed
  6d7b035

  Copy full SHA for 6d7b035

• Merge pull request jax-ml#26262 from gnecula:debug_info_one

  Show description for 7e35391
  Google-ML-Automation

  committed
  7e35391

  Copy full SHA for 7e35391

• Merge pull request jax-ml#26261 from vfdev-5:use-tsan-numpy-in-tsan-ci-2

  Show description for aa64372
  Google-ML-Automation

  committed
  aa64372

  Copy full SHA for aa64372

• Merge pull request jax-ml#25474 from jaro-sevcik:compilation-cache-mock-doc

  Show description for 12c76cd
  Google-ML-Automation

  committed
  12c76cd

  Copy full SHA for 12c76cd

• Merge pull request #219 from ROCm/ci-upstream-sync-104_1

  Show description for 0962b96
  github-actions[bot]

  authored
  0962b96

  Copy full SHA for 0962b96

• Merge branch 'rocm-main' into ci-upstream-sync-104_1

  charleshofer

  committed
  ec2c305

  Copy full SHA for ec2c305

Pagination

Previous

Next