[pull] master from tensorflow:master#1679

Merged

pull[bot] merged 23 commits into

GesuBackups:masterfrom

tensorflow:master

Mar 31, 2026

Commits on Mar 31, 2026

Allow dots with multiple batch dimensions through GemmFusion - the new emitter can handle this, so there's no reason to restrict this

vwbaker
authored and
tensorflower-gardener
committed
Automated Code Change
tensorflower-gardener
committed
[XLA Presubmit] Add SM 12.0 to the list of known SM architecture

beckerhe
authored and
tensorflower-gardener
committed
PR #39693: Fix tracked_gpu_device_buffer_test in OSS by adding cuda_platform dep

and-ivanov
authored and
tensorflower-gardener
committed
PR #39393: [xla:gpu] Add VA remapping for command buffer thunks

shawnwang18
authored and
tensorflower-gardener
committed
PR #39695: Fix xla aot compile gpu test

and-ivanov
authored and
tensorflower-gardener
committed
Rollback of Migrate XLA GPU emitters and HWIR to use SymbolicMap.
tensorflower-gardener
committed
[xla:cpu] Make small_while_loop_hoisting_pass_test use the same byte threshold as cpu_compiler.

penpornk
authored and
tensorflower-gardener
committed
PR #40141: [xla:gpu] Fix a bug when emitting degenerate collective permute

ezhulenev
authored and
tensorflower-gardener
committed
PR #40108: Update protobuf dependency version to 32.1

ezhulenev
authored and
tensorflower-gardener
committed
PR #39871: [ROCm] Fix bf16 upcast handling for libdevice calls.

zoranjovanovic-ns
authored and
tensorflower-gardener
committed
PR #39309: [ROCm] Add scope_range_id support to ROCm profiler

magaonka-amd
authored and
tensorflower-gardener
committed
PR #39991: Add test coverage for BatchNormExpander untested code paths

kredd2506
authored and
tensorflower-gardener
committed
PR #40111: Bump pygments from 2.18.0 to 2.20.0 in /xla/backends/cpu/benchmarks/e2e/gemma2/flax_2b

dependabot[bot]
authored and
tensorflower-gardener
committed
PR #40118: Fix tensor memory size check to use tcgen05 capability instead of IsBlackwell()

and-ivanov
authored and
tensorflower-gardener
committed
[XLA:GPU] Move triton emitter c++ tests closer to the lit tests.

pifon2a
authored and
tensorflower-gardener
committed
[XLA:GPU] Add slice and iota emission via the new tiling.

pifon2a
authored and
tensorflower-gardener
committed
[XLA:GPU] Simplify xla_aot_compile_gpu_test by using HloPjRtTestBase.

akuegel
authored and
tensorflower-gardener
committed
Fully serialize stream_executor::DeviceDescription to/from proto.

beckerhe
authored and
tensorflower-gardener
committed
[xla:cpu] Roll-back small_while_loop_hoisting_pass threshold refactor.

penpornk
authored and
tensorflower-gardener
committed
Remove gen_gpu_hlo_compile_tests bazel rule.

akuegel
authored and
tensorflower-gardener
committed
PR #39744: [ROCm] Porting CUB sort FFI handler consolidation to ROCm.

tsrw2048
authored and
tensorflower-gardener
committed
Automated Code Change
tensorflower-gardener
committed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

[pull] master from tensorflow:master#1679

[pull] master from tensorflow:master#1679
pull[bot] merged 23 commits into
GesuBackups:masterfrom
tensorflow:master

Commits on Mar 31, 2026

Allow dots with multiple batch dimensions through GemmFusion - the new emitter can handle this, so there's no reason to restrict this

Automated Code Change

[XLA Presubmit] Add SM 12.0 to the list of known SM architecture

PR #39693: Fix tracked_gpu_device_buffer_test in OSS by adding cuda_platform dep

PR #39393: [xla:gpu] Add VA remapping for command buffer thunks

PR #39695: Fix xla aot compile gpu test

Rollback of Migrate XLA GPU emitters and HWIR to use SymbolicMap.

[xla:cpu] Make small_while_loop_hoisting_pass_test use the same byte threshold as cpu_compiler.

PR #40141: [xla:gpu] Fix a bug when emitting degenerate collective permute

PR #40108: Update protobuf dependency version to 32.1

PR #39871: [ROCm] Fix bf16 upcast handling for libdevice calls.

PR #39309: [ROCm] Add scope_range_id support to ROCm profiler

PR #39991: Add test coverage for BatchNormExpander untested code paths

PR #40111: Bump pygments from 2.18.0 to 2.20.0 in /xla/backends/cpu/benchmarks/e2e/gemma2/flax_2b

PR #40118: Fix tensor memory size check to use tcgen05 capability instead of IsBlackwell()

[XLA:GPU] Move triton emitter c++ tests closer to the lit tests.

[XLA:GPU] Add slice and iota emission via the new tiling.

[XLA:GPU] Simplify xla_aot_compile_gpu_test by using HloPjRtTestBase.

Fully serialize stream_executor::DeviceDescription to/from proto.

[xla:cpu] Roll-back small_while_loop_hoisting_pass threshold refactor.

Remove gen_gpu_hlo_compile_tests bazel rule.

PR #39744: [ROCm] Porting CUB sort FFI handler consolidation to ROCm.

Automated Code Change