pytorch/test
Edward Z. Yang 7b9d250f06 Change _dynamo.export to be export(f)(*args, **kwargs) (#106109)
Signed-off-by: Edward Z. Yang <ezyang@meta.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/106109
Approved by: https://github.com/voznesenskym
2023-07-27 21:41:13 +00:00
..
_nvfuser
ao/sparsity [BE] Enable ruff's UP rules and autoformat ao/ (#105430) 2023-07-19 13:44:37 +00:00
autograd Fix bug in basicAutogradNotImplementedFallback (#105660) 2023-07-25 18:41:22 +00:00
backends/xeon
benchmark_utils [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
bottleneck_test
cpp Change _dynamo.export to be export(f)(*args, **kwargs) (#106109) 2023-07-27 21:41:13 +00:00
cpp_api_parity [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
cpp_extensions
custom_backend [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
custom_operator [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
distributed Clean up unsed MHA code to avoid confusion (#105956) 2023-07-27 17:10:17 +00:00
distributions use UBSAN on some tests (#103655) 2023-07-24 14:24:49 +00:00
dynamo Change _dynamo.export to be export(f)(*args, **kwargs) (#106109) 2023-07-27 21:41:13 +00:00
edge [torchgen] Rename executorch's RuntimeContext to KernelRuntimeContext (#104892) 2023-07-14 21:15:50 +00:00
error_messages [BE] Enable ruff's UP rules and autoformat dynamo / functorch and refs (#105432) 2023-07-19 13:48:44 +00:00
expect [pt2] add decomps for multilabel_margin_loss_forward ops (#105302) 2023-07-23 02:16:29 +00:00
export [export] Remove experimental runtime assertion configs from export API. (#105043) 2023-07-26 16:21:29 +00:00
forward_backward_compatibility Meff Attn Bias (#104310) 2023-07-26 15:51:59 +00:00
functorch [pt2] add meta for argsort.stable, use sort samples in OpInfo (#106025) 2023-07-27 03:49:17 +00:00
fx Change _dynamo.export to be export(f)(*args, **kwargs) (#106109) 2023-07-27 21:41:13 +00:00
inductor [Inductor] Fuse non-foreach ops with foreach ops without iterating over all subnodes (#106008) 2023-07-27 21:40:24 +00:00
jit [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
jit_hooks
lazy
mobile [BE] Enable ruff's UP rules in pyproject.toml (#105437) 2023-07-21 19:14:52 +00:00
nn [BE] Enable ruff's UP rules in pyproject.toml (#105437) 2023-07-21 19:14:52 +00:00
onnx Change _dynamo.export to be export(f)(*args, **kwargs) (#106109) 2023-07-27 21:41:13 +00:00
onnx_caffe2 [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
optim Implement NAdamW optimizer (#103881) 2023-07-24 19:29:26 +00:00
package [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
profiler [profiler] add option for kineto synchronization events in the trace (#105187) 2023-07-26 03:45:04 +00:00
quantization [quant][pt2e] store scale/zero_point as tensor attributes to support serialization (#105894) 2023-07-26 20:15:06 +00:00
scripts
test_img
typing
_test_bazel.py
allowlist_for_publicAPI.json Add torch.utils to the docs page, remove dead code and fix docstrings (#105142) 2023-07-26 14:24:58 +00:00
conftest.py
create_dummy_torchscript_model.py
delete.py
HowToWriteTestsUsingFileCheck.md
linear.py
load_torchscript_model.py
mkl_verbose.py
mkldnn_verbose.py
run_doctests.sh
run_test.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
simulate_nccl_errors.py
test_ao_sparsity.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_autocast.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
test_autograd.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_binary_ufuncs.py add lerp cpu support for half (#105607) 2023-07-21 20:29:05 +00:00
test_bundled_images.py
test_bundled_inputs.py
test_comparison_utils.py
test_compile_benchmark_util.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_complex.py
test_content_store.py
test_cpp_api_parity.py
test_cpp_extensions_aot.py Use random inputs for mps extension tests (#104597) 2023-07-06 07:14:56 +00:00
test_cpp_extensions_jit.py [ROCm] Skip test_jit_cudnn_extension on ROCm (#105805) 2023-07-24 14:19:56 +00:00
test_cpp_extensions_open_device_registration.py Unify TEST_CUDNN definition (#105594) 2023-07-20 16:10:26 +00:00
test_cuda.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
test_cuda_expandable_segments.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_cuda_multigpu.py Revert "Fix test failure in TestCudaMultiGPU.test_cuda_device_memory_allocated (#105501)" 2023-07-24 14:18:44 +00:00
test_cuda_nvml_based_avail.py
test_cuda_primary_ctx.py
test_cuda_sanitizer.py
test_cuda_trace.py
test_custom_op_testing.py
test_dataloader.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
test_datapipe.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_decomp.py
test_deploy.py
test_determination.py
test_dispatch.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
test_dlpack.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_dynamic_shapes.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_expanded_weights.py
test_fake_tensor.py Improve FakeTensor to work with mixed meta-cpu embedding bag arguments (#105924) 2023-07-26 01:19:08 +00:00
test_flop_counter.py
test_foreach.py Reland "Add forward mode AD to out-place foreach functions (#102409) (#106043) 2023-07-27 03:13:24 +00:00
test_function_schema.py
test_functional_autograd_benchmark.py
test_functional_optim.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_functionalization.py
test_functionalization_of_rng_ops.py
test_futures.py [Reland] Update mypy to 1.4.1 (#105227) 2023-07-15 20:30:20 +00:00
test_fx.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
test_fx_experimental.py Back out "Make adding buffers more like adding parameters (#104069)" (#105581) 2023-07-20 03:39:53 +00:00
test_fx_passes.py
test_fx_reinplace_pass.py
test_hub.py [BE]: Apply ruff PERF fixes to torch (#104917) 2023-07-11 20:45:21 +00:00
test_import_stats.py
test_indexing.py
test_itt.py
test_jit.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
test_jit_autocast.py
test_jit_cuda_fuser.py Remove set_default_dtype calls from jit and ops tests (#105072) 2023-07-15 03:18:33 +00:00
test_jit_disabled.py
test_jit_fuser.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_jit_fuser_legacy.py
test_jit_fuser_te.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_jit_legacy.py
test_jit_llga_fuser.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_jit_profiling.py
test_jit_simple.py
test_jit_string.py
test_jiterator.py
test_kernel_launch_checks.py
test_legacy_vmap.py [BE]: Apply ruff PERF fixes to torch (#104917) 2023-07-11 20:45:21 +00:00
test_license.py
test_linalg.py test_linalg: triangular_solve - make well_conditioned well conditioned (#105919) 2023-07-26 09:21:12 +00:00
test_logging.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_masked.py
test_maskedtensor.py
test_matmul_cuda.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_meta.py [pt2] add metas for _cholesky_solve_helper and cholesky_solve (#105867) 2023-07-25 20:21:47 +00:00
test_metal.py
test_mkl_verbose.py
test_mkldnn.py [BE]: Apply ruff PERF fixes to torch (#104917) 2023-07-11 20:45:21 +00:00
test_mkldnn_fusion.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
test_mkldnn_verbose.py
test_mobile_optimizer.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_model_dump.py
test_module_init.py
test_modules.py Added ModuleInfo test for meta device ctx init (#105871) 2023-07-26 01:57:54 +00:00
test_monitor.py
test_mps.py use UBSAN on some tests (#103655) 2023-07-24 14:24:49 +00:00
test_multiprocessing.py
test_multiprocessing_spawn.py
test_namedtensor.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
test_namedtuple_return_api.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_native_functions.py
test_native_mha.py
test_nestedtensor.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_nn.py [easy] improve hint on error message in nn.Module.load_state_dict (#106042) 2023-07-27 19:56:02 +00:00
test_nnapi.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_numba_integration.py
test_numpy_interop.py
test_nvfuser_dynamo.py
test_nvfuser_frontend.py
test_openmp.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_ops.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
test_ops_fwd_gradients.py
test_ops_gradients.py
test_ops_jit.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_optim.py
test_out_dtype_op.py [export] Remove experimental runtime assertion configs from export API. (#105043) 2023-07-26 16:21:29 +00:00
test_overrides.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_package.py
test_per_overload_api.py
test_prims.py Fix aten.logspace decomposition (#105201) 2023-07-22 04:10:20 +00:00
test_proxy_tensor.py Update matmul decomp to match eager (#105850) 2023-07-26 09:24:51 +00:00
test_pruning_op.py
test_public_bindings.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_python_dispatch.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_pytree.py Serialize empty pytree cases (#105159) 2023-07-13 23:02:59 +00:00
test_quantization.py add basic cuda support for float8 dtypes (#105807) 2023-07-25 03:43:36 +00:00
test_reductions.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_scatter_gather_ops.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_schema_check.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_segment_reductions.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_serialization.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py
test_show_pickle.py
test_sort_and_select.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_sparse.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_sparse_csr.py Add SparseCsrCPU and SparseCsrCUDA dispatch to sum.dim_IntList (#99292) 2023-07-24 17:30:58 +00:00
test_sparse_semi_structured.py Update sparse semi-structured linear operator (#104608) 2023-07-13 23:52:39 +00:00
test_spectral_ops.py
test_stateless.py Back out "Make adding buffers more like adding parameters (#104069)" (#105581) 2023-07-20 03:39:53 +00:00
test_static_runtime.py
test_subclass.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_sympy_utils.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_tensor_creation_ops.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_tensorboard.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_tensorexpr.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_tensorexpr_pybind.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_testing.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_throughput_benchmark.py
test_torch.py [PyTorch] Add tests for empty tensors w/storage null data_ptr (#101426) 2023-07-27 05:19:42 +00:00
test_transformers.py [BE] Improve test_transformers test structure (#105938) 2023-07-26 22:16:20 +00:00
test_type_hints.py
test_type_info.py
test_type_promotion.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_typing.py
test_unary_ufuncs.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
test_utils.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
test_view_ops.py
test_vulkan.py
test_weak.py
test_xnnpack_integration.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00