Commit Graph

185228 Commits

Author SHA1 Message Date
dependabot[bot]
09a34687fb
Bump the github-actions group with 6 updates
Bumps the github-actions group with 6 updates:

| Package | From | To |
| --- | --- | --- |
| [actions/github-script](https://github.com/actions/github-script) | `7.0.1` | `8.0.0` |
| [google/osv-scanner-action](https://github.com/google/osv-scanner-action) | `2.2.2` | `2.2.3` |
| [actions/setup-python](https://github.com/actions/setup-python) | `5.6.0` | `6.0.0` |
| [ossf/scorecard-action](https://github.com/ossf/scorecard-action) | `2.4.2` | `2.4.3` |
| [github/codeql-action](https://github.com/github/codeql-action) | `3.29.11` | `3.30.5` |
| [actions/stale](https://github.com/actions/stale) | `9.1.0` | `10.0.0` |


Updates `actions/github-script` from 7.0.1 to 8.0.0
- [Release notes](https://github.com/actions/github-script/releases)
- [Commits](60a0d83039...ed597411d8)

Updates `google/osv-scanner-action` from 2.2.2 to 2.2.3
- [Release notes](https://github.com/google/osv-scanner-action/releases)
- [Commits](https://github.com/google/osv-scanner-action/compare/v2.2.2...v2.2.3)

Updates `actions/setup-python` from 5.6.0 to 6.0.0
- [Release notes](https://github.com/actions/setup-python/releases)
- [Commits](a26af69be9...e797f83bcb)

Updates `ossf/scorecard-action` from 2.4.2 to 2.4.3
- [Release notes](https://github.com/ossf/scorecard-action/releases)
- [Changelog](https://github.com/ossf/scorecard-action/blob/main/RELEASE.md)
- [Commits](05b42c6244...4eaacf0543)

Updates `github/codeql-action` from 3.29.11 to 3.30.5
- [Release notes](https://github.com/github/codeql-action/releases)
- [Changelog](https://github.com/github/codeql-action/blob/main/CHANGELOG.md)
- [Commits](3c3833e0f8...3599b3baa1)

Updates `actions/stale` from 9.1.0 to 10.0.0
- [Release notes](https://github.com/actions/stale/releases)
- [Changelog](https://github.com/actions/stale/blob/main/CHANGELOG.md)
- [Commits](5bef64f19d...3a9db7e6a4)

---
updated-dependencies:
- dependency-name: actions/github-script
  dependency-version: 8.0.0
  dependency-type: direct:production
  update-type: version-update:semver-major
  dependency-group: github-actions
- dependency-name: google/osv-scanner-action
  dependency-version: 2.2.3
  dependency-type: direct:production
  update-type: version-update:semver-patch
  dependency-group: github-actions
- dependency-name: actions/setup-python
  dependency-version: 6.0.0
  dependency-type: direct:production
  update-type: version-update:semver-major
  dependency-group: github-actions
- dependency-name: ossf/scorecard-action
  dependency-version: 2.4.3
  dependency-type: direct:production
  update-type: version-update:semver-patch
  dependency-group: github-actions
- dependency-name: github/codeql-action
  dependency-version: 3.30.5
  dependency-type: direct:production
  update-type: version-update:semver-minor
  dependency-group: github-actions
- dependency-name: actions/stale
  dependency-version: 10.0.0
  dependency-type: direct:production
  update-type: version-update:semver-major
  dependency-group: github-actions
...

Signed-off-by: dependabot[bot] <support@github.com>
2025-10-01 08:14:08 +00:00
Jorge Gorbe Moya
8ba4a99bc2 Integrate LLVM at llvm/llvm-project@d28c07b755
Updates LLVM usage to match
[d28c07b7550a](https://github.com/llvm/llvm-project/commit/d28c07b7550a)

PiperOrigin-RevId: 813490970
2025-09-30 17:24:40 -07:00
Yang Chen
38e22c5a91 #tf-data-service Clean up verbose code.
PiperOrigin-RevId: 813339176
2025-09-30 11:07:10 -07:00
Evan Brown
a2b8d7ed86 Automated Code Change
PiperOrigin-RevId: 813293486
2025-09-30 09:21:05 -07:00
A. Unique TensorFlower
5103439efa compat: Update forward compatibility horizon to 2025-09-30
PiperOrigin-RevId: 813159375
2025-09-30 02:23:49 -07:00
A. Unique TensorFlower
1e5a3d1055 Update GraphDef version to 2366.
PiperOrigin-RevId: 813159344
2025-09-30 02:11:01 -07:00
A. Unique TensorFlower
594811d378 Automated Code Change
PiperOrigin-RevId: 813141030
2025-09-30 01:48:35 -07:00
A. Unique TensorFlower
12079a0018 Automated Code Change
PiperOrigin-RevId: 813140437
2025-09-30 01:35:20 -07:00
A. Unique TensorFlower
ce66fe2866 Automated Code Change
PiperOrigin-RevId: 813139767
2025-09-30 01:21:03 -07:00
Ivo Ristovski List
2235c4739d Automatically Generated Code
PiperOrigin-RevId: 813136216
2025-09-30 01:10:32 -07:00
A. Unique TensorFlower
f4a7ee8ba3 Automated Code Change
PiperOrigin-RevId: 813132470
2025-09-30 00:57:26 -07:00
A. Unique TensorFlower
811c058af7 Automated Code Change
PiperOrigin-RevId: 813129135
2025-09-30 00:46:43 -07:00
A. Unique TensorFlower
ef62b3ee88 Automated Code Change
PiperOrigin-RevId: 813126240
2025-09-30 00:35:55 -07:00
A. Unique TensorFlower
c77f0b88ba Automated Code Change
PiperOrigin-RevId: 813114119
2025-09-29 23:55:13 -07:00
Henning Becker
f43168339f Remove if_google conditions for google/protobuf dependencies in XLA BUILD files.
Due to a recent protobuf cleanup, proto dependencies now work correctly in OSS and we can remove a bunch of if_google guards.

PiperOrigin-RevId: 813107970
2025-09-29 23:43:00 -07:00
Zac Mustin
7fc641904a Optimize Convert2DCBuffersToCppBuffers.
This function iterates through the two-dimensional `c_lists` of buffers and retrieving the C++ buffers. Now we make a time saving by pre-allocating the inner vectors to their final size instead of dynamically growing them with `push_back`.

PiperOrigin-RevId: 813107171
2025-09-29 23:34:00 -07:00
Evan Brown
e7a15d2865 Automated Code Change
PiperOrigin-RevId: 813103310
2025-09-29 23:16:10 -07:00
A. Unique TensorFlower
f46522255f Automated Code Change
PiperOrigin-RevId: 813100146
2025-09-29 23:03:48 -07:00
A. Unique TensorFlower
f2dd926783 Automated Code Change
PiperOrigin-RevId: 813087190
2025-09-29 22:22:05 -07:00
Christian Sigg
aff4d09018 Remove @xla prefix from labels in repository macros.
This change updates references within repo macros from `Label(@xla//...)` to `Label(//...)`.

They are redundant, and not explicitly referencing the xla repo is better because it's called local_xla in TensorFlow.

PiperOrigin-RevId: 813083017
2025-09-29 22:07:13 -07:00
A. Unique TensorFlower
c565e894b8 Automated Code Change
PiperOrigin-RevId: 813059572
2025-09-29 20:45:39 -07:00
A. Unique TensorFlower
8f6f08437e [XLA:Original Value] Fix bugs in original value tracking
1. When recovery table is serialized, the shape index is not considered in comparator, which causes original arrays with different shape index to be squashed into one

2. When sharding doesn't change the shape, no recovery computation is needed.

3. When deriving placeholder original array, the old shape index is used so that it's more consistent. This change is not significant. Using whatever shape index does not affect correctness.

PiperOrigin-RevId: 813047161
2025-09-29 20:10:18 -07:00
Aiden Grossman
716bcf2a66 Use triple overloads of TargetRegistry functions
The overloads accepting triples as strings are deprecated and will be removed soon.

PiperOrigin-RevId: 813035688
2025-09-29 19:19:56 -07:00
Chuan He
dcbbe2c058 Remove "Pure" traits from the op otherwise this gets optimized in MLIR CSE when there is no consumer of the op.
PiperOrigin-RevId: 813004408
2025-09-29 17:26:34 -07:00
Yang Chen
2e8980f763 #tf-data-service Add a test for sampling with replicate_on_split.
PiperOrigin-RevId: 812994177
2025-09-29 16:58:47 -07:00
Junwhan Ahn
65335818d0 Clean up IFRT futures
IFRT defined aliases for futures/promises back when future/promise implementations were in PjRt as (1) it wouldn't make sense for IFRT users to directly use PjRt futures and (2) it might make it easier for IFRT to customize future implementations later. With recent move from PjRt futures to TSL futures, we decided to converge on TSL futures as (1) is no longer a concern and implementing new features on top of TSL futures may benefit other users as well.

All users of the IFRT futures have already been migrated to directly reference TSL futures.

PiperOrigin-RevId: 812967576
2025-09-29 15:38:25 -07:00
Pauline Sho
6d3931869f Add plumbing for blockwise for TFLite interpreter
PiperOrigin-RevId: 812940180
2025-09-29 14:26:01 -07:00
Yun Peng
bfc4d2d2e4 Update pip dependency reference from @pypi_XXX//:pkg to @pypi//XXX.
The `@pypi_<name>` references are deprecated and their repo names are
an implementation detail of how an underlying library is downloaded.
The modern, supported, way is to go through the hub (`@pypi`). This
also makes the code compatible with both workspace and bzlmod.

PiperOrigin-RevId: 812915387
2025-09-29 13:31:45 -07:00
Christian Sigg
dfda2351b9 Fix //third_party/nccl:archive.BUILD label.
Not sure how this ever worked.

PiperOrigin-RevId: 812913429
2025-09-29 13:19:56 -07:00
Junwhan Ahn
d4ef26295b Replace IFRT futures/promises with TSL futures/promises
IFRT futures/promises have been deprecated in favor of TSL futures/promises.

PiperOrigin-RevId: 812879965
2025-09-29 11:53:41 -07:00
Kevin Gleason
2ef289ae0a [MHLO->StableHLO] Migrate kernelgen to use StableHLO->Linalg passes.
Allows for deletion of MHLO->Linalg

PiperOrigin-RevId: 812860062
2025-09-29 11:08:18 -07:00
Penporn Koanantakool
379a70095b [xla] Update the codeql-action version comment
The PR that updated codeql-action's version was authored by a bot and didn't update the version comment:
https://github.com/openxla/xla/pull/32008

I figured it'd be easier to edit the comment in a follow-up PR instead of asking the bot to make more changes.

PiperOrigin-RevId: 812845892
2025-09-29 10:43:00 -07:00
Yun Peng
2a5f830ee8 Use common instead of build in .bazelrc file.
This change makes most Bazel options apply to all commands (e.g., `build`, `test`, `query`, `mod`) rather than just the `build` command, which prevents external repo being refetched accidentally when switching between commands (e.g `build` vs `query`).

PiperOrigin-RevId: 812842964
2025-09-29 10:30:09 -07:00
Pedro Gonnet
1792de50d1 Rename XNN_FLAG_YIELD_WORKERS to XNN_FLAG_DONT_SPIN_WORKERS.
PiperOrigin-RevId: 812833249
2025-09-29 10:14:34 -07:00
Henning Becker
8eaa5d711f Add missing header dependencies to cudnn_frontend target
PiperOrigin-RevId: 812816175
2025-09-29 09:48:29 -07:00
dependabot[bot]
694a3a7d14 PR #32008: Bump github/codeql-action from 3.30.3 to 3.30.5
Imported from GitHub PR https://github.com/openxla/xla/pull/32008

Bumps [github/codeql-action](https://github.com/github/codeql-action) from 3.30.3 to 3.30.5.
<details>
<summary>Release notes</summary>
<p><em>Sourced from <a href="https://github.com/github/codeql-action/releases">github/codeql-action's releases</a>.</em></p>
<blockquote>
<h2>v3.30.5</h2>
<h1>CodeQL Action Changelog</h1>
<p>See the <a href="https://github.com/github/codeql-action/releases">releases page</a> for the relevant changes to the CodeQL CLI and language packs.</p>
<h2>3.30.5 - 26 Sep 2025</h2>
<ul>
<li>We fixed a bug that was introduced in <code>3.30.4</code> with <code>upload-sarif</code> which resulted in files without a <code>.sarif</code> extension not getting uploaded. <a href="https://redirect.github.com/github/codeql-action/pull/3160">#3160</a></li>
</ul>
<p>See the full <a href="https://github.com/github/codeql-action/blob/v3.30.5/CHANGELOG.md">CHANGELOG.md</a> for more information.</p>
<h2>v3.30.4</h2>
<h1>CodeQL Action Changelog</h1>
<p>See the <a href="https://github.com/github/codeql-action/releases">releases page</a> for the relevant changes to the CodeQL CLI and language packs.</p>
<h2>3.30.4 - 25 Sep 2025</h2>
<ul>
<li>We have improved the CodeQL Action's ability to validate that the workflow it is used in does not use different versions of the CodeQL Action for different workflow steps. Mixing different versions of the CodeQL Action in the same workflow is unsupported and can lead to unpredictable results. A warning will now be emitted from the <code>codeql-action/init</code> step if different versions of the CodeQL Action are detected in the workflow file. Additionally, an error will now be thrown by the other CodeQL Action steps if they load a configuration file that was generated by a different version of the <code>codeql-action/init</code> step. <a href="https://redirect.github.com/github/codeql-action/pull/3099">#3099</a> and <a href="https://redirect.github.com/github/codeql-action/pull/3100">#3100</a></li>
<li>We added support for reducing the size of dependency caches for Java analyses, which will reduce cache usage and speed up workflows. This will be enabled automatically at a later time. <a href="https://redirect.github.com/github/codeql-action/pull/3107">#3107</a></li>
<li>You can now run the latest CodeQL nightly bundle by passing <code>tools: nightly</code> to the <code>init</code> action. In general, the nightly bundle is unstable and we only recommend running it when directed by GitHub staff. <a href="https://redirect.github.com/github/codeql-action/pull/3130">#3130</a></li>
<li>Update default CodeQL bundle version to 2.23.1. <a href="https://redirect.github.com/github/codeql-action/pull/3118">#3118</a></li>
</ul>
<p>See the full <a href="https://github.com/github/codeql-action/blob/v3.30.4/CHANGELOG.md">CHANGELOG.md</a> for more information.</p>
</blockquote>
</details>
<details>
<summary>Changelog</summary>
<p><em>Sourced from <a href="https://github.com/github/codeql-action/blob/main/CHANGELOG.md">github/codeql-action's changelog</a>.</em></p>
<blockquote>
<h1>CodeQL Action Changelog</h1>
<p>See the <a href="https://github.com/github/codeql-action/releases">releases page</a> for the relevant changes to the CodeQL CLI and language packs.</p>
<h2>[UNRELEASED]</h2>
<p>No user facing changes.</p>
<h2>3.30.5 - 26 Sep 2025</h2>
<ul>
<li>We fixed a bug that was introduced in <code>3.30.4</code> with <code>upload-sarif</code> which resulted in files without a <code>.sarif</code> extension not getting uploaded. <a href="https://redirect.github.com/github/codeql-action/pull/3160">#3160</a></li>
</ul>
<h2>3.30.4 - 25 Sep 2025</h2>
<ul>
<li>We have improved the CodeQL Action's ability to validate that the workflow it is used in does not use different versions of the CodeQL Action for different workflow steps. Mixing different versions of the CodeQL Action in the same workflow is unsupported and can lead to unpredictable results. A warning will now be emitted from the <code>codeql-action/init</code> step if different versions of the CodeQL Action are detected in the workflow file. Additionally, an error will now be thrown by the other CodeQL Action steps if they load a configuration file that was generated by a different version of the <code>codeql-action/init</code> step. <a href="https://redirect.github.com/github/codeql-action/pull/3099">#3099</a> and <a href="https://redirect.github.com/github/codeql-action/pull/3100">#3100</a></li>
<li>We added support for reducing the size of dependency caches for Java analyses, which will reduce cache usage and speed up workflows. This will be enabled automatically at a later time. <a href="https://redirect.github.com/github/codeql-action/pull/3107">#3107</a></li>
<li>You can now run the latest CodeQL nightly bundle by passing <code>tools: nightly</code> to the <code>init</code> action. In general, the nightly bundle is unstable and we only recommend running it when directed by GitHub staff. <a href="https://redirect.github.com/github/codeql-action/pull/3130">#3130</a></li>
<li>Update default CodeQL bundle version to 2.23.1. <a href="https://redirect.github.com/github/codeql-action/pull/3118">#3118</a></li>
</ul>
<h2>3.30.3 - 10 Sep 2025</h2>
<p>No user facing changes.</p>
<h2>3.30.2 - 09 Sep 2025</h2>
<ul>
<li>Fixed a bug which could cause language autodetection to fail. <a href="https://redirect.github.com/github/codeql-action/pull/3084">#3084</a></li>
<li>Experimental: The <code>quality-queries</code> input that was added in <code>3.29.2</code> as part of an internal experiment is now deprecated and will be removed in an upcoming version of the CodeQL Action. It has been superseded by a new <code>analysis-kinds</code> input, which is part of the same internal experiment. Do not use this in production as it is subject to change at any time. <a href="https://redirect.github.com/github/codeql-action/pull/3064">#3064</a></li>
</ul>
<h2>3.30.1 - 05 Sep 2025</h2>
<ul>
<li>Update default CodeQL bundle version to 2.23.0. <a href="https://redirect.github.com/github/codeql-action/pull/3077">#3077</a></li>
</ul>
<h2>3.30.0 - 01 Sep 2025</h2>
<ul>
<li>Reduce the size of the CodeQL Action, speeding up workflows by approximately 4 seconds. <a href="https://redirect.github.com/github/codeql-action/pull/3054">#3054</a></li>
</ul>
<h2>3.29.11 - 21 Aug 2025</h2>
<ul>
<li>Update default CodeQL bundle version to 2.22.4. <a href="https://redirect.github.com/github/codeql-action/pull/3044">#3044</a></li>
</ul>
<h2>3.29.10 - 18 Aug 2025</h2>
<p>No user facing changes.</p>
<h2>3.29.9 - 12 Aug 2025</h2>
<p>No user facing changes.</p>
<h2>3.29.8 - 08 Aug 2025</h2>
<!-- raw HTML omitted -->
</blockquote>
<p>... (truncated)</p>
</details>
<details>
<summary>Commits</summary>
<ul>
<li><a href="3599b3baa1"><code>3599b3b</code></a> Merge pull request <a href="https://redirect.github.com/github/codeql-action/issues/3161">#3161</a> from github/update-v3.30.5-0a67bd46a</li>
<li><a href="2ca0085e58"><code>2ca0085</code></a> Update changelog for v3.30.5</li>
<li><a href="0a67bd46a0"><code>0a67bd4</code></a> Merge pull request <a href="https://redirect.github.com/github/codeql-action/issues/3160">#3160</a> from github/mbg/fix/upload-sarif</li>
<li><a href="8e34f2f3bf"><code>8e34f2f</code></a> Add changelog</li>
<li><a href="0b7fc56648"><code>0b7fc56</code></a> Fix <code>upload-sarif</code> not uploading non-<code>.sarif</code> files</li>
<li><a href="94a9b7a110"><code>94a9b7a</code></a> Merge pull request <a href="https://redirect.github.com/github/codeql-action/issues/3155">#3155</a> from github/mbg/node/no-install-in-actions</li>
<li><a href="a0ae9ba202"><code>a0ae9ba</code></a> Log what the script is doing</li>
<li><a href="b27a8ef21f"><code>b27a8ef</code></a> Exit if running in an Actions workflow</li>
<li><a href="65925679a3"><code>6592567</code></a> Merge pull request <a href="https://redirect.github.com/github/codeql-action/issues/3139">#3139</a> from github/henrymercer/fix-log-message</li>
<li><a href="fa64a7dee6"><code>fa64a7d</code></a> Merge pull request <a href="https://redirect.github.com/github/codeql-action/issues/3154">#3154</a> from github/mbg/node/check-up-to-date-deps</li>
<li>Additional commits viewable in <a href="192325c861...3599b3baa1">compare view</a></li>
</ul>
</details>
<br />

[![Dependabot compatibility score](https://dependabot-badges.githubapp.com/badges/compatibility_score?dependency-name=github/codeql-action&package-manager=github_actions&previous-version=3.30.3&new-version=3.30.5)](https://docs.github.com/en/github/managing-security-vulnerabilities/about-dependabot-security-updates#about-compatibility-scores)

Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting `@dependabot rebase`.

[//]: # (dependabot-automerge-start)
[//]: # (dependabot-automerge-end)

---

<details>
<summary>Dependabot commands and options</summary>
<br />

You can trigger Dependabot actions by commenting on this PR:
- `@dependabot rebase` will rebase this PR
- `@dependabot recreate` will recreate this PR, overwriting any edits that have been made to it
- `@dependabot merge` will merge this PR after your CI passes on it
- `@dependabot squash and merge` will squash and merge this PR after your CI passes on it
- `@dependabot cancel merge` will cancel a previously requested merge and block automerging
- `@dependabot reopen` will reopen this PR if it is closed
- `@dependabot close` will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually
- `@dependabot show <dependency name> ignore conditions` will show all of the ignore conditions of the specified dependency
- `@dependabot ignore this major version` will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself)
- `@dependabot ignore this minor version` will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself)
- `@dependabot ignore this dependency` will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)

</details>
Copybara import of the project:

--
8fd1c1421dd952414ab16e9361f8f6cd99c797cc by dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>:

Bump github/codeql-action from 3.30.3 to 3.30.5

Bumps [github/codeql-action](https://github.com/github/codeql-action) from 3.30.3 to 3.30.5.
- [Release notes](https://github.com/github/codeql-action/releases)
- [Changelog](https://github.com/github/codeql-action/blob/main/CHANGELOG.md)
- [Commits](192325c861...3599b3baa1)

---
updated-dependencies:
- dependency-name: github/codeql-action
  dependency-version: 3.30.5
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>

Merging this change closes #32008

PiperOrigin-RevId: 812816138
2025-09-29 09:27:15 -07:00
Tori Baker
1f7f20dc93 Update autotuner cache to work with --xla_gpu_experimental_enable_fusion_autotuner
Some tests in gpu_compiler_test.cc test the autotuner cache using this file. The change introduced by the flag adds another autotuner pass for transposes/reductions & therefore needs more entries in the file.

PiperOrigin-RevId: 812798127
2025-09-29 08:44:56 -07:00
Henning Becker
ba651334fe Remove @com_google_protobuf//:protobuf from XLA default dependencies.
This implicit dependency is not needed anymore but breaks test targets
when this dep is added explicitly due to duplicated dependencies.

Therefore let's remove it from here.

PiperOrigin-RevId: 812794119
2025-09-29 08:31:51 -07:00
Sean Talts
90d2d2b257 [XLA:CPU] Add CPP->LLVM IR build rules
PiperOrigin-RevId: 812791562
2025-09-29 08:21:59 -07:00
spiao
43cca495d2 PR #32009: [ROCm] fixed the build error on rocm
Imported from GitHub PR https://github.com/openxla/xla/pull/32009

🐛 Bug Fix
Fixed the build error on ROCm, as cupti_tracer is not available on ROCm platform.
It is a separate PR according to the comment in https://github.com/openxla/xla/pull/32002#discussion_r2387941595.

@xla-rotation could you review my PR, please?

Copybara import of the project:

--
d66a44ecdf3d28236ea396f68e1b166684e3bbfc by Songlin <Songlin.Piao@amd.com>:

fixed build error on rocm

Merging this change closes #32009

PiperOrigin-RevId: 812790412
2025-09-29 08:09:34 -07:00
Tori Baker
92a3b52a3f Update tests to work with --xla_gpu_experimental_enable_fusion_autotuner.
The new flag adds an autotuner pass which autotunes between block_level_emitters and native emitters and therefore the config is unpredictable. These tests are trying to test the heuristics used when there is no autotuner, so we should turn the autotuner off to preserve the tests logic.

PiperOrigin-RevId: 812783680
2025-09-29 07:49:35 -07:00
TensorFlower Gardener
72f8d3e26e Merge pull request #94375 from ekalda:gatherop-folding
PiperOrigin-RevId: 812780818
2025-09-29 07:39:03 -07:00
Tori Baker
25146bd52e Update splitk checks in gemm_fusion_autotuner_test
Getting this test to work with `--xla_gpu_experimental_enable_fusion_autotuner` - previously, the tests were assuming they could check for a reduction by checking the last part was a fusion of kind=kInput. However, the change I'm making allows reductions to go through the triton or native emitter backends. I've updated the tests to check for the fusion output of both fusions, which should be more representative of the dot+reduce patter that the test is looking for.

PiperOrigin-RevId: 812770934
2025-09-29 07:10:32 -07:00
Marcin Radomski
b97a945e70 [XLA:GPU] Enable chlo.atanh -> kAtanh HloInstruction lowering.
PiperOrigin-RevId: 812739773
2025-09-29 05:29:05 -07:00
Henning Becker
48a37b4b5d Protobuf dependency cleanup
PiperOrigin-RevId: 812731794
2025-09-29 04:51:09 -07:00
A. Unique TensorFlower
97b519d156 Automated Code Change
PiperOrigin-RevId: 812718798
2025-09-29 04:24:16 -07:00
A. Unique TensorFlower
e42f4c93bf Automated Code Change
PiperOrigin-RevId: 812717746
2025-09-29 04:16:59 -07:00
Pedro Gonnet
c4d2103362 Add an interface to initialize XNNPACK with an xnn_threadpool.
This required adding `xnn_threadpool_num_threads()` and `xnn_threadpool_schedule()` functions so that the `xnn_threadpool` could be used from `C`-language code, thus allowing a `pthreadpool` to use an `xnn_threadpool` as an executor.

To run Subgraph benchmarks with the new interface, build with `bazel build --define xnn_benchmark_use_xnn_threadpool=true`, or set `-DXNN_BENCHMARK_USE_XNN_THREADPOOL`.

PiperOrigin-RevId: 812716071
2025-09-29 04:06:05 -07:00
Karlo Basioli
8ddcd964ce [XLA:CPU] Unify JIT and AOT compilation paths.
PiperOrigin-RevId: 812711375
2025-09-29 03:37:25 -07:00
Eugene Zhulenev
616a54bccd [xla:pjrt] Move ScopedAsyncTrackingEvent to a separate target
In preparation for removing pjrt_future.h header file move ScopedAsyncTrackingEvent, which is only used by PjRtDevice to a separate target.

Reverts 85467a4e15

PiperOrigin-RevId: 812691689
2025-09-29 02:57:35 -07:00