summerdo 7db6eb7156 [test_nn] add custom device support for dropout tests、lazy_modules te… (#106609) ... add custom device support for dropout tests、lazy_modules tests and multihead_attention tests. Pull Request resolved: https://github.com/pytorch/pytorch/pull/106609 Approved by: https://github.com/mikaylagawarecki		2023-08-11 09:14:34 +00:00
..
codegen
data
distributed
generated
opinfo	Remove dynamo+nvfuser (#105789)	2023-08-08 22:29:32 +00:00
optests	Improve AOTAutograd tests to do something when inputs don't require grad (#106558)	2023-08-07 00:11:30 +00:00
test_module
__init__.py
autocast_test_lists.py
autograd_function_db.py
check_kernel_launches.py
common_cuda.py	unify TEST_CUDA (#106685)	2023-08-10 09:01:36 +00:00
common_device_type.py	[test_nn] add custom device support for dropout tests、lazy_modules te… (#106609)	2023-08-11 09:14:34 +00:00
common_dist_composable.py
common_distributed.py
common_dtype.py
common_fsdp.py
common_jit.py
common_methods_invocations.py	[inductor] modify index_reduce to pass opinfo tests (#106429)	2023-08-10 18:14:00 +00:00
common_modules.py	Reland "Add nn.CircularPad{*}d for consistency + fix no_batch_dim support (#106148)" (#106632)	2023-08-07 20:10:25 +00:00
common_nn.py	Remove set_default_dtype from nn tests (#105775)	2023-08-10 14:56:13 +00:00
common_pruning.py
common_quantization.py	[pytorch][ao] Add torch.matmul in FloatFunctional/QFunctional (#106831)	2023-08-10 22:43:36 +00:00
common_quantized.py
common_subclass.py
common_utils.py	[test_nn] add custom device support for dropout tests、lazy_modules te… (#106609)	2023-08-11 09:14:34 +00:00
composite_compliance.py
control_flow_opinfo_db.py
custom_op_db.py
dist_utils.py
hypothesis_utils.py	Remove set_default_dtype from nn tests (#105775)	2023-08-10 14:56:13 +00:00
inductor_utils.py
jit_metaprogramming_utils.py
jit_utils.py
logging_tensor.py
logging_utils.py	make sure log tests are run in non-verbose mode (#106496)	2023-08-03 02:45:35 +00:00
quantization_torch_package_models.py