pytorch/torch/testing/_internal
Lillian Johnson 3b6f30824c OpInfo JIT op.output_func handling support (#50775)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50775

Test Plan: Imported from OSS

Reviewed By: mruberry

Differential Revision: D25964541

Pulled By: Lilyjjo

fbshipit-source-id: 8cf1ee9191d526cc46ae283f38c2d64bd60afdb2
2021-01-27 15:04:23 -08:00
..
codegen Enable CUDA Fuser for ROCm (#45965) 2020-10-08 10:41:56 -07:00
data remediation of S205607 2020-07-17 17:19:47 -07:00
distributed Support device map for distributed autograd while using TensorPipe. (#44859) 2021-01-27 13:01:44 -08:00
generated remediation of S205607 2020-07-17 17:19:47 -07:00
test_module Drop unused imports (#49972) 2021-01-13 12:26:17 -08:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
autocast_test_lists.py [PyTorch] Enable type check for autocast_test_lists (#45107) 2020-09-22 16:54:26 -07:00
common_cuda.py Enable BFloat support for gemms on arch other than ampere (#50442) 2021-01-26 11:07:07 -08:00
common_device_type.py OpInfo: Add DecorateInfo class similar to SkipInfo for decorators (#50501) 2021-01-25 04:51:04 -08:00
common_distributed.py Let TensorPipe detect peer access (#50676) 2021-01-20 08:04:51 -08:00
common_jit.py OpInfo JIT op.output_func handling support (#50775) 2021-01-27 15:04:23 -08:00
common_methods_invocations.py OpInfo JIT op.output_func handling support (#50775) 2021-01-27 15:04:23 -08:00
common_nn.py test, fix deepcopy of tensor with grad (#50663) 2021-01-26 16:19:53 -08:00
common_quantization.py quantization: Linear + BatchNorm1d fusion (#50748) 2021-01-20 12:59:02 -08:00
common_quantized.py [quant] Quantizable LSTM (#49671) 2020-12-30 15:21:38 -08:00
common_utils.py Rename set_deterministic to use_deterministic_algorithms (#49904) 2021-01-22 11:27:07 -08:00
dist_utils.py [RPC profiling] Extend RPC profiling to support async function execution over RPC. (#44664) 2020-09-25 13:19:26 -07:00
expecttest.py Fix test_dispatch.py when running with TORCH_SHOW_CPP_STACKTRACES=1 (#50509) 2021-01-20 10:15:01 -08:00
hypothesis_utils.py [quant] conv_transpose1d_prepack / conv_transpose1d_unpack (#40360) 2020-09-11 04:12:28 -07:00
jit_metaprogramming_utils.py OpInfo JIT op.output_func handling support (#50775) 2021-01-27 15:04:23 -08:00
jit_utils.py Workaround Python3.9 limitations in test_jit_py3 (#51088) 2021-01-26 12:49:35 -08:00
mypy_wrapper.py [reland] Refactor mypy configs list into editor-friendly wrapper (#50826) 2021-01-26 09:04:14 -08:00
te_utils.py Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00