pytorch/test
rzou 4ac77fc6bd [HOP] Don't send HOPs to torch_dispatch (#131370)
I regretted the decision in
https://github.com/pytorch/pytorch/pull/130606. Most user
torch_dispatchs don't have enough to actually handle the HOP correctly,
so for now I'd prefer that users explicitly define the interaction
between the HOP and their torch_dispatch class.

An example is FlopCounterMode: if we allow HOPs to get passed to it, it
will ignore auto_functionalized(mm) by default but it will record flops
for mm, which is weird.

Test Plan:
- tests

Pull Request resolved: https://github.com/pytorch/pytorch/pull/131370
Approved by: https://github.com/ydwu4
2024-07-23 13:41:08 +00:00
..
ao/sparsity
autograd
backends/xeon
benchmark_utils
bottleneck_test
cpp [structural binding][11/N] Replace std::tie with structural binding (#130830) 2024-07-18 00:45:06 +00:00
cpp_api_parity
cpp_extensions Re-implement pin_memory to be device-agnostic by leveraging the Accelerator concept (#126376) 2024-07-23 01:44:15 +00:00
custom_backend
custom_operator Revert "Tighten torch.library.infer_schema input types (#130705)" 2024-07-16 12:57:11 +00:00
distributed Add decomposition for t_copy (#130939) 2024-07-23 08:29:19 +00:00
distributions [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
dynamo [HOP] Don't send HOPs to torch_dispatch (#131370) 2024-07-23 13:41:08 +00:00
dynamo_expected_failures Add hasattr for tensor variable (#131008) 2024-07-19 12:43:27 +00:00
dynamo_skips [3.12, 3.13, dynamo] simplified construction for frame f_locals/localsplus (#129185) 2024-07-12 17:56:38 +00:00
edge
error_messages [Ez][BE]: Enable new stable ruff rules (#129825) 2024-07-02 14:47:10 +00:00
expect Add decomposition for t_copy (#130939) 2024-07-23 08:29:19 +00:00
export [export] Allow non-CIA ops to be preserved (#131075) 2024-07-23 00:41:48 +00:00
forward_backward_compatibility [Inductor][Quant] Change the schema of QLinear Binary (#129049) 2024-07-02 12:36:38 +00:00
functorch Add decomposition for t_copy (#130939) 2024-07-23 08:29:19 +00:00
fx Revert D59561509: Multisect successfully blamed "D59561509: [FX][export] DCE pass, check schema for node impurity (#130395)" for one test failure (#131341) 2024-07-23 05:23:47 +00:00
higher_order_ops [BE][Easy][9/19] enforce style for empty lines in import segments in test/[e-h]*/ (#129760) 2024-07-17 14:25:29 +00:00
inductor Fixes for the extension backend tests (#130933) 2024-07-23 10:46:32 +00:00
jit [BE]: Enable a few additional ruff rules (#130700) 2024-07-17 02:06:04 +00:00
jit_hooks
lazy [BE][Easy] replace import pathlib with from pathlib import Path (#129426) 2024-06-30 01:36:07 +00:00
mobile Revert "[BE][Easy] use pathlib.Path instead of dirname / ".." / pardir (#129374)" 2024-06-29 00:47:15 +00:00
nn fix for launching kernel invalid config error when calling embedding … (#130994) 2024-07-20 08:33:29 +00:00
onnx Fix out_wrapper, _make_copy_from_view to handle all signatures (#130937) 2024-07-21 20:39:24 +00:00
optim
package
profiler [Profiler] exclude gpu_user_annotation when accumulating cuda time total (#130733) 2024-07-22 04:35:21 +00:00
quantization Revert D59561509: Multisect successfully blamed "D59561509: [FX][export] DCE pass, check schema for node impurity (#130395)" for one test failure (#131341) 2024-07-23 05:23:47 +00:00
scripts
test_img
torch_np [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
typing added type hints for __contains__ (#129653) 2024-06-30 11:49:11 +00:00
xpu Add allow_xpu to enable XPU UTs (#130312) 2024-07-17 02:40:28 +00:00
_test_bazel.py
allowlist_for_publicAPI.json Evaluate symexprs on load path of cache not write (#128997) 2024-06-20 08:55:12 +00:00
conftest.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
create_dummy_torchscript_model.py
delete.py
hi.py
HowToWriteTestsUsingFileCheck.md
linear.py
load_torchscript_model.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
minioptest_failures_dict.json
mkl_verbose.py
mkldnn_verbose.py
pytest_shard_custom.py
run_doctests.sh
run_test.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
simulate_nccl_errors.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_ao_sparsity.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_autocast.py Revert "[MPS] Add support for autocast in MPS (#99272)" 2024-07-02 12:29:51 +00:00
test_autograd.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_autograd_fallback.py
test_autoload.py [RFC] Add support for device extension autoloading (#127074) 2024-07-09 06:14:13 +00:00
test_binary_ufuncs.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_bundled_images.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_bundled_inputs.py
test_ci_sanity_check_fail.py
test_comparison_utils.py
test_compile_benchmark_util.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_complex.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_content_store.py
test_cpp_api_parity.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_cpp_extensions_aot.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_cpp_extensions_jit.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_cpp_extensions_mtia_backend.py
test_cpp_extensions_open_device_registration.py Re-implement pin_memory to be device-agnostic by leveraging the Accelerator concept (#126376) 2024-07-23 01:44:15 +00:00
test_cpp_extensions_stream_and_event.py Re-implement pin_memory to be device-agnostic by leveraging the Accelerator concept (#126376) 2024-07-23 01:44:15 +00:00
test_cuda.py Add wrappers for synchronous GPUDirect Storage APIs (#130633) 2024-07-22 14:51:24 +00:00
test_cuda_expandable_segments.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_cuda_multigpu.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_cuda_nvml_based_avail.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_cuda_primary_ctx.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_cuda_sanitizer.py
test_cuda_trace.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_custom_ops.py Revert "[custom ops] Add register_vmap for custom ops (#130589)" 2024-07-23 01:44:44 +00:00
test_dataloader.py Support IPC for Expandable Segments (#130890) 2024-07-22 16:15:01 +00:00
test_datapipe.py
test_decomp.py Add decomposition for reflection_pad{1,2,3}d_backward (#130299) 2024-07-17 21:56:00 +00:00
test_deploy.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_determination.py [Caffe2] [2/N] Remove Caffe2 from tests (#128911) 2024-06-19 00:05:50 +00:00
test_dispatch.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_dlpack.py
test_dynamic_shapes.py [dynamo][automatic_dynamic] Trigger dynamism on stride changes (#130232) 2024-07-21 03:45:54 +00:00
test_expanded_weights.py
test_fake_tensor.py FakeTensor cache SymInt support (#127596) 2024-07-21 19:26:38 +00:00
test_flop_counter.py [cuDNN][SDPA] Remove TORCH_CUDNN_SDPA_ENABLED=1, enable cuDNN SDPA by default on H100 and 2nd on other archs >= sm80 (#125343) 2024-06-30 19:22:16 +00:00
test_foreach.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_function_schema.py
test_functional_autograd_benchmark.py
test_functional_optim.py
test_functionalization.py
test_functionalization_of_rng_ops.py [dynamo][aot autograd] Silently disable default saved tensor hooks during tracing (#123196) 2024-06-14 20:28:08 +00:00
test_futures.py
test_fx.py
test_fx_experimental.py [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
test_fx_passes.py
test_fx_reinplace_pass.py
test_hub.py
test_import_stats.py
test_indexing.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_itt.py
test_jit.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_jit_autocast.py
test_jit_disabled.py
test_jit_fuser.py
test_jit_fuser_legacy.py
test_jit_fuser_te.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_jit_legacy.py
test_jit_llga_fuser.py
test_jit_profiling.py
test_jit_simple.py
test_jit_string.py
test_jiterator.py
test_kernel_launch_checks.py
test_legacy_vmap.py
test_license.py
test_linalg.py TunableOp size hotfix (#130800) 2024-07-22 23:42:26 +00:00
test_logging.py
test_masked.py
test_maskedtensor.py Set seed per sample for OpInfo tests + support for restricting to a single sample input (#128238) 2024-07-08 16:06:38 +00:00
test_matmul_cuda.py Updates to scaled_mm for rowwise scaling (#130059) 2024-07-04 00:53:17 +00:00
test_meta.py [TEST] Fix _scaled_mm tests (#130897) 2024-07-18 02:15:00 +00:00
test_metal.py
test_mkl_verbose.py
test_mkldnn.py
test_mkldnn_fusion.py
test_mkldnn_verbose.py
test_mobile_optimizer.py
test_model_dump.py
test_model_exports_to_core_aten.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_module_tracker.py
test_modules.py [BE][Easy] fix ruff rule needless-bool (SIM103) (#130206) 2024-07-14 08:17:52 +00:00
test_monitor.py
test_mps.py Add decomposition for t_copy (#130939) 2024-07-23 08:29:19 +00:00
test_multiprocessing.py Support IPC for Expandable Segments (#130890) 2024-07-22 16:15:01 +00:00
test_multiprocessing_spawn.py
test_namedtensor.py
test_namedtuple_return_api.py sdp::SDPBackend::flash_attention support PrivateUse1 (#126392) 2024-06-28 17:48:40 +00:00
test_native_functions.py
test_native_mha.py
test_nestedtensor.py [NestedTensor] Integrate sum along the jagged dimension into NestedTensor (#130425) 2024-07-18 10:48:18 +00:00
test_nn.py [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
test_nnapi.py
test_numba_integration.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_numpy_interop.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_openmp.py [BE] enable UFMT for torch/storage.py (#127706) 2024-06-27 23:16:24 +00:00
test_ops.py Add hooks for execution on intel gaudi devices - 1 (#128584) 2024-07-20 05:03:36 +00:00
test_ops_fwd_gradients.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_ops_gradients.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_ops_jit.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_optim.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_out_dtype_op.py
test_overrides.py [BE][Easy] fix ruff rule needless-bool (SIM103) (#130206) 2024-07-14 08:17:52 +00:00
test_package.py
test_per_overload_api.py
test_prims.py Infer prim tags from equivalent aten ones (#130367) 2024-07-11 20:53:52 +00:00
test_proxy_tensor.py Add decomposition for view_copy (#130938) 2024-07-21 20:39:24 +00:00
test_pruning_op.py
test_public_bindings.py Make public binding test only consider files that are packaged in the wheels (#130497) 2024-07-11 13:22:04 +00:00
test_python_dispatch.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_pytree.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_quantization.py Rename generate_numeric_debug_handle to numeric_debugger (#130590) 2024-07-15 22:42:27 +00:00
test_reductions.py Errors when 0-dim tensor of complex or bool type passed to aminmax. (#128404) 2024-06-24 21:46:49 +00:00
test_scatter_gather_ops.py
test_schema_check.py
test_segment_reductions.py
test_serialization.py Blocklist certain modules for weights_only load (#131259) 2024-07-22 18:23:21 +00:00
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_show_pickle.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_sort_and_select.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_sparse.py rm duplicate index_dtype arg (#130803) 2024-07-18 18:30:13 +00:00
test_sparse_csr.py [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
test_sparse_semi_structured.py
test_spectral_ops.py
test_stateless.py [BE] bump optree version to 0.12.1 (#130139) 2024-07-20 02:41:10 +00:00
test_static_runtime.py
test_subclass.py Only test _is_param if doing instance check on Parameter base (#130578) 2024-07-12 13:55:13 +00:00
test_sympy_utils.py Keep zero check be compatible with different sympy versions (#130729) 2024-07-16 08:39:00 +00:00
test_tensor_creation_ops.py Fix Storage.filename to not track the filename when storage was mmap-ed with MAP_PRIVATE (#128725) 2024-06-17 18:55:47 +00:00
test_tensorboard.py [dynamo][user-defined] Simplify and improve scope of UserDefinedObject var_getattr (#130169) 2024-07-08 04:10:56 +00:00
test_tensorexpr.py
test_tensorexpr_pybind.py
test_testing.py [BE] bump optree version to 0.12.1 (#130139) 2024-07-20 02:41:10 +00:00
test_throughput_benchmark.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_torch.py Re-implement pin_memory to be device-agnostic by leveraging the Accelerator concept (#126376) 2024-07-23 01:44:15 +00:00
test_transformers.py [cpu][flash attention] fix nan issue (#130014) 2024-07-10 02:33:26 +00:00
test_type_hints.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_type_info.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_type_promotion.py
test_typing.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_unary_ufuncs.py
test_utils.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_view_ops.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_vulkan.py
test_weak.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_xnnpack_integration.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_xpu.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00