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
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
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
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
expect [TB] Support writing new style scalar (#53496) 2021-03-12 19:03:13 -08:00
fx 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
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
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
test_img
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
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
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
test_bundled_inputs.py [Pytorch Mobile] Improve Bundled Inputs Error Checking (#52386) 2021-02-24 13:55:45 -08:00
test_complex.py
test_cpp_api_parity.py
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
test_cuda.py torch.prod backward for complex types. (#48125) 2021-03-19 09:44:08 -07:00
test_cuda_primary_ctx.py
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
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
test_expecttest.py
test_foreach.py Refactor Unary Ops tests (#49712) 2021-03-19 09:28:00 -07:00
test_function_schema.py
test_functional_autograd_benchmark.py
test_futures.py
test_fx.py Simplifes OpInfo test matrix to reduce test time (#53255) 2021-03-22 03:48:27 -07:00
test_fx_experimental.py [FX] Experimental type annotation pass using Python signatures (#53831) 2021-03-17 20:43:17 -07:00
test_indexing.py TST Replaces assertEqualIgnoreTypes with assertEqual in test_indexing (#53115) 2021-03-16 16:06:36 -07:00
test_jit.py [JIT] Add Reinplacing to MKLDNN Subgraphs (#53908) 2021-03-22 19:21:03 -07:00
test_jit_cuda_fuser.py
test_jit_disabled.py
test_jit_fuser.py
test_jit_fuser_legacy.py
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_legacy.py
test_jit_profiling.py
test_jit_py3.py
test_jit_simple.py
test_jit_string.py
test_kernel_launch_checks.py
test_license.py
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
test_metal.py
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.py
test_multiprocessing_spawn.py
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
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
test_public_bindings.py [JIT] Add torch._C._jit submodule (#52910) 2021-02-26 16:05:05 -08:00
test_pytree.py
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
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
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.py Add NNC support for aten::hardtanh (a hot operation in mobilenet v2/v3) (#52394) 2021-02-18 22:56:03 -08:00
test_tensorexpr_pybind.py [TensorExpr] Reland: "PyBind: bind ExternalCalls." (#53063) 2021-03-01 20:44:10 -08:00
test_testing.py Move mypy wrapper to tools (#54268) 2021-03-18 15:41:27 -07:00
test_throughput_benchmark.py
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
test_type_promotion.py
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
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
test_xnnpack_integration.py