pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00

Files

Jacob Szwejbka 583c4bf7d3 [Pytorch Mobile] optimize_for_mobile: Fuse Add Relu on any function (#54441 )

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54441

Similar to previous dropout one
ghstack-source-id: 124544176

Test Plan: Printed graphs before and after fusion. verified input outputs stayed the same {P299343882}

Reviewed By: kimishpatel

Differential Revision: D27014352

fbshipit-source-id: d0a9548f8743472bdd7e194efd8e8d5fe53b95b6

2021-03-23 12:11:59 -07:00

backward_compatibility

Faster backwards for cumsum and cumprod (#53711 )

2021-03-16 13:57:43 -07:00

benchmark_utils

Revert D26907093: Add repeats to Timer.collect_callgrind(...)

2021-03-17 20:14:21 -07:00

bottleneck_test

fix possible pandas import error during tensorboard tests (#29650 )

2020-02-04 14:27:46 -08:00

cpp

Use type-erased union for Buffer. (#54251 )

2021-03-19 04:58:09 -07:00

cpp_api_parity

Huber loss (#50553 )

2021-03-02 17:30:45 -08:00

cpp_extensions

Pass Scalar by reference (#53583 )

2021-03-15 23:17:06 -07:00

custom_backend

Adds a bool is_available() method to the backend contract (#53068 )

2021-03-10 00:24:16 -08:00

custom_operator

[BE] move more unittest.main() to run_tests() (#50923 )

2021-01-25 07:23:09 -08:00

distributed

[5/n][torch/elastic][upstream] Move torchelastic/agent to torch/distributed/elastic/agent (#54343 )

2021-03-22 23:15:37 -07:00

distributions

Fix distributions which don't properly honor validate_args=False (#53600 )

2021-03-10 13:16:32 -08:00

error_messages

Use latest stable flake8-bugbear in CI and fix B011 flake8 error. (#21944 )

2019-06-24 20:48:15 -07:00

expect

[TB] Support writing new style scalar (#53496 )

2021-03-12 19:03:13 -08:00

Test forward reference annotations (#53713 )

2021-03-15 19:40:26 -07:00

jit

[JIT] Add Reinplacing to MKLDNN Subgraphs (#53908 )

2021-03-22 19:21:03 -07:00

jit_hooks

[WIP] JIT Static Hooks: cpp tests (#49547 )

2021-01-20 09:12:57 -08:00

mobile

[pytorch] reintroduce static dispatch (#51957 )

2021-02-19 11:41:39 -08:00

onnx

[ONNX] Update embedding export wrt padding_idx (#53931 )

2021-03-15 10:03:53 -07:00

optim

Implement AdamW optimizer (#21250 )

2019-07-02 09:09:10 -07:00

package

[package] Make exporters write to buffer in fbcode (#54303 )

2021-03-19 19:59:35 -07:00

quantization

ns for fx: move linear activation test case to new API (#53777 )

2021-03-12 10:02:52 -08:00

scripts

Make test_torch.py pass cuda-memcheck (#29243 )

2019-12-14 20:29:58 -08:00

test_img

[PyTorch Operator] [2/n] Adding python test

2020-06-25 18:36:44 -07:00

typing/reveal

remove type_hint_tests and convert the files to use the new test style (#53167 )

2021-03-18 17:33:53 -07:00

HowToWriteTestsUsingFileCheck.md

Grammar patch 1 (.md) (#41599 )

2020-07-20 10:19:08 -07:00

linear.py

Move conv to mkldnn (#51483 )

2021-03-01 21:19:27 -08:00

run_test.py

Change sharding in ci (#54228 )

2021-03-22 08:40:34 -07:00

simulate_nccl_errors.py

Remove py2 compatible future imports (#44735 )

2020-09-16 12:55:57 -07:00

test_autograd.py

Support non-tensor inputs and outputs for checkpointed functions. (#52422 )

2021-03-19 21:29:03 -07:00

test_binary_ufuncs.py

port at::pow to structured (#53669 )

2021-03-19 14:30:48 -07:00

test_bundled_images.py

Fixes lint (#40606 )

2020-06-25 21:53:00 -07:00

test_bundled_inputs.py

[Pytorch Mobile] Improve Bundled Inputs Error Checking (#52386 )

2021-02-24 13:55:45 -08:00

test_complex.py

Add serialization logic for complex numbers (#51287 )

2021-01-28 17:25:35 -08:00

test_cpp_api_parity.py

Always use NewModuleTest instead of ModuleTest. (#44745 )

2020-09-17 07:36:39 -07:00

test_cpp_extensions_aot.py

Make test_cpp_extensions_aot handle lack of pytest more gracefully (#53740 )

2021-03-22 07:23:41 -07:00

test_cpp_extensions_jit.py

Internal gradcheck wrapper in testing._internal that sets certain flags to True (#51133 )

2021-01-29 09:13:37 -08:00

test_cuda_primary_ctx.py

Output to stderr in distributed tests. (#42139 )

2020-07-29 19:23:34 -07:00

test_cuda.py

torch.prod backward for complex types. (#48125 )

2021-03-19 09:44:08 -07:00

test_dataloader.py

Make ideep honor torch.set_num_thread changes (#53871 )

2021-03-13 11:20:44 -08:00

test_datapipe.py

[DataLoader] ZipIterDataPipe (#53554 )

2021-03-12 08:26:21 -08:00

test_dataset.py

[DataLoader] Rename Functional DataSet to DataPipe (#51488 )

2021-02-03 07:01:09 -08:00

test_determination.py

Improve logic for S3 stats gathering. Uses automatic SLOW_TESTS. (#53549 )

2021-03-10 09:37:06 -08:00

test_dispatch.py

Improve docs around Math/DefaultBackend & add PythonDispatcher class. (#50854 )

2021-01-25 23:10:36 -08:00

test_expecttest.py

[BE] replace unittest.main with run_tests (#50451 )

2021-01-13 10:33:08 -08:00

test_foreach.py

Refactor Unary Ops tests (#49712 )

2021-03-19 09:28:00 -07:00

test_function_schema.py

Stricter backward compatibility check (#45773 )

2020-10-06 01:28:48 -07:00

test_functional_autograd_benchmark.py

Skip test_functional_autograd_benchmark during code coverage (#50183 )

2021-01-07 11:17:21 -08:00

test_futures.py

[JIT/Futures] support set_exception api (#50983 )

2021-02-04 20:22:19 -08:00

test_fx_experimental.py

[FX] Experimental type annotation pass using Python signatures (#53831 )

2021-03-17 20:43:17 -07:00

test_fx.py

Simplifes OpInfo test matrix to reduce test time (#53255 )

2021-03-22 03:48:27 -07:00

test_indexing.py

TST Replaces assertEqualIgnoreTypes with assertEqual in test_indexing (#53115 )

2021-03-16 16:06:36 -07:00

test_jit_cuda_fuser.py

[ROCm] disable tests for ROCm 4.0.1 (#51510 )

2021-02-02 14:39:08 -08:00

test_jit_disabled.py

[BE] replace unittest.main with run_tests (#50451 )

2021-01-13 10:33:08 -08:00

test_jit_fuser_legacy.py

Rename jobs and cli switches for testing GraphExecutor configurations to something a little bit more sensical. (#45715 )

2020-10-05 11:43:28 -07:00

test_jit_fuser_te.py

[jit][tensorexpr] Added aten::batch_norm into fuser when in inference mode (#54204 )

2021-03-23 04:41:52 -07:00

test_jit_fuser.py

Clean up some type annotations in caffe2/test (#49943 )

2021-01-13 10:01:55 -08:00

test_jit_legacy.py

Rename jobs and cli switches for testing GraphExecutor configurations to something a little bit more sensical. (#45715 )

2020-10-05 11:43:28 -07:00

test_jit_profiling.py

Remove redundant code for unsupported Python versions (#49486 )

2021-01-06 12:45:46 -08:00

test_jit_py3.py

Workaround Python3.9 limitations in test_jit_py3 (#51088 )

2021-01-26 12:49:35 -08:00

test_jit_simple.py

Remove redundant code for unsupported Python versions (#49486 )

2021-01-06 12:45:46 -08:00

test_jit_string.py

[JIT] Allow implicit boolean conversion of containers (#51683 )

2021-02-05 00:34:35 -08:00

test_jit.py

[JIT] Add Reinplacing to MKLDNN Subgraphs (#53908 )

2021-03-22 19:21:03 -07:00

test_kernel_launch_checks.py

Fix check_kernel_launches.py for macros and provide extended context (#49365 )

2020-12-14 22:09:33 -08:00

test_license.py

concantenate LICENSE files when building a wheel (#51634 )

2021-02-08 08:28:46 -08:00

test_linalg.py

Revert D27155845: [pytorch][PR] Fixed the size of the workspace array in functions calling MAGMA

2021-03-19 08:27:18 -07:00

test_logging.py

Move pytorch distributed tests to separate folder for contbuild. (#30445 )

2020-01-22 21:16:59 -08:00

test_metal.py

[Metal] Add the Python binding for optimize_for_mobile (#46456 )

2020-10-17 10:26:25 -07:00

test_mkldnn.py

fix mkldnn_add in-place behavior (#51687 )

2021-03-16 12:54:27 -07:00

test_mobile_optimizer.py

[Pytorch Mobile] optimize_for_mobile: Fuse Add Relu on any function (#54441 )

2021-03-23 12:11:59 -07:00

test_multiprocessing_spawn.py

Add exception classification to torch.multiprocessing.spawn (#45174 )

2020-10-09 12:59:41 -07:00

test_multiprocessing.py

[BE] unified run_process_no_exception code (#49774 )

2021-01-04 13:43:09 -08:00

test_namedtensor.py

Enhance Tensor.unflatten to support -1 as the inferred size (#51955 )

2021-02-18 08:37:41 -08:00

test_namedtuple_return_api.py

Update clang-format linux hash and yaml import calls (#53932 )

2021-03-18 17:11:58 -07:00

test_native_functions.py

De-prioritise Dimname and DimnameList in python overload resolution (#51350 )

2021-03-02 23:07:08 -08:00

test_nn.py

Add padding='same' mode to conv{1,2,3}d (#45667 )

2021-03-18 16:22:03 -07:00

test_numba_integration.py

Temporary disable TestNumbaIntegration.test_from_cuda_array_interface* (#54430 )

2021-03-22 09:17:28 -07:00

test_numpy_interop.py

Mypy fixes for pytorch master (#52090 )

2021-02-17 10:39:51 -08:00

test_op_aliases.py

OpInfo: Added clamp and trunc tests with aliases (#51167 )

2021-02-10 05:36:18 -08:00

test_openmp.py

Marginally improve pytest collection for top-level test files (#53617 )

2021-03-10 08:56:39 -08:00

test_ops.py

Simplifes OpInfo test matrix to reduce test time (#53255 )

2021-03-22 03:48:27 -07:00

test_optim.py

[ROCm] Enable test in test_linalg.py, test_optim.py and test_vmap.py … (#52818 )

2021-03-04 02:29:45 -08:00

test_overrides.py

Marginally improve pytest collection for top-level test files (#53617 )

2021-03-10 08:56:39 -08:00

test_package.py

[package] split tests (#53749 )

2021-03-10 16:07:36 -08:00

test_profiler.py

Skip tests on ROCm (#53339 )

2021-03-04 21:55:34 -08:00

test_pruning_op.py

Rowwise Prune op (Add the test to OSS run_test), Make the op private. (#46131 )

2021-01-29 06:08:18 -08:00

test_public_bindings.py

[JIT] Add torch._C._jit submodule (#52910 )

2021-02-26 16:05:05 -08:00

test_pytree.py

Implement _broadcast_to_and_flatten(pytree, spec) (#46288 )

2020-10-20 07:52:14 -07:00

test_quantization.py

NS for FX: add test for a simple sparsenn model (#52092 )

2021-02-18 08:17:57 -08:00

test_reductions.py

Enabled BFloat16 support for argmax & argmin on both CPU & CUDA (#52582 )

2021-03-23 03:38:11 -07:00

test_serialization.py

Enable tests using named temp files on Windows (#49640 )

2020-12-29 09:57:35 -08:00

test_set_default_mobile_cpu_allocator.py

[Pytorch] Add python binding to use mobile cpu allocator. (#52323 )

2021-02-17 08:42:23 -08:00

test_shape_ops.py

Simplifes OpInfo test matrix to reduce test time (#53255 )

2021-03-22 03:48:27 -07:00

test_show_pickle.py

[BE] replace unittest.main with run_tests (#50451 )

2021-01-13 10:33:08 -08:00

test_sort_and_select.py

Stable sort for the CPU take 2. (#51790 )

2021-02-19 09:28:57 -08:00

test_sparse.py

Refactor tensor_new.cpp to use TensorOptions instead of DispatchKey (#54034 )

2021-03-19 09:08:32 -07:00

test_spectral_ops.py

[ROCM] Fix hipfft transform type error (#53411 )

2021-03-17 19:26:04 -07:00

test_static_runtime.py

[static runtime] Move all heavy constructor logic into InferenceModule (renamed to StaticModule) (#51564 )

2021-03-05 10:15:26 -08:00

test_tensor_creation_ops.py

Add meta device to generic device testing framework, skip NotImplementedError (#53682 )

2021-03-14 20:41:19 -07:00

test_tensorboard.py

[TB] Support writing new style scalar (#53496 )

2021-03-12 19:03:13 -08:00

test_tensorexpr_pybind.py

[TensorExpr] Reland: "PyBind: bind ExternalCalls." (#53063 )

2021-03-01 20:44:10 -08:00

test_tensorexpr.py

Add NNC support for aten::hardtanh (a hot operation in mobilenet v2/v3) (#52394 )

2021-02-18 22:56:03 -08:00

test_testing.py

Move mypy wrapper to tools (#54268 )

2021-03-18 15:41:27 -07:00

test_throughput_benchmark.py

Enable tests using named temp files on Windows (#49640 )

2020-12-29 09:57:35 -08:00

test_torch.py

Port index_copy from TH to ATen (#52203 )

2021-03-22 22:36:35 -07:00

test_type_hints.py

Move mypy wrapper to tools (#54268 )

2021-03-18 15:41:27 -07:00

test_type_info.py

Improve repr for torch.iinfo & torch.finfo (#40488 )

2020-07-10 15:22:55 -07:00

test_type_promotion.py

[numpy] torch.erf{c} : promote integer inputs to float (#48472 )

2020-11-27 15:08:40 -08:00

test_typing.py

remove type_hint_tests and convert the files to use the new test style (#53167 )

2021-03-18 17:33:53 -07:00

test_unary_ufuncs.py

[testing] OpInfo for sgn and sign (#53885 )

2021-03-22 09:39:40 -07:00

test_utils.py

Only allow hub.load() from original repo. (#54451 )

2021-03-22 20:27:54 -07:00

test_view_ops.py

Add tensor.view(dtype) (#47951 )

2021-01-08 06:55:21 -08:00

test_vmap.py

Revert "Revert D26733731: [pytorch][PR] Skip dispatch for `is_floatin… (#53242 )

2021-03-11 09:46:25 -08:00

test_vulkan.py

[vulkan] jit passes for vulkan conv2 prepack and fuse with clamp (#39282 )

2020-06-20 14:12:21 -07:00

test_xnnpack_integration.py

Mark top 3 slowest tests as slow (#46068 )

2020-10-08 21:10:03 -07:00