All pull requests

Pull requests

• Add GetAllReduceSliceBlockAssignment to collective_opt_utils

  #51791·copybara-service[bot] opened Oct 28, 2026Bot·

  ·
  

• [XLA:GPU] Register GpuAlgebraicSimplifier in HloPassRegistry.

  #51790·copybara-service[bot] opened Oct 28, 2026Bot·

  ·
  

• [XLA:GPU] Add layout assignment constraints for kScaledDot.

  #51789·copybara-service[bot] opened Oct 28, 2026Bot·

  ·
  

• [Mosaic:TPU] Check operand rank equality in ConcatenateOp verifier

  #51785·copybara-service[bot] opened Oct 28, 2026Bot·

  ·
  

• PR #51669: Add optional WritableFile::Seek support for POSIX file system

  #51784·copybara-service[bot] opened Oct 28, 2026Bot·

  ·

• Bump github/codeql-action/upload-sarif from 4.38.3 to 4.39.1

  dependenciesgithub_actions
  #51783·dependabot[bot] opened Oct 28, 2026Bot·

  ·

• Bump lit from 23.1.3 to 23.1.4

  dependenciespython
  #51782·dependabot[bot] opened Oct 28, 2026Bot·

  ·

• Bump astral-sh/setup-uv from 10.3.0 to 10.3.1

  dependenciesgithub_actions
  #51781·dependabot[bot] opened Oct 28, 2026Bot·

  ·

• Automated Code Change

  #51773·copybara-service[bot] opened Oct 28, 2026Bot·

  ·

• Automated Code Change

  #51770·copybara-service[bot] opened Oct 28, 2026Bot·

  ·

• Add Stream::Memcpy2D and CudaStream::Memcpy2D.

  #51768·copybara-service[bot] opened Oct 28, 2026Bot·

  ·

• Support CopySlice and UpdateSlice on StreamExecutor GPU HBM.

  #51767·copybara-service[bot] opened Oct 28, 2026Bot·

  ·

• Store Fusion Explorer graphs and metadata directly in the zipapp.

  #51766·copybara-service[bot] opened Oct 28, 2026Bot·

  ·
  

• [XLA:GPU][oneAPI] Simplify OneAPIComputeCapability with Xe arch inference

  #51764·akhilgoe opened Oct 28, 2026Contributor·

  ·
  11

• Lower ApproxTopK fallback to unstable sort on CPU and GPU.

  #51762·copybara-service[bot] opened Oct 28, 2026Bot·

  ·
  

• Add GetAllReduceSliceBlockAssignment to collective_opt_utils

  #51761·copybara-service[bot] opened Oct 28, 2026Bot·

  ·
  

• [XLA] Remove HloOpcode::kDomain, HloDomainInstruction, and DomainMetadata

  #51759·copybara-service[bot] opened Oct 28, 2026Bot·

  ·
  

• [XLA] Replace AllocateBuffer copies with fresh buffers

  #51758·copybara-service[bot] opened Oct 28, 2026Bot·

  ·
  

• [PJRT] Clear PadRealToStatic metadata prefix through splits and async offloads

  #51757·copybara-service[bot] opened Oct 28, 2026Bot·

  ·

• Use JAX's Pathways profiler instead of monkey patching jax.profiler.

  #51756·copybara-service[bot] opened Oct 28, 2026Bot·

  ·

• Pass the profile session ID to profilers of on-demand captures.

  #51755·copybara-service[bot] opened Oct 28, 2026Bot·

  ·

• Add a JAX profiler that traces Pathways backends.

  #51754·copybara-service[bot] opened Oct 28, 2026Bot·

  ·

• [XLA] Speed up fake argument initialization and device loading in HLO runner

  #51753·copybara-service[bot] opened Oct 28, 2026Bot·

  ·
  

• Unify TOTAL and PARTIAL U64 key adapters in XLA:GPU TopkSpecializer.

  #51752·copybara-service[bot] opened Oct 27, 2026Bot·

  ·
  

• [Pallas:SC] Lower all_reduce_ffs to tpu.reduce_index when layout passes are on

  #51751·copybara-service[bot] opened Oct 27, 2026Bot·

  ·