Edward Z. Yang b7215de32f prod ref ... It turns out the prim is implemented incorrectly as torch.prod does not accept a dim list, so I added a little stub for this. Signed-off-by: Edward Z. Yang <ezyangfb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/78461 Approved by: https://github.com/ngimel		2022-05-31 14:18:49 +00:00
..
codegen	Clean up profiling mode and profiling executor strategy (#73875)	2022-03-29 18:38:51 +00:00
data
distributed	Revert RPC Meta device support	2022-05-19 23:47:47 +00:00
generated
test_module
__init__.py
autocast_test_lists.py	Add logsumexp to AMP autocast (#76330)	2022-05-27 17:26:20 +00:00
check_kernel_launches.py	move kernel launch checks from torch.testing to torch.testing._internal.check_kernel_launches (#60862)	2021-09-29 00:39:22 -07:00
common_cuda.py	Add torch.linalg.ldl_factor_ex and torch.linalg.ldl_solve	2022-04-28 19:23:37 +00:00
common_device_type.py	Use any_type in test_out	2022-05-18 16:24:32 +00:00
common_distributed.py	Add testing workaround for EFA and TensorPipe (#77363)	2022-05-18 22:54:15 +00:00
common_dtype.py	[complex32] real and imag (also remove unused real and imag kernels)	2022-05-05 04:36:58 +00:00
common_fsdp.py	Update linalg.*norm	2022-05-18 11:46:50 +00:00
common_fx2trt.py	[fx2trt] support ops for hf_T5 (#62)	2022-04-27 00:38:11 +00:00
common_jit.py	Extends OpInfo architecture with reference inputs, adds them for elementwise binary operators	2022-03-21 03:24:16 +00:00
common_methods_invocations.py	prod ref	2022-05-31 14:18:49 +00:00
common_modules.py	[quant][refactor] Remove the base class from __all__	2022-05-20 17:56:22 +00:00
common_nn.py	Implement Tanh Gelu Approximation (#61439)	2022-02-14 03:40:32 +00:00
common_quantization.py	[quant][fx][equalization] Fix example_inputs follow ups in test_equalize_fx	2022-05-26 01:42:24 +00:00
common_quantized.py	Add onednn quant backend (#74137)	2022-03-15 01:28:21 +00:00
common_subclass.py	Support for tensor subclasses as parameters	2022-04-27 19:28:55 +00:00
common_utils.py	Fix sparse BSR tensor validation.	2022-05-27 13:26:35 +00:00
composite_compliance.py	[Reland] Change 'python mode' to 'torch dispatch mode'	2022-05-02 20:06:43 +00:00
dist_utils.py	Use higher timeout for TSAN tests. (#65391)	2021-09-21 12:08:27 -07:00
hypothesis_utils.py
jit_metaprogramming_utils.py	[JIT] fix opinfo utils to handle tensor kwargs	2022-05-10 03:30:07 +00:00
jit_utils.py	[NVFuser] prep for on-by-default	2022-05-06 18:18:53 +00:00
logging_tensor.py	Modernize LoggingTensorMode	2022-05-24 22:41:49 +00:00
opinfo_helper.py	[opinfo] use dtypes instead of dtypesIfCPU (#68732)	2021-11-27 16:07:51 -08:00
quantization_torch_package_models.py	fx quant: add a BC test for loading old torch.package models (#65538)	2021-10-11 08:23:38 -07:00