Angela Yi 6e0b09728a [export] Remove report from draft-export output (#147558) ...

Summary: This matches the export API. To print the report, people can just do `print(ep._report)`. This information is also displayed in the terminal after the draft_export call.

Test Plan: CI

Reviewed By: SherlockNoMad

Differential Revision: D69689154

Pull Request resolved: https://github.com/pytorch/pytorch/pull/147558
Approved by: https://github.com/pianpwk

2025-02-22 00:54:29 +00:00

..

ao/sparsity

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

autograd

…

backends/xeon

…

benchmark_utils

PEP585 update - test (#145176)

2025-01-22 04:48:28 +00:00

bottleneck_test

…

cpp

Introduce cache clearing APIs for the lazy graph executor (#144489)

2025-01-29 17:38:01 +00:00

cpp_api_parity

…

cpp_extensions

xpu: support sycl with torch.utils.cpp_extension APIs (#132945)

2025-02-16 16:50:59 +00:00

custom_backend

…

custom_operator

PEP585 update - test (#145176)

2025-01-22 04:48:28 +00:00

distributed

Revert "Build a storage reader/writer to write checkpoints in HF format (#146352)"

2025-02-21 07:30:52 +00:00

distributions

Temp disable MKL in DistributionKernels.cpp (#146174)

2025-02-01 18:53:11 +00:00

dynamo

Add current cuda device index to FXGraphCache key (#147464)

2025-02-20 12:38:21 +00:00

dynamo_expected_failures

[Dynamo] support isinstance(...) check for type tuple (#146984)

2025-02-16 10:41:49 +00:00

dynamo_skips

Move Dynamo test to skip from expected_failures (#145390)

2025-01-22 19:06:39 +00:00

edge

Use std::string_view in tests (#146120)

2025-02-04 09:51:36 +00:00

error_messages

…

expect

[cuDNN][SDPA][Nested Tensor] Experimental cuDNN Nested Tensor SDPA Support (forward only) (#141178)

2025-02-21 05:22:19 +00:00

export

[export] Remove report from draft-export output (#147558)

2025-02-22 00:54:29 +00:00

forward_backward_compatibility

Clean up op BC check list (#146577)

2025-02-07 22:40:49 +00:00

functorch

[associative_scan] compile backend change to "eager" (#146973)

2025-02-21 20:21:41 +00:00

fx

[Codemod][AddExplicitStrictExportArg] caffe2/torch (#146439)

2025-02-05 22:56:54 +00:00

higher_order_ops

[BaseHOP] change hop(subgraph, operands) to hop(subgraph, *operands) (#146730)

2025-02-20 02:30:36 +00:00

inductor

[inductor][cpu] Move VNNI weight packing into AMX GEMM kernel for contiguous BMM weights (#146843)

2025-02-21 21:46:00 +00:00

inductor_expected_failures

Set enable_faithful_generator_behavior flag to True (#142513)

2025-02-08 22:42:12 +00:00

inductor_skips

[BE] Remove test_ops from FIXME_inductor_dont_reset_dynamo (#145307)

2025-01-27 18:12:39 +00:00

jit

[torch] fix exception types in custom class magic setattr/getattr (#146516)

2025-02-06 02:14:11 +00:00

jit_hooks

…

lazy

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

mobile

PEP585 update - test (#145176)

2025-01-22 04:48:28 +00:00

nn

[ROCm] miopen benchmark behavior now better aligns with cudnn (#145294)

2025-02-05 17:19:53 +00:00

onnx

PEP585: More UP006 fixes (#146392)

2025-02-20 06:18:13 +00:00

optim

Adding support for differentiable lr, weight_decay, and betas in Adam/AdamW (#143726)

2024-12-30 01:11:57 +00:00

package

Remove outdated test skipif conditions for Python3.9 (#146144)

2025-01-31 19:01:04 +00:00

profiler

Enable CUPTI on Windows (#141454)

2025-02-06 15:58:20 +00:00

quantization

add the torch.float8_e8m0fnu dtype to PyTorch (#147466)

2025-02-20 13:55:42 +00:00

scripts

…

test_img

…

torch_np

[BE]: Enable ruff SLOT checks (#146276)

2025-02-04 19:18:23 +00:00

typing

Revert "Fix non-bitwise type annotations for Tensor operators (see #145838) (#146845)"

2025-02-18 19:01:27 +00:00

xpu

[Intel GPU] Enable fp64 GEMM (#140677)

2025-02-17 08:15:55 +00:00

_test_bazel.py

…

allowlist_for_publicAPI.json

Improve typing in torch/types.py (#145237)

2025-01-28 05:29:12 +00:00

conftest.py

Apply ruff fixes to tests (#146140)

2025-02-04 05:41:01 +00:00

create_dummy_torchscript_model.py

…

delete.py

…

hi.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

[CI] Use job name to index into test times json (#147154)

2025-02-14 17:06:56 +00:00

simulate_nccl_errors.py

…

slow_tests.json

Update slow tests (#147308)

2025-02-17 12:03:40 +00:00

test_accelerator.py

Generalize pin memory logic for accelerator when non blocking copy happened (#143783)

2025-01-23 03:43:05 +00:00

test_ao_sparsity.py

…

test_autocast.py

Enable TemporaryFileName tests on Windows (#146311)

2025-02-07 06:06:18 +00:00

test_autograd_fallback.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_autograd.py

[BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408)

2025-02-04 19:07:04 +00:00

test_autoload.py

…

test_binary_ufuncs.py

Fix lerp weight type promotion (#141117)

2025-01-24 01:18:20 +00:00

test_bundled_images.py

…

test_bundled_inputs.py

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

test_ci_sanity_check_fail.py

…

test_comparison_utils.py

…

test_compile_benchmark_util.py

…

test_complex.py

…

test_content_store.py

Enable TemporaryFileName tests on Windows (#146311)

2025-02-07 06:06:18 +00:00

test_cpp_api_parity.py

Enable C++ API parity tests on AArch64 (#145370)

2025-01-30 22:42:49 +00:00

test_cpp_extensions_aot.py

xpu: support sycl with torch.utils.cpp_extension APIs (#132945)

2025-02-16 16:50:59 +00:00

test_cpp_extensions_jit.py

xpu: support sycl with torch.utils.cpp_extension APIs (#132945)

2025-02-16 16:50:59 +00:00

test_cpp_extensions_mtia_backend.py

…

test_cpp_extensions_open_device_registration.py

Update pin memory related APIs to not pass 'device' argument (#131858)

2025-01-15 17:23:35 +00:00

test_cpp_extensions_stream_and_event.py

…

test_cuda_expandable_segments.py

Revert "Use absolute path path.resolve() -> path.absolute() (#129409)"

2025-01-04 14:17:20 +00:00

test_cuda_multigpu.py

Add get_stream_from_external API for CUDA backend (#143799)

2024-12-31 11:15:59 +00:00

test_cuda_nvml_based_avail.py

…

test_cuda_primary_ctx.py

…

test_cuda_sanitizer.py

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

test_cuda_trace.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_cuda.py

Make torch.cuda.gds APIs public (#147120)

2025-02-14 17:06:50 +00:00

test_custom_ops.py

[dynamo] improved graph break messages for some common graph break sites [1/N] (#146525)

2025-02-20 00:08:13 +00:00

test_dataloader.py

Enable some tests on MacOS (#146268)

2025-02-03 05:04:24 +00:00

test_datapipe.py

Remove unactivated test (#146233)

2025-02-04 05:26:04 +00:00

test_decomp.py

[Inductor][CPU] disable bernoulli_p decomposition (#143460)

2024-12-19 11:21:35 +00:00

test_deploy.py

…

test_determination.py

…

test_dispatch.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_dlpack.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_dynamic_shapes.py

realize stride symbols in estimate_runtime (#146752)

2025-02-19 06:02:49 +00:00

test_expanded_weights.py

No actual change, just remove variable contain Tensors from global scope (#143225)

2024-12-17 16:14:25 +00:00

test_extension_utils.py

Move privateuse1 test out of test_utils and make them serial (#145380)

2025-01-23 00:31:39 +00:00

test_fake_tensor.py

support meta_tensor.to(device='cpu') under fake_mode (#146729)

2025-02-12 20:57:10 +00:00

test_file_check.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_flop_counter.py

Build RowwiseScaledMM.cu for SM89 (#145676)

2025-02-01 11:44:58 +00:00

test_foreach.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_function_schema.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_functional_autograd_benchmark.py

Enable Windows tests (#146666)

2025-02-08 00:55:20 +00:00

test_functional_optim.py

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

test_functionalization_of_rng_ops.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_functionalization.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_futures.py

…

test_fx_experimental.py

PEP585: Add noqa to necessary tests (#146391)

2025-02-12 15:29:50 +00:00

test_fx_passes.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_fx_reinplace_pass.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_fx.py

Make fx.node.map_arg() and .map_aggregate() generic (#146248)

2025-02-14 19:25:32 +00:00

test_hop_infra.py

Require that all HOPs be imported at import torch time (#145939)

2025-01-29 22:27:52 +00:00

test_hub.py

…

test_import_stats.py

…

test_indexing.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_itt.py

…

test_jit_autocast.py

PEP585 update - test (#145176)

2025-01-22 04:48:28 +00:00

test_jit_disabled.py

…

test_jit_fuser_legacy.py

…

test_jit_fuser_te.py

[BE][Ez]: FURB148 - remove useless enumerate calls (#145619)

2025-01-24 23:37:15 +00:00

test_jit_fuser.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_jit_legacy.py

…

test_jit_llga_fuser.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_jit_profiling.py

…

test_jit_simple.py

…

test_jit_string.py

PEP585 update - test (#145176)

2025-01-22 04:48:28 +00:00

test_jit.py

Enable TemporaryFileName tests on Windows (#146311)

2025-02-07 06:06:18 +00:00

test_jiterator.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_kernel_launch_checks.py

…

test_legacy_vmap.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_license.py

…

test_linalg.py

[ROCm] OCP FP8 Support for new GPUs (#146632)

2025-02-21 23:44:08 +00:00

test_logging.py

…

test_masked.py

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

test_maskedtensor.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_matmul_cuda.py

[ROCm] OCP FP8 Support for new GPUs (#146632)

2025-02-21 23:44:08 +00:00

test_meta.py

[inductor] fix index.Tensor fallback (#144736)

2025-01-16 09:38:29 +00:00

test_metal.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_mkl_verbose.py

…

test_mkldnn_fusion.py

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

test_mkldnn_verbose.py

…

test_mkldnn.py

[BE][CI] bump ruff to 0.8.4 (#143753)

2024-12-24 12:24:10 +00:00

test_mobile_optimizer.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_model_exports_to_core_aten.py

…

test_module_tracker.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_modules.py

Disable slow gradcheck for nn.Transformer ModuleInfo (#145531)

2025-01-25 00:58:03 +00:00

test_monitor.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_mps.py

[MPS] Implement and test round.decimals (#147266)

2025-02-16 23:17:13 +00:00

test_multiprocessing_spawn.py

[BE][CI] bump ruff to 0.8.4 (#143753)

2024-12-24 12:24:10 +00:00

test_multiprocessing.py

[dynamo] Re-enable test_fs family for dynamo (#145302)

2025-01-22 17:50:05 +00:00

test_namedtensor.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_namedtuple_return_api.py

…

test_native_functions.py

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

test_native_mha.py

…

test_nestedtensor.py

[cuDNN][SDPA][Nested Tensor] Experimental cuDNN Nested Tensor SDPA Support (forward only) (#141178)

2025-02-21 05:22:19 +00:00

test_nn.py

Use float data type for Half sum in fallback implementation of batchnorm backward on CPU (#147353)

2025-02-21 01:33:33 +00:00

test_nnapi.py

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

test_numba_integration.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_numpy_interop.py

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

test_openmp.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_ops_fwd_gradients.py

…

test_ops_gradients.py

…

test_ops_jit.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_ops.py

[inductor] add size-asserts for fallback ops (#145904)

2025-02-07 18:44:32 +00:00

test_optim.py

Fix loading older state_dict into AdamW after refactor (#144972)

2025-01-16 19:50:31 +00:00

test_out_dtype_op.py

[Codemod][AddExplicitStrictExportArg] caffe2/test (#143688)

2024-12-27 07:58:44 +00:00

test_overrides.py

Use std::string_view in tests (#146120)

2025-02-04 09:51:36 +00:00

test_package.py

…

test_per_overload_api.py

…

test_prims.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_proxy_tensor.py

Add fake_impl for unique_consecutive (#145649)

2025-01-29 22:33:16 +00:00

test_pruning_op.py

…

test_public_bindings.py

Remove public_allowlist from TestPublicBindings.test_correct_module_names and ensure private_allowlist-ed things are actually private (#145620)

2025-01-27 17:30:02 +00:00

test_python_dispatch.py

Delete torch._library.register_functional_op (#145110)

2025-01-18 00:58:25 +00:00

test_pytree.py

[pytree][Easy] preserve dict keys in insertion order in CXX pytree (#130140)

2025-02-12 16:41:49 +00:00

test_quantization.py

Add support for prototype affine quantization in pt2e flow (#141421)

2024-12-24 04:22:18 +00:00

test_reductions.py

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

test_scatter_gather_ops.py

…

test_schema_check.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_segment_reductions.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_serialization.py

Use __qualname__ in add_safe_globals and update Unpickling error raised for Unsupported GLOBAL (#146815)

2025-02-21 18:04:59 +00:00

test_set_default_mobile_cpu_allocator.py

…

test_shape_ops.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_show_pickle.py

…

test_sort_and_select.py

Fix linter F821 error (#146665)

2025-02-08 07:19:37 +00:00

test_sparse_csr.py

[CUDA][SDPA] Compute reference in test_triton_scaled_dot_product_attention_block_size_16_cuda_float32 in float64 (#146461)

2025-02-06 23:28:56 +00:00

test_sparse_semi_structured.py

[TEST][Sparse] Force CUTLASS backend in TestSparseSemiStructuredCUTLASS (#146398)

2025-02-04 22:07:12 +00:00

test_sparse.py

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

test_spectral_ops.py

Re-add stft option to align window for center = false (#146379)

2025-02-06 14:07:13 +00:00

test_stateless.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_static_runtime.py

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

test_subclass.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_sympy_utils.py

[Inductor] Expand Identity ops prior to block pattern matching (#146000)

2025-02-08 18:11:53 +00:00

test_tensor_creation_ops.py

[BE] Unskip some tensor creation tests on Mac (#146952)

2025-02-12 01:59:15 +00:00

test_tensorboard.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_tensorexpr_pybind.py

…

test_tensorexpr.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_testing.py

Enable some tests on Windows (#146243)

2025-02-05 03:54:28 +00:00

test_throughput_benchmark.py

Fix Throughputbenchmark issue (#144669)

2025-01-26 03:37:20 +00:00

test_torch.py

Enable some tests on Windows (#146243)

2025-02-05 03:54:28 +00:00

test_transformers.py

[cuDNN][SDPA][Nested Tensor] Experimental cuDNN Nested Tensor SDPA Support (forward only) (#141178)

2025-02-21 05:22:19 +00:00

test_type_hints.py

Revert "Use absolute path path.resolve() -> path.absolute() (#129409)"

2025-01-04 14:17:20 +00:00

test_type_info.py

…

test_type_promotion.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_typing.py

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

test_unary_ufuncs.py

Enable some tests on Windows (#146243)

2025-02-05 03:54:28 +00:00

test_utils_config_module.py

Add check that envvar configs are boolean (#145454)

2025-02-05 19:40:10 +00:00

test_utils_filelock.py

…

test_utils.py

[utils] add try_import method for importing optional modules (#145528)

2025-01-25 00:14:07 +00:00

test_view_ops.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_vulkan.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_weak.py

Fix unused Python variables in test/[e-z]* (#136964)

2024-12-18 23:02:30 +00:00

test_xnnpack_integration.py

[BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408)

2025-02-04 19:07:04 +00:00

test_xpu.py

Fix test_device_memory_allocated (#147311)

2025-02-17 19:00:53 +00:00