saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test
History
Erjia Guan 3d218e1c87 Raise warning for unpickable local function (#547) (#80232) 
Summary:
X-link: https://github.com/pytorch/data/pull/547

Fixes https://github.com/pytorch/data/issues/538
- Improve the validation function to raise warning about unpickable function when either lambda or local function is provided to DataPipe.
- The inner function from functools.partial object is extracted as well for validation
- Mimic the behavior of pickle module for local lambda function: It would only raise Error for the local function rather than lambda function. So, we will raise warning about local function not lambda function.
```py

>>> import pickle
>>> def fn():
...     lf = lambda x: x
...     pickle.dumps(lf)
>>> pickle.dumps(fn)
AttributeError: Can't pickle local object 'fn.<locals>.<lambda>'
```

This Diff also fixes the Error introduced by https://github.com/pytorch/pytorch/pull/79344

Test Plan:
CI on PyTorch and TorchData
Manually validated the tests from TorchVision

Differential Revision: D37417556

Pull Request resolved: https://github.com/pytorch/pytorch/pull/80232
Approved by: https://github.com/NivekT
2022-06-27 21:47:09 +00:00
..
ao/sparsity [ao][sparsity] Implemented state_dict() and load_state_dict() functions (#79883) 2022-06-24 16:55:06 +00:00
autograd
benchmark_utils
bottleneck_test
cpp Fix //:module_test Conversion_MultiCUDA (#79926) 2022-06-21 23:32:18 +00:00
cpp_api_parity
cpp_extensions Allow relocatable device code linking in pytorch CUDA extensions (#78225) 2022-06-02 21:35:56 +00:00
custom_backend
custom_operator set data permits requires_grad=True on integer tensor 2022-06-01 15:56:32 +00:00
distributed [Checkpoint Wrapper] Fix assert (#80283) 2022-06-27 15:19:39 +00:00
distributions Setting validation flag for Distributions tests to work with TorchDynamo (#80081) 2022-06-24 04:51:51 +00:00
error_messages
expect Validate Sparse Compressed tensor inputs (#79385) 2022-06-27 17:19:54 +00:00
forward_backward_compatibility Make linalg_det function structured (#79486) 2022-06-23 23:24:44 +00:00
fx Back out "Back out "[const_fold] Set requires_grad based on the folded tensor; add device_for_folding option"" (#79696) 2022-06-18 19:13:49 +00:00
jit Use generators with all/any in torch/optim (#78142) 2022-06-24 17:23:45 +00:00
jit_hooks
lazy functionalization <> LTC integration (take 3) (#80251) 2022-06-26 23:10:21 +00:00
mobile Fix backward of binary_cross_entropy_with_logits 2022-06-23 01:31:08 +00:00
onnx [ONNX] Move no runtime tests out of runtime test cases (#80078) 2022-06-26 18:46:25 +00:00
package [codemod][usort] apply import merging for fbcode (1 of 11) (#78973) 2022-06-06 23:44:28 +00:00
quantization [quant] Modify APoT nonuniform quantization workflow (#80075) 2022-06-27 14:54:06 +00:00
scripts
test_img
typing
allowlist_for_publicAPI.json Add __all__ to torch.nn.quantized, fx.passes, ao.nn and amp submodules (#80376) 2022-06-27 21:36:27 +00:00
create_dummy_torchscript_model.py
defs.bzl Add all bzl files per D36874458 2022-06-06 09:40:19 -07:00
delete.py
HowToWriteTestsUsingFileCheck.md
linear.py
load_torchscript_model.py
run_test.py Enable test: test/distributed/algorithms/ddp_comm_hooks/test_ddp_hooks.py 2022-06-23 00:11:53 +00:00
simulate_nccl_errors.py
test_ao_sparsity.py [ao][sparsity] Base scheduler class for Data Schedulers (#79817) 2022-06-24 16:51:52 +00:00
test_autocast.py
test_autograd.py Fix forward AD copy_ into same-sized tensor without fw grad 2022-06-17 18:55:32 +00:00
test_binary_ufuncs.py Don't check overflow for scalar arg in comparison ops (#78881) 2022-06-05 01:43:51 +00:00
test_bundled_images.py
test_bundled_inputs.py
test_complex.py
test_cpp_api_parity.py
test_cpp_extensions_aot.py Allow relocatable device code linking in pytorch CUDA extensions (#78225) 2022-06-02 21:35:56 +00:00
test_cpp_extensions_jit.py
test_cuda.py increase sleep for TestCuda.test_caching_pinned_memory_multi_gpu (#76601) 2022-06-14 21:10:35 +00:00
test_cuda_primary_ctx.py
test_dataloader.py Support percentages in random_split (#78877) 2022-06-16 02:00:25 +00:00
test_datapipe.py Raise warning for unpickable local function (#547) (#80232) 2022-06-27 21:47:09 +00:00
test_decomp.py Reference implementations for rsqrt and native_layer_norm (#79413) 2022-06-17 07:24:02 +00:00
test_deploy.py
test_determination.py
test_dispatch.py Make string serialization of C++ FunctionSchema consistent with torchgen.model.FunctionSchema 2022-05-24 19:39:26 +00:00
test_dynamic_shapes.py Revert "parse pysymints to IValues (#80066)" 2022-06-24 20:15:09 +00:00
test_expanded_weights.py
test_fake_tensor.py [FakeTensor] Use the device of the meta tensor for fallback kernel (#80193) 2022-06-24 20:00:07 +00:00
test_foreach.py
test_function_schema.py Expose is_out to python 2022-06-01 07:39:24 +00:00
test_functional_autograd_benchmark.py
test_functional_optim.py
test_functionalization.py add a new alias key for functional to view op decompositions 2022-06-15 23:18:09 +00:00
test_futures.py
test_fx.py Add Dropout1d module 2022-06-15 14:39:07 +00:00
test_fx_experimental.py Ported proxy tensor tests over to core (#78890) 2022-06-07 00:28:53 +00:00
test_fx_passes.py FX graph partitioner and fuser (#79439) 2022-06-24 18:49:37 +00:00
test_hub.py
test_import_stats.py Revert "Revert "[ci] remove remaining RDS dependency"" 2022-06-15 10:04:13 -07:00
test_indexing.py fix set item to scalar tensor missing gradient info 2022-05-25 21:22:44 +00:00
test_jit.py Revert "Enable dim=None for torch.sum (#75845)" 2022-06-16 22:01:41 +00:00
test_jit_autocast.py Conv BN folding data type issue when conv has no bias (#78241) 2022-05-26 18:42:17 +00:00
test_jit_cuda_fuser.py Add OpInfo for torch.equal and fix support for non-standard bools 2022-06-20 23:48:39 +00:00
test_jit_disabled.py
test_jit_fuser.py
test_jit_fuser_legacy.py
test_jit_fuser_te.py Add OpInfo for torch.equal and fix support for non-standard bools 2022-06-20 23:48:39 +00:00
test_jit_legacy.py Remove mentions of non-existing test_jit_py3 (#78977) 2022-06-07 02:28:45 +00:00
test_jit_llga_fuser.py
test_jit_profiling.py Remove mentions of non-existing test_jit_py3 (#78977) 2022-06-07 02:28:45 +00:00
test_jit_simple.py Remove mentions of non-existing test_jit_py3 (#78977) 2022-06-07 02:28:45 +00:00
test_jit_string.py
test_jiterator.py Python Jiterator supports multiple outputs (#78139) 2022-05-24 21:52:56 +00:00
test_kernel_launch_checks.py
test_license.py
test_linalg.py Port cholesky to structured kernels (#79300) 2022-06-24 02:37:45 +00:00
test_logging.py
test_masked.py
test_meta.py Add Div reference (#77936) 2022-06-27 14:46:17 +00:00
test_metal.py
test_mkldnn.py
test_mobile_optimizer.py
test_model_dump.py
test_module_init.py [quant] Add QuantizedMHA class (#79956) 2022-06-24 16:54:42 +00:00
test_modules.py Support both train / eval modes for ModuleInfo 2022-06-09 20:57:17 +00:00
test_monitor.py
test_mps.py [MPS] Add equal operator (#80195) 2022-06-25 12:40:52 +00:00
test_multiprocessing.py
test_multiprocessing_spawn.py
test_namedtensor.py Revert "Enable dim=None for torch.sum (#75845)" 2022-06-16 22:01:41 +00:00
test_namedtuple_return_api.py Make linalg_det function structured (#79486) 2022-06-23 23:24:44 +00:00
test_native_functions.py
test_native_mha.py [PyTorch] Round T up to next multiple of 8 in NestedTensor case 2022-05-25 20:34:19 +00:00
test_nestedtensor.py TensorImpl:::size_custom to support NestedTensor.size (#80236) 2022-06-27 17:07:42 +00:00
test_nn.py BUG: Evade segfault by throwing a RuntimeError for nn.ChannelShuffle and empty input tensors (#77029) 2022-06-23 21:14:02 +00:00
test_nnapi.py
test_numba_integration.py
test_numpy_interop.py ENH: Add a force argument to numpy() (#78564) 2022-06-06 14:14:17 +00:00
test_openmp.py
test_ops.py Add Div reference (#77936) 2022-06-27 14:46:17 +00:00
test_ops_gradients.py
test_ops_jit.py Minor fix in jit tests to pass TorchDynamo (#79903) 2022-06-21 00:42:02 +00:00
test_optim.py Fix SequentialLR initialization (#72856) 2022-06-21 20:21:13 +00:00
test_overrides.py add utils for checking that all modes are in the same scope and finding the outermost mode 2022-06-10 19:31:05 +00:00
test_package.py
test_per_overload_api.py
test_prims.py small cleanup of executor (#79973) 2022-06-22 00:35:51 +00:00
test_profiler.py [Profiler] Add queue depth computation (#79993) 2022-06-24 20:47:58 +00:00
test_profiler_tree.py [Profiler][Trivial] Update test_profiler_tree 2022-06-22 22:30:33 +00:00
test_proxy_tensor.py switched over to using faketensor in proxytensor (#79634) 2022-06-27 19:55:47 +00:00
test_pruning_op.py
test_public_bindings.py Wconstab/reland pysymint (#79795) 2022-06-20 22:55:06 +00:00
test_python_dispatch.py disable modes during constructor 2022-06-17 22:28:27 +00:00
test_pytree.py
test_quantization.py [ao] Adds framework for InputWeightEqualization Detector (#79916) 2022-06-24 14:51:15 +00:00
test_reductions.py Add complex_to_float option in ReductionOpInfo 2022-06-01 09:31:45 +00:00
test_scatter_gather_ops.py fix gather sparse_grad backward crash with empty index tensor 2022-06-04 02:25:05 +00:00
test_segment_reductions.py reland Add offsets-based reduction to segment_reduce (CPU, CUDA) 2022-06-17 15:49:31 +00:00
test_serialization.py
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py
test_show_pickle.py
test_sort_and_select.py MAINT: Harmonize argsort params with array_api (#75162) 2022-06-09 12:32:01 +00:00
test_sparse.py permute for COO sparse tensors (#79707) 2022-06-25 08:49:58 +00:00
test_sparse_csr.py Add support for BSR <-> Strided Conversion (#80354) 2022-06-27 21:09:09 +00:00
test_spectral_ops.py
test_stateless.py
test_static_runtime.py [Static Runtime] Added nested prim::fork and aten::wait test case (#79746) 2022-06-20 16:32:17 +00:00
test_subclass.py
test_tensor_creation_ops.py
test_tensorboard.py [tensorboard] update assertion error for scalar() and fix docs (#76859) 2022-06-16 05:24:20 +00:00
test_tensorexpr.py [NNC] channels last propagation within NNC fusion group (#76948) 2022-05-30 18:31:49 +00:00
test_tensorexpr_pybind.py
test_testing.py Validate Sparse Compressed tensor inputs (#79385) 2022-06-27 17:19:54 +00:00
test_throughput_benchmark.py
test_torch.py Only sync CUDA if the operation is run on GPU (#80328) 2022-06-27 14:49:39 +00:00
test_transformers.py Add test for torchscripting nn.TransformerEncoder, including fast path (#79796) 2022-06-17 22:00:49 +00:00
test_type_hints.py
test_type_info.py
test_type_promotion.py
test_typing.py
test_unary_ufuncs.py update signbit docs and add -0. to reference testing for unary and binary functions. (#78349) 2022-06-06 13:48:08 +00:00
test_utils.py Revert "To add hipify_torch as a submodule in pytorch/third_party (#74704)" 2022-06-21 23:54:00 +00:00
test_view_ops.py functionalization <> LTC integration (take 3) (#80251) 2022-06-26 23:10:21 +00:00
test_vmap.py
test_vulkan.py
test_xnnpack_integration.py