Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10000 commits
Select commit Hold shift + click to select a range
688f9e0
[XLA] Update HloAsyncStartInstruction::AddCallOperand API
tensorflower-gardener Jun 24, 2026
a0fc6f3
PR #44748: [PJRT C API] Add PJRT_DeviceEvent typedef
cerisier Jun 24, 2026
e86981f
Introduces guards to running TuplePointsToAnalysis so that it is only…
tensorflower-gardener Jun 24, 2026
14b798d
Fix int overflow by using `int64_t` instead of `int`.
tensorflower-gardener Jun 24, 2026
455cc22
record duration of RunCudnnCompilerPasses into RunBackend
ermilovmaxim Jun 24, 2026
d7db25d
Cleanup AutotunerConfig and use config_assigner options directly.
shyamliagrawal Jun 24, 2026
29c8321
Remove constraint on contracting dimension size for allowing slices i…
vwbaker Jun 24, 2026
03ba8e8
[JAX:XLA] Add inlining control with multi-state 'inline' parameter
zvikinoza Jun 24, 2026
299b67d
[XLA:PJRT] Interpreter support for HLO output callbacks
tensorflower-gardener Jun 24, 2026
a8d969b
Refactor latency_hiding_scheduler to make it polymorphic.
tensorflower-gardener Jun 24, 2026
e704fe2
[XLA:Tools] Integrate TPU Fusion Debugger with HLO Isolation API
tensorflower-gardener Jun 24, 2026
7c54142
Add CopyArrays tests for `xla::ifrt::Bundle`.
tensorflower-gardener Jun 24, 2026
910ef0c
Parameterize ReshardArrays tests to cover `Bundle::ReshardArrays`
tensorflower-gardener Jun 25, 2026
fd93394
Replace std::reverse with Abseil container algorithms.
apivovarov Jun 25, 2026
7ea9c41
rename CompileToPtx to CompileToTargetBinary
ermilovmaxim Jun 25, 2026
17f0310
[XLA] Allow multiple operands for HloAsyncUpdate in CreateFromProto.
tensorflower-gardener Jun 25, 2026
2bc0de6
Reverts 03ba8e8197e45af11b2897e67b17c4cd3faefd29
tensorflower-gardener Jun 25, 2026
1712d05
Automated Code Change
tensorflower-gardener Jun 25, 2026
565986f
Refine error messages for AsyncUpdate/AsyncDone operands in HLO parser.
tensorflower-gardener Jun 25, 2026
826e213
Refactor tensorflow::FunctionLibraryDefinition to support move semant…
tensorflower-gardener Jun 25, 2026
9133967
Fix Use-After-Free, memory leaks, and missing null checks in TFLite P…
dmiltr3 Jun 25, 2026
9a2ad43
Automated Code Change
tensorflower-gardener Jun 25, 2026
6a3bf4b
Reverts changelist 936455016
dimitar-asenov Jun 25, 2026
b9e64ee
Fix xla_dump_to flag propagation from XLA_FLAGS env var
akuegel Jun 25, 2026
34ba96c
Automated Code Change
tensorflower-gardener Jun 25, 2026
594193b
Automated Code Change
tensorflower-gardener Jun 25, 2026
fb3c213
Migrate GPU-specific emitters passes to ODS-generated constructors
akuegel Jun 25, 2026
b747fba
Upgrade Abseil-CPP dependency to LTS 20260526.0.
Jun 25, 2026
2629006
[XLA:GPU] Add xla flag which should allow users to enable NCCL symmet…
PatriosTheGreat Jun 25, 2026
8f632f6
PR #44572: [ROCm] Remove xla_gpu_use_inprocess_lld and xla_gpu_use_em…
draganmladjenovic Jun 25, 2026
cab4935
Make GetCurrentHloPassMetadata public.
tensorflower-gardener Jun 25, 2026
a22ad6b
Automated Code Change
tensorflower-gardener Jun 25, 2026
253c25a
Automated Code Change
tensorflower-gardener Jun 25, 2026
d6ec122
Automated Code Change
tensorflower-gardener Jun 25, 2026
3fded39
Remove nvshmem_stub from pywrap_tensorflow_internal dependencies.
PatriosTheGreat Jun 25, 2026
b7f660b
[tests] Anticipate upcoming restrictions to `arith.constant`.
tensorflower-gardener Jun 25, 2026
16448ce
Automated Code Change
tensorflower-gardener Jun 25, 2026
c9cb060
[XLA:GPU] Add functions which allows to determine if NCCL symmetric k…
PatriosTheGreat Jun 25, 2026
426f097
Automated Code Change
tensorflower-gardener Jun 25, 2026
834fedf
Drop now-unused MultiplyWithoutOverflow include
RadikHoroshev Jun 25, 2026
53c6c77
Migrate CPU emitters passes to ODS-generated constructors
akuegel Jun 25, 2026
037d9a9
PR #44763: [XLA:GPU][ROCm] Fix rocm_clang_local host toolchain under …
nurmukhametov Jun 25, 2026
75fe3d2
Calculate bytes per iterations and K-iterations in dot cost model
karupayun Jun 25, 2026
5f58d72
UX improvements for the split_proto_cli
EusebioDM Jun 25, 2026
4077637
Optimize ComputeRuntimeOrdering by avoiding unnecessary map insertion…
sashabu Jun 25, 2026
4bb0ba1
PR #44438: Bump github/codeql-action from 4.36.1 to 4.36.2
dependabot[bot] Jun 25, 2026
3e14bf3
[XLA] Preverse buffer aliasing/donation in XlaTranform.
Tongfei-Guo Jun 25, 2026
d8f55af
Refactor unit tests use HLO strings for input.
ekayaaslan Jun 25, 2026
8a8f547
[StableHLO] Fix 2 bugs related to `interpreter.print` inside `interpr…
Jun 25, 2026
539e63c
Check the output hlo fully on unit tests.
ekayaaslan Jun 25, 2026
ed27846
Add CompilerBuildInfo sub-message to CompilationLogEntry proto with
tensorflower-gardener Jun 25, 2026
78b5a60
Automated Code Change
tensorflower-gardener Jun 25, 2026
fb214c9
Fix nullptr dereference in HloRematerialization PeakPriorityMode.
JoelWee Jun 25, 2026
0bfad91
Fix integer overflow in CreateTensor size validation
ratgr Jun 25, 2026
bc72fab
[PJRT C] Guard the places that attempt to set the layout
hyeontaek Jun 25, 2026
d261667
Improve security of pip package build scripts.
ratgr Jun 25, 2026
262caa1
[XLA:GPU]: Wrap all collectives in async-start for GPUs
sohaibiftikhar Jun 25, 2026
46f0d3f
[XLA:GPU] emit reshape as reshape + broadcast
metaflow Jun 25, 2026
5ea7ed3
PR #37766: fix: add recursion depth limit to HLO parser
thevilledev Jun 25, 2026
41500f9
Fix Use-After-Free in TfliteInferenceStage by retrieving raw output p…
dmiltr3 Jun 25, 2026
9e79f2f
Relax tolerances of depthwise convolution test
dsharletg Jun 25, 2026
6460f67
Fix heap out-of-bounds read in TOCO ReorderReshapeTranspose graph tra…
dmiltr3 Jun 25, 2026
e9d31ab
Replace std::iota with absl::c_iota
apivovarov Jun 25, 2026
b22ab80
Adding Android workflow to Github
vkarampudi Jun 25, 2026
2cbe5cc
Remove redundant maximum_iterations args
kevinbchen Jun 25, 2026
407c427
Expand cumprod zero gradient coverage
danyalahmed1995 Jun 25, 2026
1371051
PR #44805: Fix PJRT plugin README and E0102 error docs
doehyunbaek Jun 25, 2026
ef1bfb1
PR #44780: docs: fix broken PJRT links and typo in determinism.md
Ashutosh0x Jun 25, 2026
fccba01
Add unit tests for LogicalBufferAnalysis.
tensorflower-gardener Jun 25, 2026
3d7f9f9
add multi target device to xla_compile
ermilovmaxim Jun 25, 2026
f4dbd2e
Refactor: Use HLO string parsing in tuple_points_to_analysis_test.cc
tensorflower-gardener Jun 25, 2026
3c1e373
Update XNNPACK in XLA
dsharletg Jun 26, 2026
bab2665
Make TPU API function pointer initialization thread-safe.
dmiltr3 Jun 26, 2026
80416ce
Remove redundant maximum_iterations args
tensorflower-gardener Jun 26, 2026
565159f
Refactor variable array storage in IfrtServingExecutable.
xiaoxlu Jun 26, 2026
6196802
Refactor portable sharding lookup in IfrtServingExecutable.
xiaoxlu Jun 26, 2026
87f64a0
Optimize IfrtServingExecutable by conditionally populating static sha…
xiaoxlu Jun 26, 2026
0dfae13
Automated Code Change
tensorflower-gardener Jun 26, 2026
6a12638
Automated Code Change
ClaytonKnittel Jun 26, 2026
04cfd63
Automated Code Change
tensorflower-gardener Jun 26, 2026
241baba
Add IndexProto and IndexDomainProto.
tensorflower-gardener Jun 26, 2026
e1b5c33
Automated Code Change
tensorflower-gardener Jun 26, 2026
e582355
Automated Code Change
tensorflower-gardener Jun 26, 2026
fe91749
Automated Code Change
tensorflower-gardener Jun 26, 2026
1a06686
Automated Code Change
tensorflower-gardener Jun 26, 2026
d114bd3
Remove references to XLA framework dialect and its passes
akuegel Jun 26, 2026
2a7b27b
Refactor TuplePointsToAnalysis for clarity and minor code simplificat…
tensorflower-gardener Jun 26, 2026
93a14f6
Propagate debug information to fusion search space.
vwbaker Jun 26, 2026
a3d10ab
[XLA] Clean up Literal creation deprecations and style in HloEvaluator.
tensorflower-gardener Jun 26, 2026
f280fa5
[NFC] Delete no longer needed Abseil patch.
allanrenucci Jun 26, 2026
93fa020
Automated Code Change
tensorflower-gardener Jun 26, 2026
1c265e7
PR #42876: [XLA:GPU] Add GPU support to onednn build
kanvi-nervana Jun 26, 2026
4e352a6
Reverts 262caa1b29b2ea9b6039d471522957e52e76fe3b
junwhanahn Jun 26, 2026
58073ec
PR #44224: [XLA:GPU] Migrate VMM allocator bookkeeping to AllocationR…
shawnwang18 Jun 26, 2026
fba94bc
Automated Code Change
tensorflower-gardener Jun 26, 2026
37f21a6
remove no longer needed xla_gpu_enable_llvm_module_compilation_parall…
ermilovmaxim Jun 26, 2026
5abc207
PR #44834: [xla] Forward frontend attributes in AllReduceFolder
ezhulenev Jun 26, 2026
476a9f9
Merge pull request #121710 from wilmerdooley:oss/decode-float8-e4m3fn…
tensorflower-gardener Jun 26, 2026
3c344f6
Merge pull request #121381 from Techie03:fix-matmul-err-msg
tensorflower-gardener Jun 26, 2026
8ce8439
Merge pull request #121173 from wilmerdooley:oss/issue-105642
tensorflower-gardener Jun 26, 2026
cddee73
Merge pull request #121172 from wilmerdooley:oss/issue-105644
tensorflower-gardener Jun 26, 2026
d1c89a2
Merge pull request #121081 from wilmerdooley:oss/issue-62631
tensorflower-gardener Jun 26, 2026
343203f
Merge pull request #120373 from Venkat6871:fix-xla-dynamic-slice-shap…
tensorflower-gardener Jun 26, 2026
98c9cd9
Merge pull request #120282 from danyalahmed1995:fix-cumprod-zero-grad…
tensorflower-gardener Jun 26, 2026
8c0c9cf
[mhlo] Add is_stable attribute to TopKOp definitions (Part 1/2)
apivovarov Jun 26, 2026
ebe9084
Merge pull request #120178 from Ashutosh0x:fix/cost-estimator-strides…
tensorflower-gardener Jun 26, 2026
833ec69
PR #44829: [xla:gpu] Do not sink constants that are not part of offse…
ezhulenev Jun 26, 2026
0ba170a
Remove `PLATFORM_GOOGLE` guards from error creation functions.
allanrenucci Jun 26, 2026
78b2868
Merge pull request #118032 from jbbqqf:feat/89106-boolean-mask-doc
tensorflower-gardener Jun 26, 2026
62e1e23
Merge pull request #118031 from jbbqqf:feat/117843-inverse-stft-valid…
tensorflower-gardener Jun 26, 2026
87fcb70
Merge pull request #115551 from Cyrax321:fix-posixpath-sys-path
tensorflower-gardener Jun 26, 2026
eeb67d4
Merge pull request #118269 from izzettin-ahmet:fix-variant-recursion-…
tensorflower-gardener Jun 26, 2026
d0d3dc7
Merge pull request #117965 from arpittkhandelwal:fix-fake-quant-xla-r…
tensorflower-gardener Jun 26, 2026
3c30356
Generate cuDNN literal constant tensors with rank matching their broa…
tensorflower-gardener Jun 26, 2026
990e314
Add safety check in pjrt_ifrt for argument buffer count vs computatio…
tensorflower-gardener Jun 26, 2026
aa1f88d
PR #44813: [Bazel] Add explicit loads of external rule symbols
cerisier Jun 26, 2026
2c9e910
Refactor CreateHloRegion in tiled_hlo.cc to use OperandsSpec struct
Moerafaat Jun 26, 2026
8c86acb
[XLA:GPU] Log interconnection information when ncclCommWindowRegister…
PatriosTheGreat Jun 26, 2026
f5350f6
Integrate LLVM at llvm/llvm-project@dd499f88bf4d
tensorflower-gardener Jun 26, 2026
acd7384
Relax broadcast guard: allow in_n>0 (legit large constant initializer…
RadikHoroshev Jun 26, 2026
a72ccb1
Use ASCII in IsSafeProtoAllocation doc comment
RadikHoroshev Jun 26, 2026
0405f13
Add dummy buffer_uses() implementation for HostExecuteStartThunk.
EusebioDM Jun 26, 2026
21853e9
[XLA:GPU] Prune inactive operands in EmitConcatenate.
tensorflower-gardener Jun 26, 2026
ab00235
Remove references to XLA framework dialect and its passes (#122090)
akuegel Jun 26, 2026
ef0b0b5
Upgrade Abseil to LTS 20260526.0
allanrenucci Jun 26, 2026
955fec3
Add utilities for cache implementations.
shyamliagrawal Jun 26, 2026
6eddff4
Enable ARM support for CuTe DSL
gflegar Jun 26, 2026
57a3dc2
Iron out parameter limit in Triton fusions
vwbaker Jun 26, 2026
6a8b37a
[XLA:GPU] update lit tests to make sure we run checks on them
metaflow Jun 26, 2026
ef555e2
PR #44872: [ROCm] Fix dangling reference to cubin span in RocmExecuto…
nurmukhametov Jun 26, 2026
efb6530
Implement buffer_uses for SelectKThunk.
EusebioDM Jun 26, 2026
7d8fa23
PR #44867: [XLA:GPU] zero initialize the output buffers of cuDNN SDPA…
Cjkkkk Jun 26, 2026
a5a9f53
Improve string tensor validation in SetTfTensorFromTfLite.
ratgr Jun 26, 2026
88f2d78
Add SparseCore CAE regex to TSL Profiler XPlane utilities to support …
tensorflower-gardener Jun 26, 2026
598f0ba
PR #44873: [ROCm] Suppress known RCCL leaks in LSan ignore list
nurmukhametov Jun 26, 2026
492dc97
Implement buffer_uses for CollectiveKernelThunk (All-Reduce).
EusebioDM Jun 26, 2026
27f75cd
Fix Use-Restrict clamping in StridedSliceOp to dynamic dimensions.
dmiltr3 Jun 26, 2026
3c29c13
PR #44884: [ROCm] Use proper sysroot version, fixing abi clashes with…
alekstheod Jun 26, 2026
8702081
PR #44702: increase default value for kMaxEvents
ScXfjiang Jun 26, 2026
0aea7fd
Reverts 2bc0de60ad6c16b9b17562c237ddce002a59e844
zvikinoza Jun 26, 2026
de23d91
[XLA] Carry over metadata_payload independently of op_name in Replace…
thcmbs Jun 26, 2026
7970f6e
Integrate Triton up to 2b74f117d42e70418264c8ed84a6ea9b1452def8
tensorflower-gardener Jun 26, 2026
392f795
guard uneven split check against dynamic shapes and improve error mes…
dxbjavid Jun 26, 2026
69091fe
Implement Protobuf-native PersistentCache.
shyamliagrawal Jun 26, 2026
eec46bf
Fix ArrayImplHashTest.HashValuesInconsistentReplicas to be robust aga…
tensorflower-gardener Jun 26, 2026
f983bdf
Integrate LLVM at llvm/llvm-project@030e74c2808a
tensorflower-gardener Jun 26, 2026
cfa8089
Fix MLIR parser issue when running with dataformatting Offloading by …
tensorflower-gardener Jun 26, 2026
9d2251c
Delete XLA framework dialect and its passes
akuegel Jun 26, 2026
cde2ddc
[XLA:GPU] Stop expanding variadic scatter if supported by emitters.
derdrdirk Jun 26, 2026
2a4fe07
Uncomment rules_cc loads in Triton OSS BUILD files.
vwbaker Jun 26, 2026
4b48483
Add _xla_table_name frontend attribute to grad and custom combiner ops.
tensorflower-gardener Jun 26, 2026
48c4ec7
Add is_stable attribute to TopK and plumb through custom-call rewriter
apivovarov Jun 26, 2026
041ebe3
Parse table stats attributes from frontend attributes in the XLA TPU …
tensorflower-gardener Jun 26, 2026
bbc1225
Fix compilation issues for TensorFlow Windows builds with clang-cl.
dmiltr3 Jun 26, 2026
0a58885
This change introduces a new sweep-line buffer assignment algorithm f…
tensorflower-gardener Jun 26, 2026
a409b5b
[XLA] Support non-default memory space in custom-call layouts during …
patrick-toulme Jun 26, 2026
87bbf65
[XLA] Fix error message in HloVerifier::VerifyOperand.
tensorflower-gardener Jun 26, 2026
26a416a
Add explicit check in Prepare and Eval to ensure multipliers values a…
dmiltr3 Jun 27, 2026
e5225fc
Fix SIGSEGV in xplane_utils.cc on RemoveLine. Check if the element is…
zzzaries Jun 27, 2026
35919fd
[7/?] remove old TPU runtime code
Jun 27, 2026
77f45fa
Update TransposePlan to support fused subbyte packing when the final …
pschuh Jun 27, 2026
a112d5a
Fix integer overflow and uninitialized memory read vulnerabilities in…
dmiltr3 Jun 27, 2026
c031f78
Add unit test for unsorted _read_only_resource_inputs
apoorvdarshan Jun 27, 2026
17953bf
Update XNNPACK and its dependencies.
tensorflower-gardener Jun 27, 2026
b8a1715
Automated Code Change
tensorflower-gardener Jun 27, 2026
0f2410f
Automated Code Change
tensorflower-gardener Jun 27, 2026
dfb0fe0
Automated Code Change
tensorflower-gardener Jun 27, 2026
3a40145
[XLA][HLO Value Tracking] Update original value of root instruction a…
jcai19 Jun 27, 2026
d6cf99e
Raise no-data proto allocation cap from 64 MB to 2 GB
RadikHoroshev Jun 27, 2026
b9a6990
Introduce SKIP_IF_DEVICE macro
toli-y Jun 27, 2026
f0466ec
fix sub-byte constant aliasing
ermilovmaxim Jun 28, 2026
17a59e4
Integrate LLVM at llvm/llvm-project@9f423a973b69
boomanaiden154 Jun 28, 2026
dade15f
Automated Code Change
tensorflower-gardener Jun 28, 2026
3676cd5
Add YNNPACK delegate
dsharletg Jun 28, 2026
4c6bb15
Automated Code Change
tensorflower-gardener Jun 29, 2026
a6f48ac
Integrate LLVM at llvm/llvm-project@f3a69752ac18
boomanaiden154 Jun 29, 2026
a84ad24
[XLA] Clean up dependencies in HloVerifier
tensorflower-gardener Jun 29, 2026
8778a4e
PR #44887: [XLA:GPU] Rename two Command API names
shawnwang18 Jun 29, 2026
76bbd17
PR #44861: [XLA:GPU] Reuse stale VMM reservation mappings in Map in …
shawnwang18 Jun 29, 2026
58c0b84
Refactor LogicalBufferAnalysis::GetBuffer to return absl::StatusOr.
tensorflower-gardener Jun 29, 2026
bd474d2
Automated Code Change
tensorflower-gardener Jun 29, 2026
649d192
Add AOTInterceptionPjrtClient skeleton for AOT tests.
khasanovaa Jun 29, 2026
2e8de6c
Automated Code Change
tensorflower-gardener Jun 29, 2026
54e4646
[XLA:GPU]: Reattempt wrap all collectives in async-start for GPUs
sohaibiftikhar Jun 29, 2026
d7da299
Make `Thunk::buffer_uses` pure virtual and add empty overrides
EusebioDM Jun 29, 2026
68f97b0
Automated Code Change
tensorflower-gardener Jun 29, 2026
c3227b0
Dump the HLO before we assign configs (aka autotuning) if xla_dump_to…
shyamliagrawal Jun 29, 2026
147f6f0
[XLA:GPU] fix incorrect runtime variable mapping
metaflow Jun 29, 2026
09d11d1
Merge pull request #110665 from apoorvdarshan:fix/auto-control-deps-u…
tensorflower-gardener Jun 29, 2026
7bb1273
Merge pull request #112635 from ayushozha:fix-kaiser-window-negative-…
tensorflower-gardener Jun 29, 2026
c63bd4f
Merge pull request #119409 from dxbjavid:fix-uneven-dtensor-splits
tensorflower-gardener Jun 29, 2026
9621848
Merge pull request #112641 from ayushozha:fix/encode-png-zero-dim-crash
tensorflower-gardener Jun 29, 2026
c7fc9eb
Merge pull request #115190 from ssam18:fix/issue-115046-sqrt-tflite-i…
tensorflower-gardener Jun 29, 2026
6195e27
Merge pull request #117850 from praneethhere:fix-xla-searchsorted-nan…
tensorflower-gardener Jun 29, 2026
a5b68ee
Merge pull request #115728 from RadikHoroshev:fix/cwe770-parse-tensor…
tensorflower-gardener Jun 29, 2026
b810251
Calculate `GpuProfiler::GetScratchBytes` using BufferUses in Autotuner
EusebioDM Jun 29, 2026
a7f573b
PR #43894: [GPU] Add PDL launch insertion.
sergachev Jun 29, 2026
4bb7de8
[XLA:GPU]: Remove reduction_kind from collective_kernel_thunk
sohaibiftikhar Jun 29, 2026
cd657a6
[XLA:GPU] Extract embedded Fusion Explorer HTML template to a standal…
tensorflower-gardener Jun 29, 2026
856cdf0
Automated Code Change
tensorflower-gardener Jun 29, 2026
9451ebb
[XLA:GPU] PriorityFusion: register fusion ineligibility reasons
tensorflower-gardener Jun 29, 2026
b8a24f5
[XLA] Update documentation for async operations.
tensorflower-gardener Jun 29, 2026
cf3d3c3
PR #44389: [ROCm] Enable float and buffer checker
draganmladjenovic Jun 29, 2026
79e5c52
[XLA][HLO][Verifier] Refactor shape verification for async instructions
tensorflower-gardener Jun 29, 2026
e622e92
Remove unused dependency on tsl/platform/errors from hlo_dataflow_ana…
tensorflower-gardener Jun 29, 2026
85913ff
Fix various OOB access across TensorFlow.
cantonios Jun 29, 2026
705879d
Add support for quantize, dequantize, more activations to YNNPACK del…
dsharletg Jun 29, 2026
4523ee2
Sanitize quantization options proto comment to resolve LeakR violation.
tensorflower-gardener Jun 29, 2026
b51533e
Switch CPUAllocator via TypedAllocator to aligned operator new and de…
tensorflower-gardener Jun 29, 2026
0ff85be
Migrate custom status macros to Abseil
Jun 29, 2026
8fbd0b2
Reverts c3227b0675683963b54c0dfb2f9a82935951c702
shyamliagrawal Jun 29, 2026
29597bf
lite: Fix memory leak in GPU delegates
tensorflower-gardener Jun 29, 2026
f197d45
[15/?] remove old TPU runtime code
Jun 29, 2026
fe50b7d
Remove unnecssary include
dsharletg Jun 29, 2026
34d5745
[XLA:GPU] Fix XLA LRN mixed precision error for float16 and bfloat16 …
mooskagh Jun 29, 2026
27d075b
[XLA] Decouple tensor block summary emitter for OpenXLA
tensorflower-gardener Jun 29, 2026
acb7a09
Add warnings to HloRematerialization for long execution times and slo…
tensorflower-gardener Jun 29, 2026
83552d9
[XLA][HLO Value Tracking] Append original values of implicit results …
jcai19 Jun 29, 2026
76a2e20
Integrate StableHLO at openxla/stablehlo@c2ceaecc
mrguenther Jun 29, 2026
b9fa349
Add bounds checks in TOCO ResolveConstantPack before memcpy to preven…
dmiltr3 Jun 29, 2026
eeadbb0
[19/?] remove old TPU runtime code
Jun 29, 2026
e29971e
Add GitHub Actions workflow to check for license headers.
tensorflower-gardener Jun 29, 2026
d61ec8a
This is an internal change
tensorflower-gardener Jun 29, 2026
0bc1693
Refactor async-start value set update.
tensorflower-gardener Jun 29, 2026
b373a94
Add is_stable attribute to chlo.topk.
apivovarov Jun 30, 2026
f591ebb
Integrate LLVM at llvm/llvm-project@72c5132a2e52
boomanaiden154 Jun 30, 2026
dc3e842
Fix heap buffer overflow in StableHLO Pad due to unchecked tensor ran…
dmiltr3 Jun 30, 2026
176b52e
flatbuffer change for experimental usage
sirakiin Jun 30, 2026
31b4f96
Fix HLO translation error reporting and add xprof fallback path.
petecao Jun 30, 2026
f3b9555
Automated Code Change
tensorflower-gardener Jun 30, 2026
46f5d7e
Wrap select with if_oss in tf_monitoring_framework_deps and tf_monito…
tensorflower-gardener Jun 30, 2026
fb5413d
[20/?] remove old TPU runtime code
Jun 30, 2026
529d2da
Add api_visibility.bzl to control API visiblity
terryheo Jun 30, 2026
8c505b6
[21/?] remove old TPU runtime code
Jun 30, 2026
5b25fba
Automated Code Change
tensorflower-gardener Jun 30, 2026
f9e1459
Add dependency on delegate to combined kernel tests
dsharletg Jun 30, 2026
96f0807
Checking in experimental code
amitsabne1 Jun 30, 2026
5ba6a7d
[22/?] remove old TPU runtime code
Jun 30, 2026
896a54d
Automated Code Change
tensorflower-gardener Jun 30, 2026
f2e7ced
Automated Code Change
tensorflower-gardener Jun 30, 2026
64b09cf
Bump to the latest Shardy commit.
akuegel Jun 30, 2026
4fecf89
PR #45014: [GPU] Relay control dependencies when inlining fissioned G…
sergachev Jun 30, 2026
b1a41b7
Forward CUDA_VISIBLE_DEVICES in lit.cfg.py
akuegel Jun 30, 2026
ce129b5
[XLA] Prevent overwriting xla_metadata_payload during attribute merging.
thcmbs Jun 30, 2026
0b8b58f
PR #44885: [XLA:GPU] Track XTile scaled-dot element types
and-ivanov Jun 30, 2026
52e6a40
PR #44890: Algebraic simplifier: do not fold transposes into scatters…
sergachev Jun 30, 2026
f2e1633
PR #45013: [ROCm] Add suppoprt for exp bf16 on gfx1250
zoranjovanovic-ns Jun 30, 2026
aca5129
PR #44888: [ROCm] Fix dot_algorithms_test
mmakevic-amd Jun 30, 2026
2f7a1db
[XLA:GPU] Back up read-write buffers in device memory for float checker
tensorflower-gardener Jun 30, 2026
149bbc0
Enforce grid-level constraint for scan dimension in Triton XLA.
chsigg Jun 30, 2026
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
The diff you're trying to view is too large. We only load the first 3000 changed files.
14 changes: 14 additions & 0 deletions .bazelignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
# Copyright 2023 The TensorFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ==============================================================================
1,377 changes: 891 additions & 486 deletions .bazelrc

Large diffs are not rendered by default.

3 changes: 2 additions & 1 deletion .bazelversion
Original file line number Diff line number Diff line change
@@ -1 +1,2 @@
3.7.2
7.7.0
# NOTE: Update Bazel version in tensorflow/tools/ci_build/release/common.sh.oss
4 changes: 4 additions & 0 deletions .clang-format
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
# Run manually to reformat a file:
# clang-format -i --style=file <file>
BasedOnStyle: Google
DerivePointerAlignment: false
179 changes: 179 additions & 0 deletions .gemini/styleguide.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,179 @@
# TensorFlow PR Review Guidelines (Gemini Code Assist)

## Objective

Provide high-signal, technically rigorous, and actionable feedback on pull
requests. Prioritize correctness, API stability, performance, security, and
long-term maintainability while minimizing unnecessary or low-value comments.

## Alignment with TensorFlow Contribution Guidelines

This style guide is aligned with TensorFlow’s official contribution guidelines:
https://github.com/tensorflow/tensorflow/blob/master/CONTRIBUTING.md

The following rules are derived from TensorFlow contribution requirements and
are used by Gemini Code Assist to guide PR review feedback.

These include requirements such as mandatory test coverage, adherence to coding
standards, API stability, and consistent behavior across supported environments.

## Core Review Mindset

### 1. Evaluate Necessity and Scope

* Is this change essential?
* Does it solve a real problem for TensorFlow users?
* Is it aligned with TensorFlow’s scope and design goals?
* Does it introduce unnecessary scope or complexity?
* Does the benefit justify the long-term maintenance cost?
* **Action:** If not, clearly question the need for the change.

### 2. Challenge the Implementation

* Actively identify edge cases, failure scenarios, and incorrect assumptions.
* Validate tensor shapes, dtypes, and execution paths.
* Do not assume correctness based solely on passing tests.

### 3. Ensure Robustness

* Flag fragile or environment-dependent logic.
* Identify risks across different hardware targets (CPU, GPU, TPU).
* Ensure proper error handling and defensive checks.

### 4. Detect Low-Quality or Non-Idiomatic Code

* Identify overly generic, verbose, or context-insensitive code.
* Flag patterns that do not align with established TensorFlow practices.
* Highlight inconsistencies with existing repository patterns.

### 5. Respect Existing Design Patterns

* Follow established TensorFlow APIs and architectural conventions.
* Avoid suggesting unnecessary abstractions or structural changes.
* Maintain consistency with similar modules.

## Review Priorities

### 1. Test Coverage and Reliability (Mandatory)

* Verify that unit tests are included for any new logic, feature, or bug fix.
* Flag PRs where source code is modified without adding or updating
corresponding tests.
* Ensure tests cover edge cases and expected behavior.
* Tests must be small, fast, and deterministic.
* Flag flaky tests or reliance on external systems (e.g., network, file
system).
* Ensure tests run reliably across supported platforms (Linux, macOS,
Windows).

### 2. Code Quality and Formatting

* Enforce standard Python and C++ formatting conventions.
* Flag issues such as inconsistent indentation, poor variable naming, and
missing docstrings.
* Ensure code readability and consistency with repository standards.
* Formatting issues should be flagged, but feedback should remain concise and
avoid overwhelming developers with low-value comments.

### 3. API Stability (High Priority)

* Flag breaking changes to public APIs (`tf.*`).
* Ensure backward compatibility is preserved.
* Verify adherence to the official deprecation lifecycle.
* Maintain consistency in naming and argument structure.

### 4. Correctness and Numerical Behavior

* Validate tensor operations, shapes, and broadcasting logic.
* Identify potential numerical instability (e.g., overflow, underflow,
division by zero).

### 5. Performance and Efficiency

* Flag Python-side loops over tensors and suggest vectorized TensorFlow
operations.
* Ensure compatibility with `tf.function` and avoid retracing issues.
* Identify unnecessary memory usage or redundant computations.
* Flag hardcoded device placement (e.g., `/gpu:0`).

### 6. Security and Safe Execution

* Flag unsafe data handling, deserialization, or file operations.
* Highlight potential memory safety issues, especially in C++ kernels or
Python–C++ boundaries.
* Ensure code avoids patterns that could introduce security risks.

### 7. Dependencies and Scope

* Flag the introduction of unnecessary or heavy external dependencies.
* Ensure the change aligns with TensorFlow’s scope and does not increase
maintenance burden unnecessarily.
* Flag large pull requests that combine multiple unrelated changes (e.g., bug
fix + refactor) and recommend splitting them into smaller, focused PRs for
easier review and maintainability.

### 8. Readability and Maintainability

* Flag issues only when they affect clarity or long-term maintainability.
* Avoid suggesting purely subjective stylistic preferences that do not impact
readability or consistency.

### 9. Idiomatic TensorFlow & Model Training

* **Data Pipelines:** Flag raw in-memory tensor training when dataset size or
scalability is a concern. \
Prefer `tf.data.Dataset` with `.batch()`, `.prefetch(tf.data.AUTOTUNE)`, and
optional `.cache()` where it fits in memory to improve performance.

* **Batching:** Ensure training uses appropriate batching (`batch_size` or
dataset batching). \
Avoid feeding the entire dataset at once unless it is trivially small.

* **Vectorization:** Avoid Python loops over tensors (training or inference).
\
Prefer batched/model-level operations (`model(x)` instead of per-sample
calls).

* **Reproducibility:** Require explicit seeds in data creation and model
initialization:

```python
np.random.seed(0)
tf.random.set_seed(0)
```

* **Observability:** Require validation data (`validation_split` or validation
dataset) and meaningful metrics (e.g., `mae`, `accuracy`) to monitor
training and detect overfitting.

* **Callbacks:** Encourage use of `tf.keras.callbacks.EarlyStopping` and
`tf.keras.callbacks.ModelCheckpoint` for stable and efficient training.

## Out of Scope (Do Not Comment)

* Minor subjective stylistic preferences (e.g., personal formatting opinions)
that do not impact readability, consistency, or correctness.
* Trivial or non-impactful differences.

## Feedback Standards

* Be clear, concise, and actionable.
* Provide concrete suggestions where possible.
* Avoid vague or non-specific feedback.
* **Bad:** "This might be slow."
* **Good:** "This loop introduces Python overhead; consider vectorized
TensorFlow operations to improve performance and enable better graph
optimization."

## Review Guardrails

* Avoid speculative or uncertain feedback.
* Do not comment if no meaningful issue is identified.
* Avoid duplicate or redundant comments.
* Prioritize high-impact issues over minor observations.

## Behavior

* Provide suggestions only (do not block pull requests).
* Focus on correctness, performance, security, and API stability.
* Maintain a high signal-to-noise ratio in all feedback.
46 changes: 0 additions & 46 deletions .github/ISSUE_TEMPLATE/00-bug-issue.md

This file was deleted.

30 changes: 0 additions & 30 deletions .github/ISSUE_TEMPLATE/10-build-installation-issue.md

This file was deleted.

60 changes: 0 additions & 60 deletions .github/ISSUE_TEMPLATE/20-documentation-issue.md

This file was deleted.

23 changes: 0 additions & 23 deletions .github/ISSUE_TEMPLATE/30-feature-request.md

This file was deleted.

16 changes: 0 additions & 16 deletions .github/ISSUE_TEMPLATE/50-other-issues.md

This file was deleted.

Loading