pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00

Files

rzou 10bc36fe84 Get tensor subclasses and torch.library.triton_op to dispatch correctly (#160341 )

Short-term fix for https://github.com/pytorch/pytorch/issues/160333

The problem is:
1) `triton_op` adds a decomposition for FunctionalTensorMode for this operation
2) Tensor Subclasses rely on FunctionalTensorMode's `__torch_dispatch__` returning NotImplemented.
3) `triton_op`'s FunctionalTensorMode decomposition takes precedence over FunctionalTensorMode's decomposition.

The easy fix is to copy-paste the FunctionalTensorMode's NotImplemented
return logic into the decomposition.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/160341
Approved by: https://github.com/drisspg

2025-08-12 04:09:37 +00:00

ao/sparsity

[BE][3/6] fix typos in test/ (#157637 )

2025-07-17 12:08:33 +00:00

autograd

…

backends/xeon

…

benchmark_utils

[BE][3/6] fix typos in test/ (#157637 )

2025-07-17 12:08:33 +00:00

bottleneck_test

…

compiled_autograd_skips

[ca] skip on some PYTORCH_TEST_WITH_DYNAMO=1 autograd tests (#156374 )

2025-06-21 18:33:38 +00:00

cpp

Remove tensorexpr tests (#158928 )

2025-08-09 02:21:22 +00:00

cpp_api_parity

[BE][PYFMT] migrate PYFMT for test/[a-h]*/ to ruff format (#144555 )

2025-06-24 04:53:54 +00:00

cpp_extensions

[OpenReg] Disable automatic inclusion of data files (#159845 )

2025-08-06 10:35:13 +00:00

custom_backend

…

custom_operator

Using the latest torch.library.register_fake API instead of torch.library.impl_abstract (#158839 )

2025-07-25 02:37:30 +00:00

distributed

Revert "port distributed pipeline test files for Intel GPU (#159033 )"

2025-08-11 20:44:45 +00:00

distributions

[BE] fix remaining flake8 v7 warnings (#159044 )

2025-07-25 02:56:34 +00:00

dynamo

[inductor] fix test_different_file_paths_local_pgo on Windows. (#160382 )

2025-08-12 01:35:39 +00:00

dynamo_expected_failures

[dynamo] support itertools.permutations (#159694 )

2025-08-09 03:01:58 +00:00

dynamo_skips

Add CPython dict tests (#150791 )

2025-05-30 18:17:09 +00:00

error_messages

…

expect

[cuDNN][SDPA] cuDNN SDPA refactor/cleanup, nested tensor backward, test priority bump for sm90, sm100 (#149282 )

2025-08-08 22:22:48 +00:00

export

Fix requires_cuda to requires_cuda_and_triton (#160222 )

2025-08-10 07:05:52 +00:00

forward_backward_compatibility

Fused RMSNorm implementation (#153666 )

2025-07-22 22:25:44 +00:00

functorch

rename-HAS_CUDA-to-HAS_CUDA_AND_TRITON (#159883 )

2025-08-08 15:44:52 +00:00

Partitioner: Fix to align partition node order with original graph (#157892 )

2025-08-06 22:12:47 +00:00

higher_order_ops

[hop][exc] make UncapturedHigherOrderOpError print user code and avoid re-raise (#159296 )

2025-08-11 22:48:10 +00:00

inductor

Get tensor subclasses and torch.library.triton_op to dispatch correctly (#160341 )

2025-08-12 04:09:37 +00:00

inductor_expected_failures

…

inductor_skips

…

jit

[TorchScript, PT2] Add torch._check compatibility support (#159988 )

2025-08-08 23:14:13 +00:00

jit_hooks

…

lazy

[BE][PYFMT] migrate PYFMT for test/[i-z]*/ to ruff format (#144556 )

2025-07-29 03:26:09 +00:00

mobile

[BE][PYFMT] migrate PYFMT for test/[i-z]*/ to ruff format (#144556 )

2025-07-29 03:26:09 +00:00

[Testing] Add MPS to NATIVE_DEVICES (#153835 )

2025-08-05 18:57:35 +00:00

onnx

[ONNX] Fix the export of the model having none as output (#160200 )

2025-08-08 23:09:34 +00:00

optim

[BE][PYFMT] migrate PYFMT for {torch,test}/{nn,optim}/** to ruff format (#144548 )

2025-06-14 11:27:04 +00:00

package

[Torch Package] Make get names of OrderedImporters support fallback to importers (#155743 )

2025-08-06 02:26:10 +00:00

profiler

[Profiler] Fix unexpected C return events (#159574 )

2025-08-07 01:17:55 +00:00

quantization

Revert "unskipped mobilenet_v3 quantization and mobilenet_v2 quantization plus tests from https://github.com/pytorch/pytorch/issues/125438 (#157786 )"

2025-08-07 13:09:33 +00:00

scripts

[BE][PYFMT] migrate PYFMT for test/[i-z]*/ to ruff format (#144556 )

2025-07-29 03:26:09 +00:00

strobelight/examples

…

test_img

…

torch_np

[BE][PYFMT] migrate PYFMT for test/[i-z]*/ to ruff format (#144556 )

2025-07-29 03:26:09 +00:00

typing

More testing of Python arithmetic operators between tensors and scalars (see 157266) (#157632 )

2025-07-05 17:48:27 +00:00

xpu

Enable _int_mm on Intel GPU (#157769 )

2025-08-02 05:16:01 +00:00

_test_bazel.py

…

allowlist_for_publicAPI.json

remove guard_or_x from allowlist_for_publicAPI (#159181 )

2025-07-26 01:22:17 +00:00

bench_mps_ops.py

[BE] Remove macos-13 guard from bench_mps_ops (#159732 )

2025-08-03 20:53:58 +00:00

conftest.py

[CI][run_test] Fix rerun logic for failing at exit (#155853 )

2025-06-17 17:51:40 +00:00

create_dummy_torchscript_model.py

…

HowToWriteTestsUsingFileCheck.md

…

linear.py

…

load_torchscript_model.py

…

minioptest_failures_dict.json

…

mkl_verbose.py

…

mkldnn_verbose.py

…

pytest_shard_custom.py

…

run_doctests.sh

…

run_test.py

Actually run the einops tests in CI (#159776 )

2025-08-07 15:23:06 +00:00

simulate_nccl_errors.py

…

slow_tests.json

Update slow tests (#158222 )

2025-08-11 12:00:13 +00:00

test_accelerator.py

Add UT for torch.accelerator memory-related API (#155200 )

2025-08-08 17:41:22 +00:00

test_ao_sparsity.py

Add __main__ guards to ao tests (#154612 )

2025-06-10 18:33:09 +00:00

test_appending_byte_serializer.py

…

test_autocast.py

…

test_autograd_fallback.py

Fix TestAutogradFallback flaky tests under Dynamo: migrate to lib._destroy() (#159443 )

2025-07-30 19:30:55 +00:00

test_autograd.py

[autograd] match 0-dim gradients device type regardless of subclassness (#160165 )

2025-08-11 17:57:32 +00:00

test_autoload.py

…

test_binary_ufuncs.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_bundled_images.py

Add __main__ guards to tests (#154716 )

2025-06-04 14:38:13 +00:00

test_bundled_inputs.py

…

test_ci_sanity_check_fail.py

…

test_comparison_utils.py

…

test_compile_benchmark_util.py

…

test_complex.py

…

test_content_store.py

…

test_cpp_api_parity.py

Revert "Stop parsing command line arguments every time common_utils is imported. (#156703 )"

2025-08-04 20:37:39 +00:00

test_cpp_extensions_aot.py

[build] modernize build-frontend: python setup.py develop/install -> [uv ]pip install --no-build-isolation [-e ]. (#156027 )

2025-07-09 11:24:27 +00:00

test_cpp_extensions_jit.py

[CI] Fixes CI for CUDA Version > 12.9 (#157385 )

2025-07-19 06:51:57 +00:00

test_cpp_extensions_mtia_backend.py

…

test_cpp_extensions_stream_and_event.py

…

test_cuda_expandable_segments.py

…

test_cuda_multigpu.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_cuda_nvml_based_avail.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_cuda_primary_ctx.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_cuda_sanitizer.py

…

test_cuda_trace.py

…

test_cuda.py

Add UT for torch.accelerator memory-related API (#155200 )

2025-08-08 17:41:22 +00:00

test_custom_ops.py

[dynamo] update pytorch-labs -> meta-pytorch in graph break URLs (#159975 )

2025-08-06 23:57:31 +00:00

test_dataloader.py

skip XPU for dataloader CPU only unit test (#159811 )

2025-08-05 03:44:01 +00:00

test_datapipe.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_decomp.py

Fix full_like decomposition to preserve strides (#158898 )

2025-07-25 20:21:36 +00:00

test_determination.py

…

test_dispatch.py

Typo fixes for "overridden" in comments and function names (#155944 )

2025-06-14 03:37:38 +00:00

test_dlpack.py

[Testing] Add MPS to NATIVE_DEVICES (#153835 )

2025-08-05 18:57:35 +00:00

test_dynamic_shapes.py

Revert "Fix get_free_symbol_uses for several nodes. (#160134 )"

2025-08-10 02:37:40 +00:00

test_expanded_weights.py

Revert "Stop parsing command line arguments every time common_utils is imported. (#156703 )"

2025-08-04 20:37:39 +00:00

test_extension_utils.py

…

test_fake_tensor.py

[MTIA] Allow users who know what they are doing to ignore all device mismatches in tracing and take a preferred device. (#159931 )

2025-08-07 22:37:15 +00:00

test_file_check.py

…

test_flop_counter.py

PYFMT lint grandfathered files 1 (#154261 )

2025-05-25 17:36:14 +00:00

test_foreach.py

Fix requires_cuda to requires_cuda_and_triton (#160222 )

2025-08-10 07:05:52 +00:00

test_function_schema.py

…

test_functional_autograd_benchmark.py

…

test_functional_optim.py

…

test_functionalization_of_rng_ops.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_functionalization.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_futures.py

…

test_fx_experimental.py

Add flag to fx.passes.split_module to normalize input names (#157733 )

2025-07-08 13:47:24 +00:00

test_fx_passes.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_fx_reinplace_pass.py

[dynamic shapes] avoid unnecessary slices (#157528 )

2025-07-10 06:34:46 +00:00

test_fx.py

Extend torch function support to ALL arguments, not just scalar type (but not insides of list) (#145089 )

2025-08-07 23:43:53 +00:00

test_hop_infra.py

…

test_hub.py

Add __main__ guards to tests (#154716 )

2025-06-04 14:38:13 +00:00

test_import_stats.py

…

test_indexing.py

[Testing] Add MPS to NATIVE_DEVICES (#153835 )

2025-08-05 18:57:35 +00:00

test_itt.py

…

test_jit_autocast.py

Revert "Stop parsing command line arguments every time common_utils is imported. (#156703 )"

2025-08-04 20:37:39 +00:00

test_jit_disabled.py

…

test_jit_fuser_legacy.py

Revert "Stop parsing command line arguments every time common_utils is imported. (#156703 )"

2025-08-04 20:37:39 +00:00

test_jit_fuser_te.py

Remove tensorexpr tests (#158928 )

2025-08-09 02:21:22 +00:00

test_jit_fuser.py

Revert "Stop parsing command line arguments every time common_utils is imported. (#156703 )"

2025-08-04 20:37:39 +00:00

test_jit_legacy.py

Revert "Stop parsing command line arguments every time common_utils is imported. (#156703 )"

2025-08-04 20:37:39 +00:00

test_jit_llga_fuser.py

…

test_jit_profiling.py

…

test_jit_simple.py

…

test_jit_string.py

…

test_jit.py

Revert "Stop parsing command line arguments every time common_utils is imported. (#156703 )"

2025-08-04 20:37:39 +00:00

test_jiterator.py

…

test_kernel_launch_checks.py

…

test_legacy_vmap.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_license.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_linalg.py

Revert "unskipped mobilenet_v3 quantization and mobilenet_v2 quantization plus tests from https://github.com/pytorch/pytorch/issues/125438 (#157786 )"

2025-08-07 13:09:33 +00:00

test_logging.py

…

test_masked.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_maskedtensor.py

Fix MaskedTensor to device ignored mask (#151205 )

2025-07-21 21:44:49 +00:00

test_matmul_cuda.py

Fix grouped MM load along K when TMA loads are not used (#159485 )

2025-07-31 17:58:02 +00:00

test_meta.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_metal.py

…

test_mkl_verbose.py

…

test_mkldnn_fusion.py

…

test_mkldnn_verbose.py

…

test_mkldnn.py

Enable TF32 as fp32 internal precision for matmul/linear/conv (#157520 )

2025-07-17 08:57:34 +00:00

test_mobile_optimizer.py

…

test_model_exports_to_core_aten.py

…

test_module_tracker.py

…

test_modules.py

…

test_monitor.py

…

test_mps.py

Fix MPS autocast for ConvTranspose3d (#160345 )

2025-08-11 21:01:52 +00:00

test_multiprocessing_spawn.py

[BE] fix remaining flake8 v7 warnings (#159044 )

2025-07-25 02:56:34 +00:00

test_multiprocessing.py

Revert "[CI] Use setup-python from for Mac tests (#155698 )"

2025-06-12 14:42:32 +00:00

test_namedtensor.py

…

test_namedtuple_return_api.py

…

test_native_functions.py

…

test_native_mha.py

…

test_nestedtensor.py

[cuDNN][SDPA] cuDNN SDPA refactor/cleanup, nested tensor backward, test priority bump for sm90, sm100 (#149282 )

2025-08-08 22:22:48 +00:00

test_nn.py

[Testing] Add MPS to NATIVE_DEVICES (#153835 )

2025-08-05 18:57:35 +00:00

test_nnapi.py

…

test_numa_binding.py

NUMA binding integration with elastic agent and torchrun (#149334 )

2025-07-25 21:19:49 +00:00

test_numba_integration.py

…

test_numpy_interop.py

Throw invalid_argument instead of RuntimeError when parameters exceed… (#158267 )

2025-07-25 23:49:46 +00:00

test_openmp.py

…

test_openreg.py

[OpenReg] move fallback tests into test_openreg.py (#158441 )

2025-07-25 02:39:41 +00:00

test_ops_fwd_gradients.py

…

test_ops_gradients.py

…

test_ops_jit.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_ops.py

Relax unclaimed successes in dtype op tests when running under TEST_WITH_DYNAMO/TEST_WITH_INDUCTOR (#159976 )

2025-08-07 02:38:45 +00:00

test_optim.py

…

test_out_dtype_op.py

…

test_overrides.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_package.py

…

test_per_overload_api.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_prims.py

…

test_proxy_tensor.py

[attempt 2] Compute contiguity symbolically to avoid dde, and introduce c++ sym_is_contiguous (#157472 )

2025-07-02 23:12:29 +00:00

test_pruning_op.py

…

test_public_bindings.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_python_dispatch.py

[autograd] match 0-dim gradients device type regardless of subclassness (#160165 )

2025-08-11 17:57:32 +00:00

test_pytree.py

Preserve Enum types during torch.export serialization and deserialization (#154821 )

2025-06-08 17:30:31 +00:00

test_quantization.py

Remove pytorch quant docs since we are moving to torchao (#157766 )

2025-07-11 03:21:47 +00:00

test_reductions.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_rename_privateuse1_to_existing_device.py

[Device] Add support for PrivateUse1 device type in parse_type function (#157609 )

2025-07-17 01:27:44 +00:00

test_scatter_gather_ops.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_schema_check.py

[inductor] slow test some Windows UTs. (#160267 )

2025-08-10 18:35:42 +00:00

test_segment_reductions.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_serialization.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_set_default_mobile_cpu_allocator.py

…

test_shape_ops.py

…

test_show_pickle.py

…

test_sort_and_select.py

[Test][Easy] Use float16 dtype in test_sort_large (#159939 )

2025-08-08 09:56:44 +00:00

test_sparse_csr.py

[BE] remove torch deploy - conditionals (#158288 )

2025-07-29 17:40:49 +00:00

test_sparse_semi_structured.py

[BE] fix remaining flake8 v7 warnings (#159044 )

2025-07-25 02:56:34 +00:00

test_sparse.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_spectral_ops.py

fix numpy compatibility for 2d small list indices (#154806 )

2025-06-04 01:58:52 +00:00

test_stateless.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_static_runtime.py

…

test_subclass.py

…

test_sympy_utils.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_tensor_creation_ops.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_tensorboard.py

Fix test_tensorboard when started w/o tensorboard package (#154709 )

2025-05-30 19:18:43 +00:00

test_tensorexpr_pybind.py

…

test_tensorexpr.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_testing.py

Fix XPU CI UT test_circular_dependencies (#158189 )

2025-07-13 09:30:57 +00:00

test_throughput_benchmark.py

…

test_torch.py

[inductor] slow test some Windows UTs. (#160267 )

2025-08-10 18:35:42 +00:00

test_transformers_privateuse1.py

Refactor and Improve the OpenReg Module (#158090 )

2025-07-15 08:10:05 +00:00

test_transformers.py

[cuDNN][SDPA] cuDNN SDPA refactor/cleanup, nested tensor backward, test priority bump for sm90, sm100 (#149282 )

2025-08-08 22:22:48 +00:00

test_type_hints.py

…

test_type_info.py

…

test_type_promotion.py

[BE] Raise ValueError from torch.cat meta func (#158249 )

2025-07-20 23:49:18 +00:00

test_typing.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_unary_ufuncs.py

[inductor] slow test some Windows UTs. (#160267 )

2025-08-10 18:35:42 +00:00

test_utils_config_module.py

…

test_utils_filelock.py

…

test_utils.py

[docs] Decorator to create a deprecation warning (#155127 )

2025-06-25 18:09:04 +00:00

test_view_ops.py

[Testing] Add MPS to NATIVE_DEVICES (#153835 )

2025-08-05 18:57:35 +00:00

test_vulkan.py

…

test_weak.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +00:00

test_xnnpack_integration.py

…

test_xpu.py

Add UT for torch.accelerator memory-related API (#155200 )

2025-08-08 17:41:22 +00:00