RepoJournal
PyTorch

@pytorch

PyTorch and the broader machine-learning ecosystem

Keep up with PyTorch in about 3 minutes a day: what actually shipped — the commits, pull requests, releases, and security advisories that matter.

One email a day. Unsubscribe in one click.

Pick a date

Topics: Python AI / ML Full archive →

The Wire · Showcase

TORCH.COMPILE GAINS PYTHON COMPARISON OPERATORS, TESTS GO DEVICE-AGNOSTIC ACROSS THREE REPOS

By RepoJournal · Filed · About PyTorch

Dynamo now handles Python's richcompare protocol, closing a critical gap in torch.compile's operator support while the ecosystem pivots hard toward device-generic testing.

The biggest move overnight: PyTorch core shipped richcompare support [1] by implementing CPython's full do_richcompare dispatch on VariableTracker, meaning torch.compile can now handle all comparison operators the way Python does. This fixes a class of compile failures that hit when models use comparison chains or custom __eq__ implementations.

In parallel, the test infrastructure is reshaping around device generics. TestFP8Matmul [2] got device-agnostic by replacing 37 hardcoded cuda references with a device parameter, and test_modes.py [3] ditched GPU_TYPE constants for torch.accelerator APIs. This matters: out-of-tree accelerators can now run quantization and dynamo tests without forking test files. ExecutorTorch's NXP backend [6] is already moving in lockstep with Neutron MLIR flow tests for tensor ops.

Two infrastructure cleanups landed: dead THNN/UCC/TH CMake references [4] got removed, and graph partition buffer signatures [5] now preserve globals correctly. The ao repo [7] added a defensive check to skip PT2E inductor registration when mkldnn isn't available, preventing silent failures on CPU builds.

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] [dynamo] mimic tp_richcompare handling (#182759) pytorch/pytorch
  2. [2] [Test] Make TestFP8Matmul device-generic for out-of-tree accelerators (#184771) pytorch/pytorch
  3. [3] [TEST]Make test_modes.py device-generic: Replace GPU_TYPE constant with torch.accelerator API (#179893) pytorch/pytorch
  4. [4] Remove dead THNN/UCC/TH references from CMakeLists (#185062) pytorch/pytorch
  5. [5] Fix graph partition removed buffer signatures (#184004) pytorch/pytorch
  6. [6] NXP backend: Enable Add Tensor with new Neutron flow ↗ pytorch/executorch
  7. [7] [CPU] skip registration of PT2E inductor passes if mkldnn not available ↗ pytorch/ao

Quick answers

What shipped in PyTorch on May 25, 2026?
Dynamo now handles Python's richcompare protocol, closing a critical gap in torch.compile's operator support while the ecosystem pivots hard toward device-generic testing. In total, 31 commits and 2 pull requests landed.
Who contributed to PyTorch on May 25, 2026?
2 developers shipped this update, including roman-janik-nxp and Xia-Weiwen.
What were the notable PyTorch updates?
[dynamo] mimic tp_richcompare handling (#182759), [Test] Make TestFP8Matmul device-generic for out-of-tree accelerators (#184771), and [TEST]Make test_modes.py device-generic: Replace GPU_TYPE constant with torch.accelerator API (#179893).

More from @pytorch

Daily updates, in your inbox

Follow PyTorch

Keep up with PyTorch in about 3 minutes a day: what actually shipped — the commits, pull requests, releases, and security advisories that matter.

or

One email a day. Unsubscribe in one click. Read a past issue →

Elsewhere on the wire

Want every project, not just this one?