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
Supriya Rao 7cec4b3d4a [quant][fx] add _remove_qconfig flag to convert_fx (#53166) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53166

Context: For fx modules that consist of scriptmodules, calling
delattr(module, 'qconfig') throws an attribute error. will follow up
with a separate issue/repro to fix this problem

This PR adds a temporary flag to convert_fx API to preserve the qconfig attributes on the converted model
We will remove this flag once we reach a conclusion on calling delattr on scriptmodules

Test Plan:
python test/test_quantization.py test_preserve_qconfig

Imported from OSS

Reviewed By: jerryzh168

Differential Revision: D26771518

fbshipit-source-id: 9fd72816576856ffb4aa11f8fde08303d1df10a2
2021-03-03 12:58:05 -08:00
..
backward_compatibility Make meta a device (getting rid of empty_meta) (#53143) 2021-03-03 11:24:13 -08:00
benchmark_utils
bottleneck_test
cpp [nnc] Test cases for uneven split + reorder (#53091) 2021-03-02 20:36:48 -08:00
cpp_api_parity Huber loss (#50553) 2021-03-02 17:30:45 -08:00
cpp_extensions Register DefaultBackend implementations for functional/inplace structured operators (#53037) 2021-03-02 14:13:08 -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 Deduplicate shared params before constructing Reducer in DDP (#51929) 2021-03-03 10:13:24 -08:00
distributions Add sample validation for LKJCholesky.log_prob (#52763) 2021-02-25 16:12:29 -08:00
error_messages
expect Trace linear as aten::linear (#51897) 2021-02-19 10:20:42 -08:00
fx Extend subgraph_rewriter logic (#51532) 2021-02-03 18:14:37 -08:00
jit Add default arguments to cuda stream and events (#53025) 2021-03-02 14:37:24 -08: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
namespace_b
onnx [ONNX] Update LayerNorm symbolic to handle autocasting (#52199) (#52350) 2021-02-19 10:57:15 -08:00
optim
package_a [fx/package] make GraphModules packageable (#51976) 2021-02-23 22:43:00 -08:00
quantization [quant][fx] add _remove_qconfig flag to convert_fx (#53166) 2021-03-03 12:58:05 -08:00
scripts
test_img
type_hint_tests
HowToWriteTestsUsingFileCheck.md
linear.py Move conv to mkldnn (#51483) 2021-03-01 21:19:27 -08:00
module_a.py
run_test.py enable coverage test for dataloader on Windows (#52550) 2021-03-02 18:40:47 -08:00
simulate_nccl_errors.py
test_autograd.py Codegen python bindings to access attributes of grad_fn (#52451) 2021-03-02 15:20:56 -08:00
test_binary_ufuncs.py remove pointless test (#52609) 2021-02-22 16:25:04 -08: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 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 [ROCm] Enable test cases in test_cuda.py for ROCm (#52739) 2021-03-01 12:54:40 -08:00
test_cuda_primary_ctx.py
test_dataloader.py enable coverage test for dataloader on Windows (#52550) 2021-03-02 18:40:47 -08:00
test_datapipe.py Add more datapipe to functional API (#53123) 2021-03-03 07:01:00 -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
test_foreach.py Probable fix for out of place BinaryOpScalar bad values and/or IMAs on 11.2 (ci-all edition) (#52634) 2021-03-01 12:41:24 -08:00
test_function_schema.py
test_functional_autograd_benchmark.py
test_futures.py [JIT/Futures] support set_exception api (#50983) 2021-02-04 20:22:19 -08:00
test_fx.py [WIP][FX] Optionally record stack traces when symtracing (#53081) 2021-03-03 12:30:43 -08:00
test_fx_experimental.py [fx] Separate globals assignment from code generation (#51974) 2021-02-23 13:48: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 Add support to compare devices (#53045) 2021-03-01 21:04:43 -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
test_jit_fuser.py
test_jit_fuser_legacy.py
test_jit_fuser_te.py add a full pipeline test for a TypeCheck (#52933) 2021-03-01 10:58:08 -08:00
test_jit_legacy.py
test_jit_profiling.py
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
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 Revert D26727918: [pytorch][PR] Added CUDA support for torch.orgqr 2021-03-03 10:06:49 -08:00
test_logging.py
test_metal.py
test_mkldnn.py Dont error on 0-dim in convolution (#51922) 2021-03-01 21:22:30 -08:00
test_mobile_optimizer.py [Reland] Update XNNPACK (#52691) 2021-02-24 06:40:38 -08: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 Implements torch.linalg.lstsq (#49093) 2021-03-02 19:00:07 -08: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 Huber loss (#50553) 2021-03-02 17:30:45 -08:00
test_numba_integration.py
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 Revert D26727660: [pytorch][PR] Add OpInfo for bitwise_not and make ROCM and CUDA OpInfo tests consistent 2021-03-03 06:08:48 -08:00
test_optim.py [optim] bugfix when all parameters have no grad (#52944) 2021-03-03 11:56:09 -08:00
test_overrides.py Fix pickling for Tensor subclasses (redo) (#47732) 2021-02-01 07:32:52 -08:00
test_package.py [fx/package] make GraphModules packageable (#51976) 2021-02-23 22:43:00 -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_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 Enable min & max for Float16 & BFloat16 (#51244) 2021-02-18 23:13:51 -08: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 Removes nonzero method warning (#51618) 2021-02-04 17:43:43 -08: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 [BE] _get_torch_cuda_version should return tuple (#52409) 2021-02-18 09:28:38 -08:00
test_spectral_ops.py [ROCm] enable fft tests (#51581) 2021-02-17 13:27:55 -08:00
test_static_runtime.py
test_tensor_creation_ops.py
test_tensorboard.py
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 Store test file in S3 as well for every TestSuite (#52869) 2021-03-02 07:36:00 -08:00
test_throughput_benchmark.py
test_torch.py Make meta a device (getting rid of empty_meta) (#53143) 2021-03-03 11:24:13 -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 Revert D26727660: [pytorch][PR] Add OpInfo for bitwise_not and make ROCM and CUDA OpInfo tests consistent 2021-03-03 06:08:48 -08:00
test_utils.py Store test file in S3 as well for every TestSuite (#52869) 2021-03-02 07:36:00 -08:00
test_view_ops.py
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