pytorch/test/jit
Antonio Kim 7fc292930c Add support for torch.Generator type in TorchScript (#110413)
- Add support for `torch.Generator` type in TorchScript
- Add `generator` args to all `torch.nn.init` functions that call `uniform_` or `normal_`
- Add support for `torch.Generator` in LTC's TorchScript backend (CC: @wconstab)

CC: @eellison @davidberard98 @GlebKazantaev @behzad-a
Pull Request resolved: https://github.com/pytorch/pytorch/pull/110413
Approved by: https://github.com/wconstab, https://github.com/albanD, https://github.com/glebk-cerebras, https://github.com/davidberard98
2023-11-21 23:07:21 +00:00
..
_imported_class_test
fixtures
fixtures_srcs [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
xnnpack
__init__.py
myexception.py
test_alias_analysis.py
test_async.py
test_aten_pow.py
test_attr.py
test_autodiff_subgraph_slicing.py
test_autodiff.py
test_await.py
test_backend_nnapi.py [jit] Skip NNAPI, test_ivalue, CPU NNC tests in fbcode (#108937) 2023-09-11 22:42:30 +00:00
test_backends.py
test_batch_mm.py
test_builtins.py
test_class_type.py
test_complex.py
test_complexity.py [jit] test_complexity.py - don't set default dtype in global scope (#106486) 2023-08-03 02:50:15 +00:00
test_convert_activation.py
test_cuda.py unify TEST_CUDA (#106685) 2023-08-10 09:01:36 +00:00
test_custom_operators.py Raise AttributeError in _OpsNamespace if __self__ attribute is requested (#104096) 2023-06-27 01:42:06 +00:00
test_data_parallel.py
test_dataclasses.py
test_dce.py
test_device_analysis.py
test_dtype_analysis.py [BE]: Apply ruff PERF fixes to torch (#104917) 2023-07-11 20:45:21 +00:00
test_enum.py [JIT] Fix typed enum handling in 3.11 (#109717) 2023-09-20 22:09:41 +00:00
test_exception.py
test_freezing.py Enable flake8-bugbear B020 lint (#110823) 2023-10-24 22:43:47 +00:00
test_functional_blocks.py
test_fuser_common.py
test_generator.py Add support for torch.Generator type in TorchScript (#110413) 2023-11-21 23:07:21 +00:00
test_graph_rewrite_passes.py
test_hash.py
test_hooks_modules.py
test_hooks.py
test_ignorable_args.py
test_ignore_context_manager.py
test_isinstance.py
test_jit_utils.py
test_list_dict.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
test_logging.py
test_misc.py torch.ops.aten.*: sort aten ops before jit overloads (#107138) 2023-08-17 03:05:59 +00:00
test_models.py Revert "Remove deprecated fbgemm operators (#104535)" 2023-10-25 16:34:16 +00:00
test_module_apis.py
test_module_containers.py [JIT] clarify errors due to non-literal indexing into ModuleList, ModuleDict (#98606) 2023-04-18 02:53:53 +00:00
test_module_interface.py
test_modules.py
test_op_decompositions.py
test_optimize_for_mobile_preserve_debug_info.py
test_parametrization.py
test_pdt.py
test_peephole.py
test_profiler.py
test_python_bindings.py
test_python_builtins.py
test_python_ir.py
test_recursive_script.py torch.jit.script escape hatch (#106229) 2023-08-11 18:24:46 +00:00
test_remove_mutation.py Remove set_default_dtype calls from jit and ops tests (#105072) 2023-07-15 03:18:33 +00:00
test_save_load_for_op_version.py
test_save_load.py Back out "Reland "Make adding buffers more like adding parameters (#104069)" (#106224)" (#106743) 2023-08-08 15:27:34 +00:00
test_script_profile.py
test_scriptmod_ann.py
test_slice.py
test_sparse.py
test_string_formatting.py
test_symbolic_shape_analysis.py Fix shape function for transpose convolution (#102139) 2023-06-21 17:50:56 +00:00
test_tensor_creation_ops.py
test_tensor_methods.py
test_torchbind.py
test_tracer.py update type() calling to not use unneeded device (#110163) 2023-09-28 17:34:46 +00:00
test_type_sharing.py
test_types.py
test_typing.py [jit] Fix inspect.get_annotations usage in python >= 3.10 (#104485) 2023-07-06 00:37:47 +00:00
test_union_pep604.py Python 3.10 Union operator | support for JIT (#109293) 2023-09-25 15:35:54 +00:00
test_union.py
test_unsupported_ops.py
test_upgraders.py
test_warn.py
test_with.py [BE]: enable PLE error codes in ruff and fix bugs (#101079) 2023-05-11 23:57:25 +00:00