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
Elias Ellison e1d927e552 [JIT] Update freezing api (#52337) 
Summary:
Update freezing api  for 1.8,  and add a corresponding C++ API. The `optimize` flag hasn't been publicly released yet, so we are able to change it without breaking BC. I will submit a PR to branch release as well, there are a few more days to do that

Pull Request resolved: https://github.com/pytorch/pytorch/pull/52337

Reviewed By: ejguan

Differential Revision: D26491833

Pulled By: eellison

fbshipit-source-id: 6dcd74eb8f76db64ac53183d03dabdd0f101f4b5
2021-02-18 00:17:27 -08:00
..
backward_compatibility make forward AD API private (#51693) 2021-02-04 19:02:29 -08:00
benchmark_utils
bottleneck_test
cpp [JIT] Update freezing api (#52337) 2021-02-18 00:17:27 -08:00
cpp_api_parity Add PixelUnshuffle (#49334) 2020-12-22 20:14:55 -08:00
cpp_extensions Move generator state APIs to ATen (#49589) 2021-01-06 18:26:56 -08:00
custom_backend Removes deprecated preprocess method from the backend interface (#52258) 2021-02-17 17:53:36 -08:00
custom_operator [BE] move more unittest.main() to run_tests() (#50923) 2021-01-25 07:23:09 -08:00
distributed [Resubmit] Add compare_set operation and test to TCPStore (#51815) 2021-02-08 13:44:31 -08:00
distributions Fix Dirichlet.arg_constraints event_dim (#51369) 2021-02-02 10:26:45 -08:00
error_messages
expect [Usability] Capture argument names for traced functions and modules (#51775) 2021-02-10 18:28:08 -08:00
fx Extend subgraph_rewriter logic (#51532) 2021-02-03 18:14:37 -08:00
jit [JIT] Update freezing api (#52337) 2021-02-18 00:17:27 -08:00
jit_hooks [WIP] JIT Static Hooks: cpp tests (#49547) 2021-01-20 09:12:57 -08:00
mobile [PyTorch Mobile] Support torchbind custom classes in lite interpreter (#51432) 2021-02-03 21:57:19 -08:00
namespace_b
onnx Warn about floor_divide performing incorrect rounding (#50281) (#50281) 2021-02-10 03:13:34 -08:00
optim
package_a [package] mangle imported module names (#50049) 2021-01-13 16:32:36 -08:00
quantization [quant] Quantizable MultiheadAttention (#49866) 2021-02-17 12:36:30 -08:00
scripts
test_img
type_hint_tests Fix mypy type hint for AdaptiveAvgPool2,3d, AdaptiveMaxPool2,3d (#49963) 2021-01-06 09:47:15 -08:00
HowToWriteTestsUsingFileCheck.md
module_a.py
run_test.py [Pytorch] Add python binding to use mobile cpu allocator. (#52323) 2021-02-17 08:42:23 -08:00
simulate_nccl_errors.py
test_autograd.py Fix autograd when inputs contains tensors without materialized grad_fn (#51940) 2021-02-11 09:22:15 -08:00
test_binary_ufuncs.py Warn about floor_divide performing incorrect rounding (#50281) (#50281) 2021-02-10 03:13:34 -08:00
test_bundled_images.py
test_bundled_inputs.py [Pytorch] Expanded Bundled Inputs To Any Public Function (#51153) 2021-02-02 10:33:59 -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
test_cpp_extensions_aot.py
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.py fix torch.nn.parallel.scatter_gather.gather to handle NamedTuples and handle moving output to CPU (#51104) 2021-02-11 15:50:28 -08:00
test_cuda_primary_ctx.py
test_dataloader.py Clean up usage of torch._six partially (#49785) 2021-02-08 13:58:34 -08:00
test_datapipe.py [DataLoader] Fix module import (#52224) 2021-02-16 16:12:33 -08:00
test_dataset.py [DataLoader] Rename Functional DataSet to DataPipe (#51488) 2021-02-03 07:01:09 -08:00
test_determination.py
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 Revert D25074763: [WIP] Update foreach APIs to use scalar lists 2021-02-03 17:06:40 -08:00
test_function_schema.py
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.py Support AST rewriting for submodules (#52297) 2021-02-17 09:08:07 -08:00
test_fx_experimental.py [WIP][FX] Normalize torch.nn.functional calls (#51816) 2021-02-17 15:18:03 -08:00
test_indexing.py index_put_ for complex tensors on CUDA (#51148) 2021-01-27 09:11:37 -08:00
test_jit.py [WIP] Add a _flush_compilation_cache for testing (#52001) 2021-02-16 10:49:38 -08: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.py Clean up some type annotations in caffe2/test (#49943) 2021-01-13 10:01:55 -08:00
test_jit_fuser_legacy.py
test_jit_fuser_te.py [TensorExpr] Use wider type for scalars (#50774) 2021-01-20 15:12:27 -08:00
test_jit_legacy.py
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_kernel_launch_checks.py
test_license.py concantenate LICENSE files when building a wheel (#51634) 2021-02-08 08:28:46 -08:00
test_linalg.py [ROCm] skip one more magma test that is flaky (#52064) 2021-02-11 14:02:52 -08:00
test_logging.py
test_metal.py
test_mkldnn.py BFloat16: enable prepacked weights's inference (#48922) 2021-02-17 11:20:00 -08:00
test_mobile_optimizer.py cleaned up ModuleAttributeError (#50298) 2021-01-14 06:58:01 -08:00
test_multiprocessing.py [BE] unified run_process_no_exception code (#49774) 2021-01-04 13:43:09 -08:00
test_multiprocessing_spawn.py
test_namedtensor.py
test_namedtuple_return_api.py make forward AD API private (#51693) 2021-02-04 19:02:29 -08:00
test_native_functions.py
test_nn.py Skip OneDNN Convolution in case of groups = 24 #50042 (#52327) 2021-02-17 14:49:23 -08:00
test_numba_integration.py Drop unused imports from test (#49973) 2021-01-07 12:09:38 -08: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
test_ops.py OpInfo: Added clamp and trunc tests with aliases (#51167) 2021-02-10 05:36:18 -08:00
test_optim.py Replace CUDA 11.1 Linux CI with CUDA 11.2 (#51905) 2021-02-10 11:43:50 -08:00
test_overrides.py Fix pickling for Tensor subclasses (redo) (#47732) 2021-02-01 07:32:52 -08:00
test_package.py [package] patch inspect.getfile to work with PackageImporter (#51568) 2021-02-02 11:29:29 -08:00
test_profiler.py Add FLOPS support to the new profiler API. (#51734) 2021-02-05 15:03:35 -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_pytree.py
test_quantization.py reland - ns for fx - stubs of the three APIs (compare weights, activations, activations with shadow) (#52302) 2021-02-16 19:59:32 -08:00
test_reductions.py Clean up usage of torch._six partially (#49785) 2021-02-08 13:58:34 -08: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 Removes nonzero method warning (#51618) 2021-02-04 17:43:43 -08: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 Revert "Stable sort for CPU (#50052)" (#50752) 2021-01-19 18:21:25 -08:00
test_sparse.py Warn about floor_divide performing incorrect rounding (#50281) (#50281) 2021-02-10 03:13:34 -08:00
test_spectral_ops.py [ROCm] enable fft tests (#51581) 2021-02-17 13:27:55 -08:00
test_static_runtime.py Clean up some type annotations in caffe2/test (#49943) 2021-01-13 10:01:55 -08:00
test_tensor_creation_ops.py [testing] Port torch.{repeat, tile} tests to use OpInfo machinery (#50199) 2021-01-19 06:02:27 -08:00
test_tensorboard.py
test_tensorexpr.py [BE] replace unittest.main with run_tests (#50451) 2021-01-13 10:33:08 -08:00
test_tensorexpr_pybind.py Make te a hidden package for now (#51690) 2021-02-04 07:58:38 -08:00
test_testing.py [reland] Early terminate CUDA on common_utils TestCases (#52126) 2021-02-12 07:32:42 -08: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 Fix test to use proper condition. (#52216) 2021-02-12 12:59:35 -08:00
test_type_hints.py Mypy fixes for pytorch master (#52090) 2021-02-17 10:39:51 -08:00
test_type_info.py
test_type_promotion.py
test_unary_ufuncs.py Fix complex acos edge cases (#52287) 2021-02-17 11:36:09 -08:00
test_utils.py Re-enable test_standalone_load for Windows 11.1 (#51596) 2021-02-03 08:58:34 -08:00
test_view_ops.py Add tensor.view(dtype) (#47951) 2021-01-08 06:55:21 -08:00
test_vmap.py Add private mechanism to toggle vmap fallback warnings (#51218) 2021-01-28 13:05:00 -08:00
test_vulkan.py
test_xnnpack_integration.py