Selective activation checkpointing policies are defined per ATen operator. Point users to torch_remat when they instead want to annotate source-code regions, including call-site-specific decisions and arbitrary…
std::vector<X>{a, b} copies every element: the initializer_list backing array is const X[], so the vector's init-list constructor can only copy out of it. Here X is a list of N tensors. Some sites already wrote…
add hardware classification to existing test classes rename test class TestNNInitDeviceType -> TestNNInitDevice replace @parametrize("dtype", ...) with @dtypes decorator add allow_xpu=True to…
Fix the cpp_extension testing cleanup util so that it works even if there is no rm from git for windows on path while keeping the ability to navigate around windows permission error: [WinError 5] via an on error…
The checked-in file is two lines saying the real content is generated by cmake. It is: cmake/Codegen.cmake:47 writes it to ${CMAKE_BINARY_DIR}, build.bzl:16 to the Bazel output tree, and only the generated one is…
Correct spelling errors, missing words, punctuation issues, and extra whitespace across torch source files. These are comment-only and string-only changes with no functional impact. Authored with Claude…
Update the torch-xpu-ops commit to intel/torch-xpu-ops@1a3415: Avoid register spilling in elementwise ops for mixed fp32/bf16/fp16 inputs by adding vectorized type-casting loads Fix group_reduce assert on XPU to avoid…
Update the torch-xpu-ops commit to intel/torch-xpu-ops@1a3415: Avoid register spilling in elementwise ops for mixed fp32/bf16/fp16 inputs by adding vectorized type-casting loads Fix group_reduce assert on XPU to avoid…
This PR is auto-generated nightly by this action. Updates the Metamates merge rule with active team members. Pull Request resolved: #194306 Approved by: https://github.com/pytorchbot
Summary Add scheduled and manually dispatched Better Benchmark CI on an eight-GPU B200 runner. Run the merged upstream eellison/better-benchmark implementation. Export measured kernel latency and projected model latency…
Description drafted with an AI assistant, quoted below; I've reviewed the change. LpNormFunctor and lpnorm_cleanup each spell the sum-vs-max predicate twice in mutually inverse forms -- an L0 || L1 || L2 enumeration at…
Description drafted with an AI assistant, quoted below; I've reviewed the change. LpNormFunctor and lpnorm_cleanup each spell the sum-vs-max predicate twice in mutually inverse forms -- an L0 || L1 || L2 enumeration at…
This test cases executes the functional_autograd_benchmark.py, which had CUDA hardcoded as the GPU device. This commit generalizes it to the torch.accelerator API Replace hardcoded cuda references with torch.accelerator…
_PyThreadState_PushFrame is public in 3.15 (and also renamed from _PyThreadState_BumpFramePointerSlow). Using it directly allows us to remove THP_PyThreadState_BumpFramePointerSlow and its supporing definitions Pull…
Following the CRCR Onboarding Guide - Step 2, add Isalia20/pytorch-mps-ci to the allowlist at L1. The repository will run MPS tests against upstream PyTorch events on macOS arm64 runners. The relay GitHub App is…