saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test
History
Angela Yi 7e3f79914c Support functionalization for torch.map (#94558) 
We restrict:
* Output of each map iteration aliasing the input
* In-place mutation on the list element or inputs given to the map function
Pull Request resolved: https://github.com/pytorch/pytorch/pull/94558
Approved by: https://github.com/tugsbayasgalan
2023-02-14 02:40:38 +00:00
..
ao/sparsity [BE] Apply almost all remaining flake8-comprehension checks (#94676) 2023-02-12 01:01:25 +00:00
autograd
backends/xeon [BE] Prefer dash over underscore in command-line options (#94505) 2023-02-09 20:16:49 +00:00
benchmark_utils
bottleneck_test [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
cpp [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
cpp_api_parity [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
cpp_extensions Revert "update aten op overload to not use from to avoid compile errors (#89797)" 2023-02-10 11:32:25 +00:00
custom_backend [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
custom_operator [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
distributed Make DDPOptimizer work with torch._dynamo.explain() (#94749) 2023-02-14 01:33:47 +00:00
distributions [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
dynamo Raise warning if torch.compile options change without reset (#94680) 2023-02-13 20:21:04 +00:00
edge [BE] Prefer dash over underscore in command-line options (#94505) 2023-02-09 20:16:49 +00:00
error_messages
expect port sparse_mm.reduce to pytorch and optimize it on CPU (#83727) 2023-02-10 15:56:40 +00:00
forward_backward_compatibility Revert "update aten op overload to not use from to avoid compile errors (#89797)" 2023-02-10 11:32:25 +00:00
functorch Support functionalization for torch.map (#94558) 2023-02-14 02:40:38 +00:00
fx [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
inductor fix view + detach graph case for inductor (#94744) 2023-02-14 01:35:23 +00:00
jit jit trace will fail for parameter check if it contains param whose ki… (#94032) 2023-02-13 20:33:30 +00:00
jit_hooks
lazy [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
mobile [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
nn [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
onnx [ONNX] Remove the deprecated monkey patches to torch.Graph (#94747) 2023-02-14 00:08:09 +00:00
onnx_caffe2 [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
package [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
profiler [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
quantization ao migration: remove package test as this behavior is tested by other things (#94422) 2023-02-13 16:33:40 +00:00
scripts
test_img
typing
allowlist_for_publicAPI.json temp fix for segment reduce undocumented FC window (#94242) 2023-02-07 18:27:01 +00:00
conftest.py
create_dummy_torchscript_model.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
delete.py
HowToWriteTestsUsingFileCheck.md
linear.py
load_torchscript_model.py
mkl_verbose.py
mkldnn_verbose.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
run_doctests.sh
run_test.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
simulate_nccl_errors.py
test_ao_sparsity.py
test_autocast.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_autograd.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_binary_ufuncs.py [BE] Apply almost all remaining flake8-comprehension checks (#94676) 2023-02-12 01:01:25 +00:00
test_bundled_images.py
test_bundled_inputs.py [BE] Apply almost all remaining flake8-comprehension checks (#94676) 2023-02-12 01:01:25 +00:00
test_comparison_utils.py
test_complex.py
test_cpp_api_parity.py
test_cpp_extensions_aot.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_cpp_extensions_jit.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_cpp_extensions_open_device_registration.py
test_cuda.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_cuda_nvml_based_avail.py Fix NVML visible device parsing (#92315) 2023-02-13 04:25:04 +00:00
test_cuda_primary_ctx.py
test_cuda_sanitizer.py
test_cuda_trace.py
test_dataloader.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_datapipe.py [BE] Apply almost all remaining flake8-comprehension checks (#94676) 2023-02-12 01:01:25 +00:00
test_decomp.py [BE] Apply almost all remaining flake8-comprehension checks (#94676) 2023-02-12 01:01:25 +00:00
test_deploy.py
test_determination.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_dispatch.py
test_dlpack.py
test_dynamic_shapes.py Prevent sym_int from showing up in FX graph (#94595) 2023-02-11 01:43:05 +00:00
test_expanded_weights.py [decompositions] GRU decompositon with and without packed sequence (#91466) 2023-02-08 14:16:30 +00:00
test_fake_tensor.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_foreach.py [BE] Apply almost all remaining flake8-comprehension checks (#94676) 2023-02-12 01:01:25 +00:00
test_function_schema.py [frontend] Expose real_type getter for torch.Argument (#91938) 2023-01-12 01:26:50 +00:00
test_functional_autograd_benchmark.py
test_functional_optim.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_functionalization.py functionalization: fix x.is_contiguous(channels_last) (#94195) 2023-02-11 21:07:08 +00:00
test_futures.py
test_fx.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_fx_experimental.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_fx_passes.py
test_fx_reinplace_pass.py
test_hub.py
test_import_stats.py
test_indexing.py
test_itt.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_jit.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_jit_autocast.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_jit_cuda_fuser.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_jit_disabled.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_jit_fuser.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_jit_fuser_legacy.py [BE] Prefer dash over underscore in command-line options (#94505) 2023-02-09 20:16:49 +00:00
test_jit_fuser_te.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_jit_legacy.py [BE] Prefer dash over underscore in command-line options (#94505) 2023-02-09 20:16:49 +00:00
test_jit_llga_fuser.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_jit_profiling.py [BE] Prefer dash over underscore in command-line options (#94505) 2023-02-09 20:16:49 +00:00
test_jit_simple.py [BE] Prefer dash over underscore in command-line options (#94505) 2023-02-09 20:16:49 +00:00
test_jit_string.py
test_jiterator.py
test_kernel_launch_checks.py
test_legacy_vmap.py Remove deprecated torch.symeig (#70988) 2023-01-31 11:59:11 +00:00
test_license.py
test_linalg.py [BE] Add empty tensor check to _compute_linear_combination (#94245) 2023-02-07 11:31:11 +00:00
test_logging.py
test_masked.py
test_maskedtensor.py
test_matmul_cuda.py [CUDA] Drop CUDA < 11.0 test flags (#92605) 2023-01-24 04:34:06 +00:00
test_meta.py enable bf16 emb (#94163) 2023-02-12 00:05:09 +00:00
test_metal.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_mkl_verbose.py
test_mkldnn.py Optimize ConvTransposed2D with mkldnn float32 and bfloat16 on CPU (#92530) 2023-02-06 10:11:25 +00:00
test_mkldnn_fusion.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_mkldnn_verbose.py
test_mobile_optimizer.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_model_dump.py [BE] Modernize PyTorch even more for 3.8 with pyupgrade (#94520) 2023-02-10 18:02:50 +00:00
test_module_init.py AO migration: replace torch internal callsites (#94170) 2023-02-07 02:32:23 +00:00
test_modules.py [BE] Apply almost all remaining flake8-comprehension checks (#94676) 2023-02-12 01:01:25 +00:00
test_monitor.py
test_mps.py [MPS] Fix remainder op for integral dtypes (#94757) 2023-02-14 01:06:49 +00:00
test_multiprocessing.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_multiprocessing_spawn.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_namedtensor.py
test_namedtuple_return_api.py [BE] Enable more flake8-comprehensions checks (#94601) 2023-02-10 23:40:29 +00:00
test_native_functions.py
test_native_mha.py
test_nestedtensor.py Fix bug in unsqueeze_nested stride calculation (#88688) 2023-02-10 17:00:04 +00:00
test_nn.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_nnapi.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_numba_integration.py
test_numpy_interop.py
test_nvfuser_dynamo.py Refactor dynamo training backends (#93409) 2023-02-03 03:07:15 +00:00
test_nvfuser_frontend.py [NVFUSER] refactor nvfuser build (#89621) 2023-01-26 02:50:44 +00:00
test_openmp.py
test_ops.py [BE] Apply almost all remaining flake8-comprehension checks (#94676) 2023-02-12 01:01:25 +00:00
test_ops_fwd_gradients.py
test_ops_gradients.py
test_ops_jit.py
test_optim.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_overrides.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_package.py
test_per_overload_api.py
test_prims.py [NVFUSER] refactor nvfuser build (#89621) 2023-01-26 02:50:44 +00:00
test_proxy_tensor.py Streamlining guard expect tests (#94404) 2023-02-13 23:36:21 +00:00
test_pruning_op.py
test_public_bindings.py [jit] Support Awaitable type (#90863) 2023-01-30 17:38:59 +00:00
test_python_dispatch.py [BE]: Apply pyupgrade yield from and unit test alias upgrades (#94309) 2023-02-07 20:08:58 +00:00
test_pytree.py
test_quantization.py [quant][pt2e] Add a test to confirm we can set qconfig according to module_name (#91977) 2023-01-12 21:59:02 +00:00
test_reductions.py Logcumsumexp for CUDA (build-time optimized) (#94310) 2023-02-13 16:00:52 +00:00
test_scatter_gather_ops.py optimize gather performance for gnn usage on CPU (#87586) 2023-01-12 00:43:43 +00:00
test_schema_check.py
test_segment_reductions.py Make segment_reduce properly private. (#93166) 2023-02-06 18:32:23 +00:00
test_serialization.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py Added vectorized horizontal flip path for channels last for NcHW (#91806) 2023-01-23 20:15:30 +00:00
test_show_pickle.py
test_sort_and_select.py
test_sparse.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_sparse_csr.py port sparse_mm.reduce to pytorch and optimize it on CPU (#83727) 2023-02-10 15:56:40 +00:00
test_spectral_ops.py
test_stateless.py [functorch][nn] Refactor NN stateless APIs by swapping module tensors (#92536) 2023-02-08 17:31:38 +00:00
test_static_runtime.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_subclass.py
test_tensor_creation_ops.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_tensorboard.py Add missing super().setUp() to test_freezing and test_tensorboard (#94553) 2023-02-13 19:56:12 +00:00
test_tensorexpr.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_tensorexpr_pybind.py
test_testing.py [ONNX] Skip import test for experimental files (#94552) 2023-02-10 15:58:49 +00:00
test_throughput_benchmark.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_torch.py turn functionalization on in aot_autograd inference (#92857) 2023-02-13 17:48:00 +00:00
test_transformers.py [SDPA] Adds basic correctness checks (#94274) 2023-02-09 08:05:26 +00:00
test_type_hints.py
test_type_info.py
test_type_promotion.py
test_typing.py
test_unary_ufuncs.py
test_utils.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_view_ops.py Deprecate .mT,.T,.mH,.H on 0D tensors (#92143) 2023-01-17 16:54:35 +00:00
test_vulkan.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_weak.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_xnnpack_integration.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00