tree: ba6f1b02f3454f4124c182c7f4b2b377da934cf1 [path history] [tgz]
  1. ActivationCheckpointingTests.test_cond_with_kwargs
  2. AotAutogradFallbackTests.test_aot_sequence_nr
  3. AutogradFunctionTests.test_graph_break_if_lifted_free_variable
  4. AutogradFunctionTests.test_print_in_bwd
  5. AutogradFunctionTests.test_stride_in_bwd
  6. ComptimeTests.test_graph_break
  7. CtxManagerTests.test_autograd_profiler_enabled
  8. CtxManagerTests.test_disable_saved_tensors_hooks
  9. CtxManagerTests.test_disable_saved_tensors_hooks_prev_disabled
  10. CtxManagerTests.test_disable_saved_tensors_hooks_prev_disabled_nested
  11. DecoratorTests.test_mark_static_address_guarded
  12. DecoratorTests.test_mark_static_address_unguarded
  13. DefaultsTests.test_cast_tensor_single_elem
  14. DefaultsTests.test_compare_constant_and_tensor
  15. DefaultsTests.test_func_default_torch_args
  16. DefaultsTests.test_in_set_inplace
  17. DefaultsTests.test_in_set_would_fail_broadcast
  18. DynamoProfilerTests.test_dynamo_timed_profiling_isolated
  19. DynamoProfilerTests.test_profiler_cache_lookup
  20. DynamoProfilerTests.test_profiler_dynamo_compiled_region
  21. End2EndTests.test_init_group
  22. ExampleTests.test_exportdb_supported_case_assume_constant_result
  23. ExampleTests.test_exportdb_supported_case_constrain_as_size_example
  24. ExcTests.test_backend_suppress_line
  25. ExcTests.test_graph_break_log
  26. ExcTests.test_internal_error_no_suppress
  27. ExcTests.test_internal_error_suppress_errors
  28. ExcTests.test_not_implemented_error
  29. ExcTests.test_trigger_bisect_on_error
  30. ExcTests.test_trigger_on_error
  31. ExportTests.test_cond_raise_user_error_on_non_list_operands
  32. ExportTests.test_cond_raise_user_error_on_non_tensor_operands
  33. ExportTests.test_cond_raise_user_error_on_unsupported_pred
  34. ExportTests.test_export_decomp
  35. ExportTests.test_export_dynamic_dim_cleanup
  36. ExportTests.test_export_dynamic_dim_range_constraint
  37. ExportTests.test_export_mark_dynamic_conflict_dynamic_dim
  38. ExportTests.test_export_multi_dynamic_dim_constraint
  39. ExportTests.test_export_multi_dynamic_dim_unsafe_relationship
  40. ExportTests.test_export_no_raise
  41. ExportTests.test_export_preserve_constraints_as_metadata_scalar
  42. ExportTests.test_export_raise_on_relationship
  43. ExportTests.test_export_with_constant_dict_values
  44. ExportTests.test_export_with_constant_free_function
  45. ExportTests.test_export_with_constant_free_function_and_class_method
  46. ExportTests.test_export_with_constant_free_function_and_class_method_multiarg
  47. ExportTests.test_export_with_constant_free_function_and_class_method_multiarg_diff
  48. ExportTests.test_export_with_constant_list_nonzero
  49. ExportTests.test_export_with_constant_list_nonzero_free_function
  50. ExportTests.test_export_with_constant_method_on_module
  51. ExportTests.test_export_with_constant_method_on_module_invoke_twice
  52. ExportTests.test_export_with_constant_none_control_flow
  53. ExportTests.test_export_with_constant_none_control_flow_free_func
  54. ExportTests.test_export_with_constant_not_none_control_flow
  55. ExportTests.test_export_with_constant_not_none_control_flow_free_func
  56. ExportTests.test_export_with_constant_not_none_control_flow_pos
  57. ExportTests.test_export_with_constant_not_return_const
  58. ExportTests.test_export_with_constant_tuple_nonzero
  59. ExportTests.test_export_with_map_cond
  60. ExportTests.test_exported_graph_serialization
  61. ExportTests.test_map_cond_param_buffer_lifted
  62. ExportTests.test_torch_inference_mode_ctx
  63. ExportTests.test_untracked_inputs_in_constraints
  64. FakeTensorOperatorInvariants.test_like_ops
  65. FakeTensorOperatorInvariants.test_non_kwarg_only_device
  66. FakeTensorOperatorInvariants.test_tensor_constructors_all_have_kwarg_device
  67. FunctionTests.test_default_dict
  68. FunctionTests.test_default_dict_closure
  69. FunctionTests.test_default_dict_lambda
  70. FunctionTests.test_is_contiguous_frame_counts
  71. FunctionTests.test_math_radians
  72. FunctionTests.test_partials_as_input_partials_lambda
  73. FunctionTests.test_partials_as_input_partials_mod
  74. FuncTorchHigherOrderOpTests.test_grad
  75. FuncTorchHigherOrderOpTests.test_grad_disable_capture
  76. FuncTorchHigherOrderOpTests.test_grad_fn_with_kwargs
  77. FuncTorchHigherOrderOpTests.test_grad_freevar_python_scalar
  78. FuncTorchHigherOrderOpTests.test_grad_has_aux
  79. FuncTorchHigherOrderOpTests.test_grad_non_tensor_input
  80. FuncTorchHigherOrderOpTests.test_grad_pytree
  81. FuncTorchHigherOrderOpTests.test_grad_two_tensor_all_grad_has_aux
  82. FuncTorchHigherOrderOpTests.test_grad_two_tensor_has_aux
  83. FuncTorchHigherOrderOpTests.test_grad_with_graph_break
  84. FuncTorchHigherOrderOpTests.test_grad_with_side_effect
  85. FuncTorchHigherOrderOpTests.test_vmap
  86. FuncTorchHigherOrderOpTests.test_vmap_disable_capture
  87. FuncTorchHigherOrderOpTests.test_vmap_free_const
  88. FuncTorchHigherOrderOpTests.test_vmap_free_tensor
  89. FuncTorchHigherOrderOpTests.test_vmap_kwargs
  90. FuncTorchHigherOrderOpTests.test_vmap_multiple_invocation_in_dims
  91. FuncTorchHigherOrderOpTests.test_vmap_multiple_invocation_out_dims
  92. FuncTorchHigherOrderOpTests.test_vmap_multiple_outputs
  93. FuncTorchHigherOrderOpTests.test_vmap_multiple_outputs_diff_dims
  94. FuncTorchHigherOrderOpTests.test_vmap_multiple_outputs_out_dims_tuple
  95. FuncTorchHigherOrderOpTests.test_vmap_over_vmap_captured
  96. FuncTorchHigherOrderOpTests.test_vmap_over_vmap_two_inputs
  97. FuncTorchHigherOrderOpTests.test_vmap_previous_illegal_op_no_graph_break
  98. FuncTorchHigherOrderOpTests.test_vmap_pytree_inputs
  99. FuncTorchHigherOrderOpTests.test_vmap_recompile_same_config
  100. FuncTorchHigherOrderOpTests.test_vmap_recompile_with_randomness
  101. FuncTorchHigherOrderOpTests.test_vmap_side_effects
  102. FuncTorchHigherOrderOpTests.test_vmap_two_inputs
  103. FuncTorchHigherOrderOpTests.test_vmap_two_inputs_tuple_in_dims
  104. FuncTorchHigherOrderOpTests.test_vmap_with_conditional_graph_break
  105. FuncTorchHigherOrderOpTests.test_vmap_with_graph_break
  106. FuncTorchHigherOrderOpTests.test_vmap_with_graph_break_2
  107. FuncTorchHigherOrderOpTests.test_vmap_with_graph_break_lambda
  108. HigherOrderOpTests.test_capture_untracked_global_nested
  109. HigherOrderOpTests.test_capture_value_created_in_subgraph
  110. HigherOrderOpTests.test_cond_branches_no_arguments
  111. HigherOrderOpTests.test_cond_branches_no_arguments_no_closure
  112. HigherOrderOpTests.test_cond_pytree_operands
  113. HigherOrderOpTests.test_cond_pytree_operands_with_non_tensor_leaves
  114. HigherOrderOpTests.test_cond_source_fn_stack
  115. HigherOrderOpTests.test_cond_subgraph_name_is_valid
  116. HigherOrderOpTests.test_fallback_on_python_primitives_output
  117. HigherOrderOpTests.test_map_lowers_to_graph
  118. HigherOrderOpTests.test_map_multi_return
  119. HigherOrderOpTests.test_map_pytree_return
  120. HigherOrderOpTests.test_map_source_fn_stack
  121. HigherOrderOpTests.test_map_subgraph_name_is_valid
  122. HigherOrderOpTests.test_map_symint_input
  123. HigherOrderOpTests.test_modules
  124. HigherOrderOpTests.test_nested_tuple_output
  125. HigherOrderOpTests.test_output_with_dict
  126. HigherOrderOpTests.test_side_effect_in_body
  127. HigherOrderOpTests.test_vmap_source_fn_stack
  128. HigherOrderOpTests.test_wrap_kwarg_recompile
  129. HigherOrderOpTests.test_wrap_subgraph_name_is_valid
  130. HooksTests.test_functools_arg_vary
  131. HooksTests.test_post_acc_grad_hook
  132. HooksTests.test_tensor_register_global_hook
  133. HooksTests.test_tensor_register_hook_in_graph_local
  134. HooksTests.test_tensor_register_multiple_hooks
  135. HooksTests.test_tensor_register_multiple_hooks_handles_in_list
  136. InteropTests.test_vmap_in_graph
  137. LoggingTests.test_aot
  138. LoggingTests.test_aot_graphs
  139. LoggingTests.test_aot_joint_graph
  140. LoggingTests.test_bytecode
  141. LoggingTests.test_custom_format
  142. LoggingTests.test_custom_format_exc
  143. LoggingTests.test_dynamo_debug
  144. LoggingTests.test_dynamo_error
  145. LoggingTests.test_dynamo_info
  146. LoggingTests.test_graph
  147. LoggingTests.test_graph_breaks
  148. LoggingTests.test_graph_code
  149. LoggingTests.test_graph_sizes
  150. LoggingTests.test_graph_sizes_dynamic
  151. LoggingTests.test_guards_recompiles
  152. LoggingTests.test_inductor_debug
  153. LoggingTests.test_inductor_error
  154. LoggingTests.test_inductor_info
  155. LoggingTests.test_multiline_format
  156. LoggingTests.test_output_code
  157. LoggingTests.test_recompiles
  158. LoggingTests.test_trace_source_cond
  159. LoggingTests.test_trace_source_nested
  160. MiscTests.test_add_to_set
  161. MiscTests.test_any_all_symnode
  162. MiscTests.test_backend_match_guard
  163. MiscTests.test_boolarg
  164. MiscTests.test_callpacked
  165. MiscTests.test_closure_out_of_scope_cell
  166. MiscTests.test_compare_shapes_with_constant
  167. MiscTests.test_cond_nested
  168. MiscTests.test_cond_side_effects
  169. MiscTests.test_dataclass_fields
  170. MiscTests.test_deque_append_left
  171. MiscTests.test_deque_input
  172. MiscTests.test_deterministic_algorithms_mutated
  173. MiscTests.test_dict_order_keys
  174. MiscTests.test_dict_order_keys_modules
  175. MiscTests.test_dict_order_keys_tensors
  176. MiscTests.test_dictcomp
  177. MiscTests.test_dtypes_no_graphbreaks
  178. MiscTests.test_dunder_new_function_inlining
  179. MiscTests.test_frozenset_torch_func_contains
  180. MiscTests.test_get_cache_entry
  181. MiscTests.test_guard_failure_fn
  182. MiscTests.test_guard_failure_fn2
  183. MiscTests.test_guard_failure_fn_shape_control
  184. MiscTests.test_guard_failure_fn_tensor_iter
  185. MiscTests.test_inference_mode
  186. MiscTests.test_inline_closure_not_loaded_by_parent
  187. MiscTests.test_inplace_param_update
  188. MiscTests.test_inplace_view_on_graph_input
  189. MiscTests.test_intermediary_tensor_grad_access
  190. MiscTests.test_is_compiling
  191. MiscTests.test_iter_set
  192. MiscTests.test_itertools_accumulate_symint_default_sum
  193. MiscTests.test_itertools_accumulate_tensors_default_sum
  194. MiscTests.test_itertools_groupby_pure_python_default_identify_func
  195. MiscTests.test_itertools_groupby_pure_python_key_func
  196. MiscTests.test_itertools_infinite_cycle
  197. MiscTests.test_itertools_infinite_repeat
  198. MiscTests.test_itertools_infinite_repeat_mutation
  199. MiscTests.test_itertools_repeat
  200. MiscTests.test_listcomp
  201. MiscTests.test_mandelbrot_numpy
  202. MiscTests.test_mark_static
  203. MiscTests.test_namedtuple1
  204. MiscTests.test_namedtuple2
  205. MiscTests.test_nan
  206. MiscTests.test_nested_closure
  207. MiscTests.test_nested_closure_mutation
  208. MiscTests.test_nested_optimize
  209. MiscTests.test_nested_optimize_decorator
  210. MiscTests.test_nested_optimize_run
  211. MiscTests.test_nn_module_getattr
  212. MiscTests.test_no_raise_guard_partial_constraint
  213. MiscTests.test_no_raise_guard_partial_constraint_across_break
  214. MiscTests.test_numpy_array_of_arrays
  215. MiscTests.test_numpy_force
  216. MiscTests.test_numpy_int_constant
  217. MiscTests.test_numpy_iter
  218. MiscTests.test_numpy_non_torch_dtype
  219. MiscTests.test_numpy_readonly
  220. MiscTests.test_numpy_recompilation_scalar
  221. MiscTests.test_numpy_size_attr
  222. MiscTests.test_numpy_subdtype
  223. MiscTests.test_numpy_tolist
  224. MiscTests.test_numpy_torch_operators
  225. MiscTests.test_numpy_with_builtin_type
  226. MiscTests.test_out_variants_with_resizing_on_graph_inputs
  227. MiscTests.test_pure_python_accumulate
  228. MiscTests.test_py_guards_mark_dynamic
  229. MiscTests.test_raise_guard_full_constraint
  230. MiscTests.test_raise_guard_partial_constraint_no_graph_break
  231. MiscTests.test_raise_on_backend_error
  232. MiscTests.test_recompile_on_global_state_change
  233. MiscTests.test_release_input_memory
  234. MiscTests.test_release_module_memory
  235. MiscTests.test_release_scope_memory
  236. MiscTests.test_repeat_interleave_graphbreaks
  237. MiscTests.test_return_nested_function
  238. MiscTests.test_set_aliasing_recompiles
  239. MiscTests.test_setattr_mutation1
  240. MiscTests.test_simple_set_usage
  241. MiscTests.test_size_input
  242. MiscTests.test_str_format_assert2
  243. MiscTests.test_tensor_build_list_unpack
  244. MiscTests.test_tensor_dict1
  245. MiscTests.test_tensor_dict2
  246. MiscTests.test_tensor_dict3
  247. MiscTests.test_tensor_item_capture
  248. MiscTests.test_tolist_0d
  249. MiscTests.test_tolist_1d
  250. MiscTests.test_tolist_kd
  251. MiscTests.test_tolist_kd_dynamic
  252. MiscTests.test_tolist_scalar
  253. MiscTests.test_torch_seed
  254. MiscTests.test_tracing_nested_py_tree
  255. MiscTests.test_tracing_nested_py_tree_dicts
  256. MiscTests.test_tracing_nested_py_tree_mixed_all
  257. MiscTests.test_tracing_nested_py_tree_tuples
  258. MiscTests.test_tracing_py_tree
  259. MiscTests.test_tracing_py_tree_tensor_subclass
  260. MiscTests.test_tracing_tree_map_only
  261. MiscTests.test_type_copy
  262. MiscTests.test_typing_typevar
  263. MiscTests.test_user_getattribute
  264. MiscTests.test_with_builtin_type
  265. MiscTests.test_yield_from
  266. MiscTests.test_yield_gen_and_from
  267. MiscTests.test_yield_send_to_subgenerator_graph_break
  268. NNModuleTests.test_lazy_module1
  269. NNModuleTests.test_lazy_module2
  270. NNModuleTests.test_lazy_module4
  271. NNModuleTests.test_lazy_module5
  272. NNModuleTests.test_lazy_module6
  273. NNModuleTests.test_lazy_module_no_cls_to_become
  274. NNModuleTests.test_self_mutating1
  275. NNModuleTests.test_unsupportedmethod
  276. NNModuleTests.test_unsupportedmodule
  277. NonStrictExportTestExport.test__scaled_dot_product_flash_attention_non_strict
  278. NonStrictExportTestExport.test_basic_non_strict_fake_tensor_non_strict
  279. NonStrictExportTestExport.test_basic_non_strict_real_tensor_non_strict
  280. NonStrictExportTestExport.test_buffer_util_non_strict
  281. NonStrictExportTestExport.test_cond_with_module_stack_export_with_non_strict
  282. NonStrictExportTestExport.test_export_decomps_dynamic_non_strict
  283. NonStrictExportTestExport.test_export_decomps_simple_non_strict
  284. NonStrictExportTestExport.test_export_with_wrong_inputs_non_strict
  285. NonStrictExportTestExport.test_external_call_non_strict_real_tensor_non_strict
  286. NonStrictExportTestExport.test_fqn_non_strict
  287. NonStrictExportTestExport.test_nn_module_stack_non_strict
  288. NonStrictExportTestExport.test_nn_module_stack_shared_submodule_non_strict
  289. NonStrictExportTestExport.test_non_strict_dynamic_shapes_non_strict
  290. NonStrictExportTestExport.test_non_strict_dynamic_shapes_suggested_fixes_non_strict
  291. NonStrictExportTestExport.test_param_util_non_strict
  292. NonStrictExportTestExport.test_raise_user_error_when_guard_on_data_dependent_operation_non_strict
  293. NonStrictExportTestExport.test_sym_sqrt_non_strict
  294. NonStrictExportTestExport.test_to_module_with_mutated_buffer_multiple_non_strict
  295. NonStrictExportTestExport.test_to_module_with_mutated_buffer_multiple_update_sub_later_non_strict
  296. NonStrictExportTestExport.test_to_module_with_mutated_buffer_non_strict
  297. NumpyTestsCPU.test_boolean_indexing_weirdness_cpu
  298. NumpyTestsCPU.test_boolean_shape_mismatch_cpu
  299. NumpyTestsCPU.test_empty_fancy_index_cpu
  300. NumpyTestsCPU.test_index_no_floats_cpu
  301. OptimizedModuleTest.test_backward_hooks
  302. OptimizedModuleTest.test_cache_size_limit_on_guarded_nn_modules
  303. OptimizedModuleTest.test_composition_with_opt_mod
  304. OptimizedModuleTest.test_hooks_inner
  305. OptimizedModuleTest.test_hooks_outer
  306. OptimizedModuleTest.test_hooks_skip_guards
  307. OptimizedModuleTest.test_module_dict_iter_name
  308. OptimizedModuleTest.test_nn_module
  309. OptimizedModuleTest.test_no_recompile_on_nn_guarded_modules
  310. OptimizedModuleTest.test_to
  311. OptimizerTests.test_adagrad
  312. OptimizerTests.test_adam
  313. OptimizerTests.test_adamax
  314. OptimizerTests.test_adamw
  315. OptimizerTests.test_asgd
  316. OptimizerTests.test_nadam
  317. OptimizerTests.test_rmsprop
  318. OptimizerTests.test_rprop
  319. OptimizerTests.test_sgd
  320. PackedSequenceTest.test_pack_sequence
  321. PackedSequenceTest.test_to
  322. PackedSequenceTest.test_total_length
  323. PackedSequenceTest.test_type_casts
  324. RecompileTests.test_aliasing_guard_failures_with_globals
  325. RecompileTests.test_automatic_dynamic_reduce_recompiles
  326. RecompileTests.test_automatic_dynamic_tensor_scalar_change
  327. RecompileTests.test_dynamic_shape_parameter_recompile
  328. RecompileTests.test_recompiles_true_false_flop
  329. RecompileUxTests.test_mismatched_type
  330. RecompileUxTests.test_verbose_tensor_check
  331. ReproTests.test_add_sub_alpha_out
  332. ReproTests.test_addr_alpha_beta_out
  333. ReproTests.test_boxes_len
  334. ReproTests.test_chunk_reformer_ff
  335. ReproTests.test_convert_boxes_to_pooler_format
  336. ReproTests.test_create_rand_mask_from_inputs
  337. ReproTests.test_do_paste_mask
  338. ReproTests.test_dynamic_shapes_float_guard
  339. ReproTests.test_dynamic_shapes_implicit_guard
  340. ReproTests.test_empty_list_contains_with_jump
  341. ReproTests.test_function_in_skipfiles
  342. ReproTests.test_functools_wraps
  343. ReproTests.test_hf_t5_forward
  344. ReproTests.test_hf_xsoftmax_training
  345. ReproTests.test_issue175
  346. ReproTests.test_jit_trace_errors
  347. ReproTests.test_list_aliasing
  348. ReproTests.test_list_self_reference
  349. ReproTests.test_longformer_chunk
  350. ReproTests.test_merge_criteria_processor_list1
  351. ReproTests.test_merge_criteria_processor_list2
  352. ReproTests.test_module_in_skipfiles
  353. ReproTests.test_multi_import
  354. ReproTests.test_negative_shape_guard
  355. ReproTests.test_numpy_not_ndarray_recompiles
  356. ReproTests.test_recursive_map
  357. ReproTests.test_reformer_eval
  358. ReproTests.test_reformer_min_chunk_len
  359. ReproTests.test_reformer_sorting
  360. ReproTests.test_relative_import
  361. ReproTests.test_relative_import_no_modulename
  362. ReproTests.test_restricted_list_subclass1
  363. ReproTests.test_rewrite_assert_noop
  364. ReproTests.test_rewrite_assert_with_msg
  365. ReproTests.test_rewrite_assert_with_non_string_msg
  366. ReproTests.test_seq_append_list
  367. ReproTests.test_size_typematch
  368. ReproTests.test_tensor_data_kwarg
  369. ReproTests.test_threading_local
  370. ReproTests.test_validate_model_kwargs
  371. RetraceExportTestDynamismExpression.test_export_inline_constraints_retraceability
  372. RetraceExportTestExport.test_cond_with_module_stack_export_with_retraceability
  373. RetraceExportTestExport.test_constrain_size_in_eager_retraceability
  374. RetraceExportTestExport.test_constrain_size_with_constrain_value_retraceability
  375. RetraceExportTestExport.test_constrain_size_with_various_cases_retraceability
  376. RetraceExportTestExport.test_nn_module_stack_retraceability
  377. RetraceExportTestExport.test_nn_module_stack_shared_submodule_retraceability
  378. RetraceExportTestExport.test_non_strict_dynamic_shapes_retraceability
  379. RetraceExportTestExport.test_non_strict_dynamic_shapes_suggested_fixes_retraceability
  380. SerDesExportTestDynamismExpression.test_export_inline_constraints_serdes
  381. SerDesExportTestExport.test_basic_non_strict_fake_tensor_serdes
  382. SerDesExportTestExport.test_basic_non_strict_real_tensor_serdes
  383. SerDesExportTestExport.test_cond_with_module_stack_export_with_serdes
  384. SerDesExportTestExport.test_constrain_size_in_eager_serdes
  385. SerDesExportTestExport.test_constrain_size_with_constrain_value_serdes
  386. SerDesExportTestExport.test_constrain_size_with_various_cases_serdes
  387. SerDesExportTestExport.test_external_call_non_strict_real_tensor_serdes
  388. SerDesExportTestExport.test_nn_module_stack_serdes
  389. SerDesExportTestExport.test_nn_module_stack_shared_submodule_serdes
  390. SerDesExportTestExport.test_non_strict_dynamic_shapes_serdes
  391. SerDesExportTestExport.test_non_strict_dynamic_shapes_suggested_fixes_serdes
  392. SkipNonTensorTests.test_add_tensor1
  393. SkipNonTensorTests.test_add_tensor2
  394. SkipNonTensorTests.test_add_tensor_dict
  395. SkipNonTensorTests.test_add_tensor_list
  396. SkipNonTensorTests.test_do_not_skip_side_effects
  397. SkipNonTensorTests.test_recursive_list
  398. SubclassTests.test_compile_higher_order_with_functionalization
  399. SubclassTests.test_compile_with_functionalization
  400. SubclassTests.test_torch_function_state_guards
  401. SubclassTests.test_wrapper_subclass_guards_on_inner_tensor
  402. SubclassTests.test_wrapper_subclass_with_differently_sized_inner_tensor
  403. SubclassTests.test_wrapper_subclass_with_same_sized_inner_tensor
  404. SubGraphTests.test_dynamic_duck_size
  405. SubGraphTests.test_dynamic_getitem
  406. SubGraphTests.test_dynamic_kwarg
  407. SubGraphTests.test_dynamic_order_dependence
  408. SubGraphTests.test_dynamic_zero_inference
  409. SubGraphTests.test_enumerate_not_break_graph
  410. SubGraphTests.test_no_graph_break_on_item
  411. TestAnalyze.test_trace_dependencies
  412. TestAOTAutograd.test_input_mutation_aliases_bases_out_of_order
  413. TestAOTAutograd.test_input_mutation_false_aliasing
  414. TestAOTAutograd.test_input_mutation_noncontiguous
  415. TestAOTAutograd.test_output_aliases_multiple_inputs_get_correct_one
  416. TestAOTAutograd.test_set__and_data_mutation_good
  417. TestAOTAutograd.test_some_output_requires_grad_input_doesnt
  418. TestAOTAutograd.test_some_outputs_dont_require_grad_non_view
  419. TestAOTAutograd.test_some_outputs_dont_require_grad_view
  420. TestAOTAutograd.test_view_and_inplace_view
  421. TestAOTDispatch.test_aot_dispatch_input_mutation
  422. TestAOTDispatch.test_aot_dispatch_input_mutation_and_output_alias
  423. TestAOTDispatch.test_aot_dispatch_output_alias
  424. TestAOTDispatch.test_aot_dispatch_simple
  425. TestApplyAlongAxis.test_0d_array
  426. TestApplyAlongAxis.test_axis_insertion
  427. TestApplyAlongAxis.test_scalar_array
  428. TestArange.test_infinite
  429. TestArange.test_nan_step
  430. TestArange.test_require_range
  431. TestArange.test_zero_step
  432. TestArgmax.test_combinations_data62
  433. TestArgmax.test_combinations_data63
  434. TestArgmax.test_combinations_data65
  435. TestArgmax.test_combinations_data66
  436. TestArrayAttributeDeletion.test_multiarray_not_writable_attributes_deletion
  437. TestArrayAttributeDeletion.test_multiarray_writable_attributes_deletion
  438. TestArrayCreationCopyArgument.test_scalars
  439. TestArrayCreationCopyArgument.test_striding_not_ok
  440. TestArrayFromScalar.test_integers_np_byte_np_longlong
  441. TestArrayFromScalar.test_integers_np_int__np_longlong
  442. TestArrayFromScalar.test_integers_np_intc_np_longlong
  443. TestArrayFromScalar.test_integers_np_longlong_np_longlong
  444. TestArrayFromScalar.test_integers_np_longlong_t26
  445. TestArrayFromScalar.test_integers_np_short_np_longlong
  446. TestArrayFromScalar.test_integers_t15_np_longlong
  447. TestArraySplit.test_integer_0_split
  448. TestAsArrayCPU.test_copy_list_cpu_bfloat16
  449. TestAsArrayCPU.test_copy_list_cpu_bool
  450. TestAsArrayCPU.test_copy_list_cpu_complex128
  451. TestAsArrayCPU.test_copy_list_cpu_complex64
  452. TestAsArrayCPU.test_copy_list_cpu_float16
  453. TestAsArrayCPU.test_copy_list_cpu_float32
  454. TestAsArrayCPU.test_copy_list_cpu_float64
  455. TestAsArrayCPU.test_copy_list_cpu_int16
  456. TestAsArrayCPU.test_copy_list_cpu_int32
  457. TestAsArrayCPU.test_copy_list_cpu_int64
  458. TestAsArrayCPU.test_copy_list_cpu_int8
  459. TestAsArrayCPU.test_copy_list_cpu_uint8
  460. TestAssignment.test_assignment_broadcasting
  461. TestAttributes.test_fill_max_uint64
  462. TestAttributes.test_fill_readonly
  463. TestAttributes.test_fill_struct_array
  464. TestAutograd.test_access_saved_tensor_twice_without_recomputation_works
  465. TestAutograd.test_anomaly_mode_no_check_nan
  466. TestAutograd.test_autograd_simple_views_python
  467. TestAutograd.test_autograd_views_codegen
  468. TestAutograd.test_backward_with_inputs
  469. TestAutograd.test_checkpoint_detects_non_determinism
  470. TestAutograd.test_checkpointing_non_reentrant_autocast_cpu
  471. TestAutograd.test_checkpointing_without_reentrant_custom_function_works
  472. TestAutograd.test_checkpointing_without_reentrant_with_context_fn
  473. TestAutograd.test_custom_function_forward_mode_inplace_checks
  474. TestAutograd.test_custom_function_forward_mode_view_checks
  475. TestAutograd.test_custom_function_saved_tensors
  476. TestAutograd.test_gradcheck_forward_ad
  477. TestAutograd.test_gradcheck_nondeterministic
  478. TestAutograd.test_hook_closure_cycle_use_custom_function_True_use_tensor_hook_False
  479. TestAutograd.test_nested_anomaly_detect_nan
  480. TestAutograd.test_nested_anomaly_printstack_cleanup
  481. TestAutograd.test_profiler
  482. TestAutograd.test_profiler_aggregation_table
  483. TestAutograd.test_profiler_propagation
  484. TestAutograd.test_profiler_seq_nr
  485. TestAutograd.test_profiler_shapes
  486. TestAutograd.test_record_function
  487. TestAutograd.test_record_function_callbacks
  488. TestAutograd.test_record_function_legacy
  489. TestAutograd.test_return_duplicate
  490. TestAutograd.test_return_duplicate_inplace
  491. TestAutograd.test_save_on_cpu_and_checkpoint
  492. TestAutograd.test_set_grad_coroutines
  493. TestAutograd.test_set_grad_coroutines_exit
  494. TestAutograd.test_set_grad_generator_functions
  495. TestAutograd.test_set_grad_generator_functions_recursive
  496. TestAutograd.test_sparse_mm_backward
  497. TestAutogradDeviceTypeCPU.test_inplace_on_view_modify_base_cpu
  498. TestAutogradDeviceTypeCPU.test_inplace_on_view_python_cpu
  499. TestAutogradForwardMode.test_create_new_zeros_with_same_meta
  500. TestAutogradForwardMode.test_detach_view_tracking
  501. TestAutogradForwardMode.test_forward_level_cleanup
  502. TestAutogradForwardMode.test_make_dual_inference_tensor_in_inference_mode
  503. TestAutogradForwardMode.test_view_inplace_always_creates_a_view
  504. TestAutogradForwardMode.test_view_inplace_differentiable_views
  505. TestAutogradForwardMode.test_view_inplace_non_differentiable_views
  506. TestAutogradInferenceMode.test_inference_mode_decorator
  507. TestAutogradInferenceMode.test_inference_mode_inf_tensor_in_inf_mode_inplace_op
  508. TestAutogradLogging.test_logging
  509. TestBaseMath.test_lower_align
  510. TestBaseSparsifier.test_state_dict
  511. TestBitsCPU.test_cat_cpu
  512. TestBool.test_sum_2
  513. TestBooleanIndexing.test_bool_as_int_argument_errors
  514. TestBooleanIndexing.test_boolean_indexing_weirdness
  515. TestBroadcast.test_broadcast_error_kwargs
  516. TestBroadcast.test_broadcast_single_arg
  517. TestBufferProtocolCPU.test_byte_to_int_cpu
  518. TestClip.test_clip_func_takes_out
  519. TestClip.test_simple_complex
  520. TestColumnStack.test_non_iterable
  521. TestCommonPass.test_correctness_CSEPass_Mutation_cpu
  522. TestCommonPass.test_correctness_CSEPass_MutationInput_cpu
  523. TestCommonPass.test_correctness_CSEPass_MutationMetadata_cpu
  524. TestCommonPass.test_correctness_CSEPass_MutationTorchTensorCall_cpu
  525. TestCommonPass.test_correctness_CSEPass_ReturnList_cpu
  526. TestCommonPass.test_correctness_CSEPass_TakeList_cpu
  527. TestCommonPass.test_correctness_factory_CSEPass_FactoryFunctionCall_cpu
  528. TestCommonPass.test_correctness_factory_CSEPass_MutationFactory_cpu
  529. TestCompileTransformsCPU.test_compile_vmap_hessian_cpu
  530. TestComposability.test_convert_without_squash_mask
  531. TestComposability.test_fusion_before_s_prep
  532. TestComposability.test_q_prep_before_s_prep
  533. TestComposability.test_qat_prep_before_s_prep
  534. TestComposability.test_s_prep_before_fusion
  535. TestComposability.test_s_prep_before_q_prep
  536. TestComposability.test_s_prep_before_qat_prep
  537. TestComposabilityCPU.test_autograd_function_no_setup_context_transform_hessian_cpu
  538. TestComposabilityCPU.test_autograd_function_no_setup_context_transform_jacfwd_cpu
  539. TestComposabilityCPU.test_deprecation_transforms_transform_functionalize_cpu
  540. TestComposabilityCPU.test_requires_grad_inside_transform_cpu
  541. TestConcatenate.test_bad_out_shape
  542. TestConcatenate.test_concatenate
  543. TestConcatenate.test_exceptions
  544. TestConcatenate.test_large_concatenate_axis_None
  545. TestCond.test_empty_sq_cases
  546. TestCond.test_sq_cases
  547. TestContentStoreCPU.test_repeated_hash_cpu
  548. TestControlFlow.test_map_autograd_nested_list
  549. TestControlFlow.test_map_autograd_no_grad_output
  550. TestControlFlow.test_map_dict_in_out
  551. TestControlFlow.test_map_list_in_out
  552. TestControlFlowTraced.test_map_functionalized
  553. TestControlFlowTraced.test_map_functionalized_aot_func
  554. TestControlFlowTraced.test_nested_cond_map_cond_symbolic
  555. TestControlFlowTraced.test_nested_map_cond_real
  556. TestControlFlowTraced.test_nested_map_cond_symbolic
  557. TestControlFlowTraced.test_tracing_map_autograd_aot_functionalized
  558. TestControlFlowTraced.test_tracing_map_autograd_symbolic_dict
  559. TestControlFlowTraced.test_tracing_map_autograd_symbolic_list
  560. TestControlFlowTraced.test_tracing_map_autograd_symbolic_simple
  561. TestControlFlowTraced.test_tracing_map_real
  562. TestControlFlowTraced.test_tracing_map_symbolic_dict
  563. TestControlFlowTraced.test_tracing_map_symbolic_list
  564. TestControlFlowTraced.test_tracing_map_symbolic_simple
  565. TestConversion.test_to_int_scalar
  566. TestCov.test_aweights
  567. TestCov.test_fweights
  568. TestCppExtensionJIT.test_cpp_frontend_module_has_same_output_as_python
  569. TestCppExtensionJIT.test_cpp_frontend_module_has_up_to_date_attribute
  570. TestCppExtensionJIT.test_cpp_frontend_module_has_up_to_date_attributes
  571. TestCppExtensionOpenRgistration.test_open_device_registration
  572. TestCross.test_broadcasting_shapes
  573. TestCustomBackendAPI.test_aot_autograd_api
  574. TestCustomBackendAPI.test_lookup_backend
  575. TestCustomBackendAPI.test_register_backend_api
  576. TestCustomOp.test_impl_device_cpu
  577. TestCustomOp.test_legacy_define
  578. TestCustomOpTestingCPU.test_missing_functionalization_cpu
  579. TestDelete.test_slices
  580. TestDeserialize.test_tensor_tensor_list
  581. TestDiag.test_failure
  582. TestDiff.test_append
  583. TestDiff.test_axis
  584. TestDiff.test_n
  585. TestDiff.test_prepend
  586. TestDistributionShapes.test_mixture_same_family_shape
  587. TestDLPack.test_dlpack_device
  588. TestDLPack.test_dtype_passthrough_dtype0
  589. TestDLPack.test_dtype_passthrough_dtype1
  590. TestDLPack.test_dtype_passthrough_dtype2
  591. TestDLPack.test_dtype_passthrough_dtype3
  592. TestDLPack.test_dtype_passthrough_dtype4
  593. TestDLPack.test_dtype_passthrough_dtype5
  594. TestDLPack.test_dtype_passthrough_dtype6
  595. TestDLPack.test_dtype_passthrough_dtype7
  596. TestDLPack.test_dtype_passthrough_dtype8
  597. TestDLPack.test_dtype_passthrough_dtype9
  598. TestDLPack.test_dunder_dlpack_refcount
  599. TestDLPack.test_from_dlpack_refcount
  600. TestDLPack.test_from_torch
  601. TestDLPack.test_higher_dims_ndim_0
  602. TestDLPack.test_higher_dims_ndim_1
  603. TestDLPack.test_higher_dims_ndim_10
  604. TestDLPack.test_higher_dims_ndim_11
  605. TestDLPack.test_higher_dims_ndim_12
  606. TestDLPack.test_higher_dims_ndim_13
  607. TestDLPack.test_higher_dims_ndim_14
  608. TestDLPack.test_higher_dims_ndim_15
  609. TestDLPack.test_higher_dims_ndim_16
  610. TestDLPack.test_higher_dims_ndim_17
  611. TestDLPack.test_higher_dims_ndim_18
  612. TestDLPack.test_higher_dims_ndim_19
  613. TestDLPack.test_higher_dims_ndim_2
  614. TestDLPack.test_higher_dims_ndim_20
  615. TestDLPack.test_higher_dims_ndim_21
  616. TestDLPack.test_higher_dims_ndim_22
  617. TestDLPack.test_higher_dims_ndim_23
  618. TestDLPack.test_higher_dims_ndim_24
  619. TestDLPack.test_higher_dims_ndim_25
  620. TestDLPack.test_higher_dims_ndim_26
  621. TestDLPack.test_higher_dims_ndim_27
  622. TestDLPack.test_higher_dims_ndim_28
  623. TestDLPack.test_higher_dims_ndim_29
  624. TestDLPack.test_higher_dims_ndim_3
  625. TestDLPack.test_higher_dims_ndim_30
  626. TestDLPack.test_higher_dims_ndim_31
  627. TestDLPack.test_higher_dims_ndim_32
  628. TestDLPack.test_higher_dims_ndim_4
  629. TestDLPack.test_higher_dims_ndim_5
  630. TestDLPack.test_higher_dims_ndim_6
  631. TestDLPack.test_higher_dims_ndim_7
  632. TestDLPack.test_higher_dims_ndim_8
  633. TestDLPack.test_higher_dims_ndim_9
  634. TestDLPack.test_ndim0
  635. TestDropoutNN.test_invalid_dropout_p
  636. TestDsplit.test_0D_array
  637. TestDsplit.test_1D_array
  638. TestDsplit.test_2D_array
  639. TestDsplit.test_non_iterable
  640. TestDstack.test_non_iterable
  641. TestDynamicQuantizedOps.test_qlinear
  642. TestDynamismExpression.test_export_inline_constraints
  643. TestEigh.test_invalid
  644. TestEigvalsh.test_invalid
  645. TestEmbeddingNN.test_embedding_max_norm
  646. TestEmbeddingNN.test_embedding_sparse_basic
  647. TestEmbeddingNN.test_embedding_sparse_empty_tensor
  648. TestEmbeddingNN.test_embeddingbag_include_last_offset
  649. TestExamplesCorrectnessCPU.test_resnet18_per_sample_grads_mechanism_functional_call_cpu
  650. TestExamplesCorrectnessCPU.test_resnet18_per_sample_grads_mechanism_make_functional_cpu
  651. TestExpandDims.test_axis_out_of_range
  652. TestExpandDims.test_repeated_axis
  653. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_conv1d_cpu_float64
  654. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_conv2d_cpu_float64
  655. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_conv3d_cpu_float64
  656. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_group_norm_cpu_float64
  657. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_instance_norm_cpu_float64
  658. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_layer_norm_cpu_float64
  659. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_conv1d_cpu_float64
  660. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_conv2d_cpu_float64
  661. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_conv3d_cpu_float64
  662. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_group_norm_cpu_float64
  663. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_instance_norm_cpu_float64
  664. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_layer_norm_cpu_float64
  665. TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_conv1d_cpu_float64
  666. TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_conv2d_cpu_float64
  667. TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_conv3d_cpu_float64
  668. TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_group_norm_cpu_float64
  669. TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_instance_norm_cpu_float64
  670. TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_layer_norm_cpu_float64
  671. TestExperiment.test_with_buffer_as_submodule
  672. TestExperimentalUtils.test_profiler_name_pattern
  673. TestExperimentalUtils.test_profiler_optimizer_single_tensor_pattern
  674. TestExperimentalUtils.test_profiler_pattern_match_helper
  675. TestExperimentalUtils.test_profiler_pattern_matcher_json_report
  676. TestExperimentalUtils.test_profiler_synchronized_dataloader_pattern
  677. TestExperimentalUtils.test_utils_compute_queue_depth_when_no_cuda_events
  678. TestExperimentalUtils.test_utils_compute_self_time
  679. TestExport.test_basic_non_strict_fake_tensor
  680. TestExport.test_basic_non_strict_real_tensor
  681. TestExport.test_cond_with_module_stack_export_with
  682. TestExport.test_constrain_size_in_eager
  683. TestExport.test_constrain_size_with_constrain_value
  684. TestExport.test_constrain_size_with_various_cases
  685. TestExport.test_external_call_non_strict_real_tensor
  686. TestExport.test_nn_module_stack
  687. TestExport.test_nn_module_stack_shared_submodule
  688. TestExport.test_non_strict_dynamic_shapes
  689. TestExport.test_non_strict_dynamic_shapes_suggested_fixes
  690. TestFakeQuantizeOps.test_backward_per_channel
  691. TestFakeQuantizeOps.test_backward_per_tensor
  692. TestFakeQuantizeOps.test_learnable_backward_per_channel_cpu
  693. TestFakeQuantizeOps.test_learnable_backward_per_channel_cuda
  694. TestFakeQuantizeOps.test_learnable_backward_per_tensor_cuda
  695. TestFakeQuantizeOps.test_learnable_forward_per_channel_cuda
  696. TestFakeQuantizeOps.test_learnable_forward_per_tensor_cpu
  697. TestFakeQuantizeOps.test_learnable_forward_per_tensor_cuda
  698. TestFlag.test_writeable_from_readonly
  699. TestFlip.test_axes
  700. TestFliplr.test_basic
  701. TestFloatNonIntegerArgument.test_non_integer_argument_errors
  702. TestFloatNonIntegerArgument.test_reduce_axis_float_index
  703. TestFromBuffer.test_basic_little_dtype0
  704. TestFromBuffer.test_basic_little_dtype1
  705. TestFromBuffer.test_basic_little_dtype2
  706. TestFromBuffer.test_empty
  707. TestFunctionalizeCPU.test_multioutput_view_cpu
  708. TestFunctionalizeCPU.test_simple_view_cpu
  709. TestFunctionalizeCPU.test_vmap_functionalize_jvp_cpu
  710. TestFunctors.test_cat_transform
  711. TestFunctors.test_cat_transform_non_uniform
  712. TestFXGraphMatcherModels.test_mobilenet_v2_qat
  713. TestGenerateNumericDebugHandle.test_quantize_pt2e_preserve_handle
  714. TestGenerateOpcheckTests.test_opcheck_bad_op
  715. TestGenericCumSumProd.test_bad_axis_func0
  716. TestGenericCumSumProd.test_bad_axis_func1
  717. TestGenericPytree.test_flatten_unflatten_deque_cxx
  718. TestGenericPytree.test_flatten_unflatten_deque_py
  719. TestGenericReductions.test_bad_axis_func0
  720. TestGenericReductions.test_bad_axis_func1
  721. TestGenericReductions.test_bad_axis_func10
  722. TestGenericReductions.test_bad_axis_func11
  723. TestGenericReductions.test_bad_axis_func2
  724. TestGenericReductions.test_bad_axis_func3
  725. TestGenericReductions.test_bad_axis_func4
  726. TestGenericReductions.test_bad_axis_func5
  727. TestGenericReductions.test_bad_axis_func6
  728. TestGenericReductions.test_bad_axis_func7
  729. TestGenericReductions.test_bad_axis_func8
  730. TestGenericReductions.test_bad_axis_func9
  731. TestGradient.test_badargs
  732. TestGradient.test_second_order_accurate
  733. TestGradient.test_specific_axes
  734. TestGradient.test_values
  735. TestGradNewOnesOverride.test_newones
  736. TestHessianCPU.test_hessian_vectorize_correctness_simple_cpu
  737. TestHessianCPU.test_jacfwd_different_levels_cpu
  738. TestHigherOrderOperatorInteractionCPU.test_grad_name_wrapping_cpu
  739. TestHistogram.test_error_binnum_type
  740. TestHistogram.test_finite_range
  741. TestHistogram.test_one_bin
  742. TestHistogram.test_unsigned_monotonicity_check
  743. TestHistogram2d.test_binparameter_combination
  744. TestHistogramdd.test_bins_error_2
  745. TestHistogramdd.test_bins_errors
  746. TestHistogramdd.test_equal_edges
  747. TestHistogramdd.test_inf_edges
  748. TestHistogramdd.test_weights
  749. TestHsplit.test_0D_array
  750. TestHsplit.test_non_iterable
  751. TestHstack.test_empty_input
  752. TestHstack.test_non_iterable
  753. TestIndexing.test_index_no_floats
  754. TestIndexing.test_slicing_no_floats
  755. TestIndexingCPU.test_byte_mask_cpu
  756. TestIndexingCPU.test_empty_ndim_index_bool_cpu
  757. TestIndexingCPU.test_index_cpu
  758. TestIndexingCPU.test_index_limits_cpu
  759. TestIndexingCPU.test_out_of_bound_index_cpu
  760. TestIndexingCPU.test_zero_dim_index_cpu
  761. TestInputAttrTracking.test_complex_attr_access_without_graph_breaks
  762. TestInputAttrTracking.test_const_property_on_tensor
  763. TestInputAttrTracking.test_set_data_on_input_tensor
  764. TestInputAttrTracking.test_tensor_property_assigned_on_tensor
  765. TestInputAttrTracking.test_tensor_property_on_tensor
  766. TestIsScalar.test_is_not_scalar_value6
  767. TestIterator.test_iterator
  768. TestIterDataPipeGraphFastForward.test_simple_snapshot_custom_non_generator
  769. TestIterDataPipeGraphFastForward.test_simple_snapshot_custom_self_next
  770. TestJacCPU.test_against_reference_correctness_different_devices_cpu
  771. TestJacCPU.test_against_reference_default_arg_cpu
  772. TestJacCPU.test_against_reference_default_arg_jacfwd_cpu
  773. TestJacCPU.test_against_reference_multi_input_cpu
  774. TestJacCPU.test_against_reference_multi_input_multi_output_cpu
  775. TestJacCPU.test_against_reference_simple_cpu
  776. TestJacCPU.test_against_reference_simple_jacfwd_cpu
  777. TestJacCPU.test_against_reference_unrelated_outputs_cpu
  778. TestJacCPU.test_against_reference_zero_dim_cpu
  779. TestJacCPU.test_argnums_defaults_to_zero_cpu
  780. TestJacCPU.test_argnums_defaults_to_zero_jacfwd_cpu
  781. TestJacCPU.test_aux_pytree_cpu
  782. TestJacCPU.test_complex_error_cpu
  783. TestJacCPU.test_dimensionality_cpu
  784. TestJacCPU.test_empty_output_cpu
  785. TestJacCPU.test_inplace_cpu
  786. TestJacCPU.test_jac_with_non_tensor_args_cpu
  787. TestJacCPU.test_multiple_inputs_outputs_pytree_cpu
  788. TestJacCPU.test_multiple_inputs_outputs_pytree_jacfwd_cpu
  789. TestJacCPU.test_multiple_inputs_pytree_cpu
  790. TestJacCPU.test_multiple_inputs_pytree_jacfwd_cpu
  791. TestJacCPU.test_multiple_outputs_multiple_argnums_cpu
  792. TestJacCPU.test_multiple_outputs_multiple_argnums_jacfwd_cpu
  793. TestJacCPU.test_multiple_outputs_pytree_jacfwd_cpu
  794. TestJacCPU.test_multiple_outputs_single_argnums_cpu
  795. TestJacCPU.test_multiple_outputs_single_argnums_jacfwd_cpu
  796. TestJacCPU.test_outputs_can_any_pytree_cpu
  797. TestJacCPU.test_unrelated_input_cpu
  798. TestJacCPU.test_unrelated_output_cpu
  799. TestJit.test_batchnorm
  800. TestJit.test_function_default_values
  801. TestJitGeneratedModule.test_nn_Bilinear
  802. TestJitGeneratedModule.test_nn_Embedding
  803. TestJitGeneratedModule.test_nn_Embedding_discontiguous
  804. TestJitGeneratedModule.test_nn_Embedding_sparse
  805. TestJitGeneratedModule.test_nn_EmbeddingBag_discontiguous
  806. TestJitGeneratedModule.test_nn_EmbeddingBag_max
  807. TestJitGeneratedModule.test_nn_EmbeddingBag_max_padding_idx
  808. TestJitGeneratedModule.test_nn_EmbeddingBag_mean
  809. TestJitGeneratedModule.test_nn_EmbeddingBag_mean_padding_idx
  810. TestJitGeneratedModule.test_nn_EmbeddingBag_sparse
  811. TestJitGeneratedModule.test_nn_EmbeddingBag_sum
  812. TestJitGeneratedModule.test_nn_EmbeddingBag_sum_padding_idx
  813. TestJitGeneratedModule.test_nn_Linear
  814. TestJitGeneratedModule.test_nn_Linear_no_batch_dim
  815. TestJitGeneratedModule.test_nn_PReLU_no_batch_dim
  816. TestJvpCPU.test_multiple_inputs_outputs_cpu
  817. TestJvpCPU.test_pytree_inputs_cpu
  818. TestLazyModules.test_lazy_batchnorm_with_dict_input
  819. TestLazyModules.test_lazy_conv1d
  820. TestLazyModules.test_lazy_conv1d_pickle
  821. TestLazyModules.test_lazy_conv2d
  822. TestLazyModules.test_lazy_conv2d_pickle
  823. TestLazyModules.test_lazy_conv3d
  824. TestLazyModules.test_lazy_conv3d_pickle
  825. TestLazyModules.test_lazy_conv_transpose1d_kwargs
  826. TestLazyModules.test_lazy_conv_transpose1d_pickle
  827. TestLazyModules.test_lazy_conv_transpose2d
  828. TestLazyModules.test_lazy_conv_transpose2d_kwargs
  829. TestLazyModules.test_lazy_conv_transpose2d_pickle
  830. TestLazyModules.test_lazy_conv_transpose3d
  831. TestLazyModules.test_lazy_conv_transpose3d_kwargs
  832. TestLazyModules.test_lazy_conv_transpose3d_pickle
  833. TestLazyModules.test_lazy_conv_transposed1d
  834. TestLazyModules.test_lazy_linear_pickle
  835. TestLazyModules.test_linear
  836. TestLexsort.test_datetime
  837. TestLexsort.test_mixed
  838. TestLinalgCPU.test_addmm_sizes_cpu_float32
  839. TestLinalgCPU.test_addmm_sizes_cpu_float64
  840. TestLinalgCPU.test_addr_integral_cpu_int16
  841. TestLinalgCPU.test_addr_integral_cpu_int32
  842. TestLinalgCPU.test_addr_integral_cpu_int64
  843. TestLinalgCPU.test_addr_integral_cpu_int8
  844. TestLinalgCPU.test_addr_integral_cpu_uint8
  845. TestLinalgCPU.test_geqrf_cpu_complex128
  846. TestLinalgCPU.test_geqrf_cpu_complex64
  847. TestLinalgCPU.test_geqrf_cpu_float32
  848. TestLinalgCPU.test_geqrf_cpu_float64
  849. TestLinalgCPU.test_householder_product_cpu_complex128
  850. TestLinalgCPU.test_householder_product_cpu_complex64
  851. TestLinalgCPU.test_householder_product_cpu_float32
  852. TestLinalgCPU.test_householder_product_cpu_float64
  853. TestLinalgCPU.test_lobpcg_torchscript_cpu_float64
  854. TestLinalgCPU.test_pinv_cpu_complex128
  855. TestLinalgCPU.test_pinv_cpu_complex64
  856. TestLinalgCPU.test_pinv_cpu_float32
  857. TestLinalgCPU.test_pinv_cpu_float64
  858. TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_complex128
  859. TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_complex64
  860. TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_float32
  861. TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_float64
  862. TestLinalgCPU.test_solve_cpu_complex128
  863. TestLinalgCPU.test_solve_cpu_complex64
  864. TestLinalgCPU.test_solve_cpu_float32
  865. TestLinalgCPU.test_solve_cpu_float64
  866. TestLinalgCPU.test_tensorinv_errors_and_warnings_cpu_complex128
  867. TestLinalgCPU.test_tensorinv_errors_and_warnings_cpu_complex64
  868. TestLinalgCPU.test_tensorinv_errors_and_warnings_cpu_float32
  869. TestLinalgCPU.test_tensorinv_errors_and_warnings_cpu_float64
  870. TestMatmul.test_exceptions
  871. TestMedian.test_extended_axis_invalid
  872. TestMeshgrid.test_indexing
  873. TestMetaKernel.test_addmm_invalid_dtype
  874. TestMin.test_adapt
  875. TestMin.test_attn
  876. TestMin.test_big_split
  877. TestMin.test_c
  878. TestMin.test_compare_dims
  879. TestMin.test_diag
  880. TestMin.test_dim_args
  881. TestMin.test_dims_with_size
  882. TestMin.test_dir
  883. TestMin.test_doc
  884. TestMin.test_embed
  885. TestMin.test_eq
  886. TestMin.test_expand
  887. TestMin.test_functorch
  888. TestMin.test_hello
  889. TestMin.test_index
  890. TestMin.test_index_placement
  891. TestMin.test_inplace
  892. TestMin.test_manual_stuff
  893. TestMin.test_mask
  894. TestMin.test_max
  895. TestMin.test_mm
  896. TestMin.test_mm_fuse
  897. TestMin.test_monkey
  898. TestMin.test_network
  899. TestMin.test_order
  900. TestMin.test_order_keyword
  901. TestMin.test_parse
  902. TestMin.test_permute_orig
  903. TestMin.test_seg
  904. TestMin.test_simple
  905. TestMin.test_softmax_split
  906. TestMin.test_stack
  907. TestMin.test_time_mm_fuse
  908. TestMin.test_with_dims_split
  909. TestMinFunctorchOnly.test_adapt
  910. TestMinFunctorchOnly.test_attn
  911. TestMinFunctorchOnly.test_big_split
  912. TestMinFunctorchOnly.test_c
  913. TestMinFunctorchOnly.test_compare_dims
  914. TestMinFunctorchOnly.test_diag
  915. TestMinFunctorchOnly.test_dim_args
  916. TestMinFunctorchOnly.test_dims_with_size
  917. TestMinFunctorchOnly.test_dir
  918. TestMinFunctorchOnly.test_doc
  919. TestMinFunctorchOnly.test_embed
  920. TestMinFunctorchOnly.test_eq
  921. TestMinFunctorchOnly.test_expand
  922. TestMinFunctorchOnly.test_functorch
  923. TestMinFunctorchOnly.test_hello
  924. TestMinFunctorchOnly.test_index
  925. TestMinFunctorchOnly.test_index_placement
  926. TestMinFunctorchOnly.test_inplace
  927. TestMinFunctorchOnly.test_manual_stuff
  928. TestMinFunctorchOnly.test_mask
  929. TestMinFunctorchOnly.test_max
  930. TestMinFunctorchOnly.test_mm
  931. TestMinFunctorchOnly.test_mm_fuse
  932. TestMinFunctorchOnly.test_monkey
  933. TestMinFunctorchOnly.test_network
  934. TestMinFunctorchOnly.test_order
  935. TestMinFunctorchOnly.test_order_keyword
  936. TestMinFunctorchOnly.test_parse
  937. TestMinFunctorchOnly.test_permute_orig
  938. TestMinFunctorchOnly.test_seg
  939. TestMinFunctorchOnly.test_simple
  940. TestMinFunctorchOnly.test_softmax_split
  941. TestMinFunctorchOnly.test_stack
  942. TestMinFunctorchOnly.test_with_dims_split
  943. TestMinMax.test_axis
  944. TestMinMax.test_scalar
  945. TestMisc.test_byteorder_check
  946. TestMisc.test_generalized_raise_multiloop
  947. TestMkldnnCPU.test_add_cpu
  948. TestMkldnnCPU.test_copy_cpu
  949. TestMkldnnCPU.test_mul_cpu
  950. TestMkldnnCPU.test_sigmoid_cpu
  951. TestMkldnnCPU.test_tanh_cpu
  952. TestModuleHookNN.test_hook_inplace
  953. TestMoveaxis.test_errors
  954. TestMultiDot.test_dynamic_programming_optimization_and_out
  955. TestMultiDot.test_three_arguments_and_out
  956. TestMultiDot.test_too_few_input_arrays
  957. TestMultiDot.test_two_arguments_and_out
  958. TestMultiprocessing.test_empty_shared
  959. TestMultiprocessing.test_fs_is_shared
  960. TestMultiprocessing.test_inherit_tensor
  961. TestMultiprocessing.test_is_shared
  962. TestNamedTensor.test_addmm
  963. TestNamedTensor.test_bmm
  964. TestNamedTensor.test_cat
  965. TestNamedTensor.test_flatten
  966. TestNamedTensor.test_info_smoke
  967. TestNamedTensor.test_logical_ops
  968. TestNamedTensor.test_matmul
  969. TestNamedTensor.test_mm
  970. TestNamedTensor.test_reduction_fns
  971. TestNamedTensor.test_set_names_property
  972. TestNamedTensor.test_transpose_variants
  973. TestNamedTensor.test_unary_propagate_names_fns
  974. TestNamedTensor.test_unflatten
  975. TestNamedTensor.test_using_unseen_interned_string_bumps_refcount_permanently
  976. TestNamedTensor.test_using_unseen_uninterned_string_refcounts
  977. TestNamedTuple.test_max
  978. TestNegative.test_exceptions
  979. TestNestedCheckpoint.test_nested_checkpoint_early_stop_False
  980. TestNestedCheckpoint.test_nested_checkpoint_early_stop_True
  981. TestNestedCheckpoint.test_nested_checkpoint_kwargs_early_stop_False
  982. TestNestedCheckpoint.test_nested_checkpoint_kwargs_early_stop_True
  983. TestNestedCheckpoint.test_nested_checkpoint_non_tensor_inputs_and_outputs_early_stop_False
  984. TestNestedCheckpoint.test_nested_checkpoint_non_tensor_inputs_and_outputs_early_stop_True
  985. TestNestedCheckpoint.test_nested_checkpoint_reentrant_backwards_early_stop_False
  986. TestNestedCheckpoint.test_nested_checkpoint_reentrant_backwards_early_stop_True
  987. TestNestedCheckpoint.test_nested_checkpoint_same_graph_early_stop_False
  988. TestNestedCheckpoint.test_nested_checkpoint_same_graph_early_stop_True
  989. TestNestedCheckpoint.test_nested_checkpoint_set_early_stop
  990. TestNestedCheckpoint.test_nested_checkpoint_set_early_stop_no_recompution_needed
  991. TestNestedCheckpoint.test_nested_checkpoint_two_children_early_stop_False
  992. TestNestedCheckpoint.test_nested_checkpoint_two_children_early_stop_True
  993. TestNestedTensor.test_binary_recompiles
  994. TestNestedTensor.test_unbind
  995. TestNN.test_bilinear_broadcasting
  996. TestNN.test_fb_fc_packed
  997. TestNN.test_layer_norm_grads_with_create_graph_flag
  998. TestNN.test_linear_autograd_device_cpu_bias_weightCOO
  999. TestNN.test_linear_autograd_device_cpu_nobias_weightCOO
  1000. TestNN.test_linear_broadcasting
  1001. TestNN.test_module_apply_inplace_op
  1002. TestNN.test_ParameterDict
  1003. TestNN.test_ParameterList
  1004. TestNN.test_Sequential_iadd
  1005. TestNNDeviceTypeCPU.test_invalid_reduction_strings_cpu
  1006. TestNNDeviceTypeCPU.test_module_to_empty_cpu_float32
  1007. TestNNDeviceTypeCPU.test_module_to_empty_cpu_float64
  1008. TestNNDeviceTypeCPU.test_nll_loss_byte_target_matches_long_cpu
  1009. TestNNDeviceTypeCPU.test_threshold_inplace_overlap_cpu
  1010. TestNNDeviceTypeCUDA.test_nll_loss_byte_target_matches_long_cuda
  1011. TestNNParametrization.test_deepcopy_after_parametrization_swap_True
  1012. TestNNParametrization.test_errors_unparametrized_tensor_parametrization_swap_False
  1013. TestNNParametrization.test_errors_unparametrized_tensor_parametrization_swap_True
  1014. TestNNParametrization.test_initialization_parametrization_swap_True
  1015. TestNNParametrization.test_register_and_remove_buffer_parametrization_swap_True
  1016. TestNNParametrization.test_register_and_remove_nested_parametrization_swap_True
  1017. TestNNParametrization.test_register_and_remove_parametrization_swap_True
  1018. TestNNParametrization.test_serialization_parametrization_swap_True
  1019. TestNNParametrization.test_transfer_parametrizations_and_params_right_inverse_swap_True
  1020. TestNNParametrization.test_transfer_parametrizations_and_params_swap_True
  1021. TestNNParametrization.test_wrapper_subclass_parametrization_swap_True
  1022. TestNonarrayArgs.test_dunder_round_edgecases_val_2147483647_ndigits_-1
  1023. TestNonarrayArgs.test_dunder_round_edgecases_val_2147483647_ndigits_-10
  1024. TestNonarrayArgs.test_dunder_round_edgecases_val_2147483647_ndigits_-9
  1025. TestNonzeroAndCountNonzero.test_count_nonzero_axis
  1026. TestNormDouble.test_axis
  1027. TestNormDouble.test_bad_args
  1028. TestNormDouble.test_matrix_2x2
  1029. TestNormInt64.test_axis
  1030. TestNormInt64.test_bad_args
  1031. TestNormInt64.test_matrix_2x2
  1032. TestNormSingle.test_axis
  1033. TestNormSingle.test_bad_args
  1034. TestNormSingle.test_matrix_2x2
  1035. TestNumPyInteropCPU.test_numpy_non_writeable_cpu
  1036. TestObserver.test_per_channel_observers
  1037. TestObserver.test_per_tensor_observers
  1038. TestOpCPU.test_cat_cpu_bfloat16
  1039. TestOpCPU.test_cat_cpu_float32
  1040. TestOptimizations.test_example_inputs
  1041. TestOptimizations.test_example_inputs_runtime_use
  1042. TestOutDtypeOp.test_out_dtype_non_op_overload
  1043. TestOutDtypeOp.test_out_dtype_wrong_output
  1044. TestPadding.test_constant_padNd
  1045. TestPasses.test_functionalize_inline_contraints
  1046. TestPasses.test_views_op_having_view_copy
  1047. TestPassManager.test_pass_manager
  1048. TestPercentile.test_extended_axis_invalid
  1049. TestPercentile.test_scalar_q
  1050. TestPickle.test_pickle
  1051. TestPoolingNN.test_MaxUnpool2d_output_size
  1052. TestProfiler.test_concrete_inputs_profiling
  1053. TestProfiler.test_export_stacks
  1054. TestProfiler.test_flops
  1055. TestProfiler.test_guarded_record_function_fast
  1056. TestProfiler.test_high_level_trace
  1057. TestProfiler.test_is_profiler_enabled
  1058. TestProfiler.test_profiler_correlation_id
  1059. TestProfiler.test_profiler_fwd_bwd_link
  1060. TestProfiler.test_profiler_op_event_args
  1061. TestProfiler.test_profiler_tracing
  1062. TestProfiler.test_source
  1063. TestProfiler.test_source_multithreaded_basic_work_in_main_thread_True
  1064. TestProfiler.test_source_multithreaded_close_in_scope_work_in_main_thread_True
  1065. TestProfiler.test_source_multithreaded_complex_work_in_main_thread_True
  1066. TestProfiler.test_source_multithreaded_multiple_preexisting_work_in_main_thread_True
  1067. TestProfiler.test_source_multithreaded_open_in_scope_work_in_main_thread_True
  1068. TestProfilerTree.test_profiler_experimental_tree_with_memory
  1069. TestProfilerTree.test_profiler_experimental_tree_with_memory_and_stack
  1070. TestProfilerTree.test_profiler_experimental_tree_with_record_function
  1071. TestProfilerTree.test_profiler_experimental_tree_with_stack_and_torch_dispatch
  1072. TestProfilerTree.test_profiler_experimental_tree_with_stack_and_torch_function
  1073. TestPruningNN.test_identity_pruning
  1074. TestPruningNN.test_random_pruning_0perc
  1075. TestPT2ERepresentation.test_add
  1076. TestPT2ERepresentation.test_add_relu
  1077. TestPT2ERepresentation.test_conv2d
  1078. TestPT2ERepresentation.test_dynamic_linear
  1079. TestPT2ERepresentation.test_maxpool2d
  1080. TestPT2ERepresentation.test_qdq
  1081. TestPT2ERepresentation.test_qdq_per_channel
  1082. TestPT2ERepresentation.test_static_linear
  1083. TestPutAlongAxis.test_broadcast
  1084. TestPutmask.test_byteorder_greater_False
  1085. TestPutmask.test_byteorder_greater_True
  1086. TestPutmask.test_record_array
  1087. TestPythonAutograd.test_backwards2
  1088. TestPythonAutograd.test_forwards1
  1089. TestPythonAutograd.test_forwards2
  1090. TestPythonAutograd.test_split
  1091. TestPythonDispatch.test_make_subclass_with_modes
  1092. TestPythonRegistration.test_alias_analysis
  1093. TestPythonRegistration.test_finalizer
  1094. TestQR.test_mode_raw
  1095. TestQR.test_qr_empty_m_0_n_0
  1096. TestQR.test_qr_empty_m_0_n_3
  1097. TestQR.test_qr_empty_m_3_n_0
  1098. TestQuantile.test_quantile_monotonic_method_averaged_inverted_cdf
  1099. TestQuantile.test_quantile_monotonic_method_closest_observation
  1100. TestQuantile.test_quantile_monotonic_method_hazen
  1101. TestQuantile.test_quantile_monotonic_method_interpolated_inverted_cdf
  1102. TestQuantile.test_quantile_monotonic_method_inverted_cdf
  1103. TestQuantile.test_quantile_monotonic_method_median_unbiased
  1104. TestQuantile.test_quantile_monotonic_method_normal_unbiased
  1105. TestQuantile.test_quantile_monotonic_method_weibull
  1106. TestQuantizedEmbeddingOps.test_embedding_bag_2d_indices
  1107. TestQuantizedOps.test_custom_module_lstm
  1108. TestQuantizedOps.test_max_pool2d_cudnn
  1109. TestQuantizedOps.test_qtanh
  1110. TestQuantizedTensor.test_choose_qparams
  1111. TestQuantizedTensor.test_choose_qparams_optimized
  1112. TestQuantizedTensor.test_decomposed_dequantize_per_tensor
  1113. TestQuantizedTensor.test_decomposed_quantize_per_tensor
  1114. TestQuantizedTensor.test_decomposed_quantize_per_tensor_bfloat16_input
  1115. TestQuantizedTensor.test_fp16_saturate_op
  1116. TestQuantizedTensor.test_qtensor_cpu
  1117. TestQuantizedTensor.test_quantize_per_channel_sub_byte
  1118. TestQuantizeFx.test_prepare_custom_config_set_standalone_module_class
  1119. TestQuantizeFx.test_static_lstm_with_custom_fixed_qparams
  1120. TestQuantizePT2E.test_composable_quantizer_linear_conv
  1121. TestQuantizePT2E.test_constant_prop_preserve_metadata
  1122. TestQuantizePT2E.test_embedding_conv_linear_quantization
  1123. TestQuantizePT2E.test_fold_all_ops_before_quantize
  1124. TestQuantizePT2E.test_fold_quantize
  1125. TestQuantizePT2E.test_fold_quantize_per_channel
  1126. TestQuantizePT2E.test_groupwise_per_channel_quant
  1127. TestQuantizePT2E.test_reentrant
  1128. TestQuantizePT2E.test_save_load
  1129. TestQuantizePT2E.test_speed
  1130. TestQuantizePT2EQAT_ConvBn1d.test_prepare_qat_conv_bn_fusion_getitem_placeholder
  1131. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_fusion
  1132. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_fusion_literal_args
  1133. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_fusion_no_conv_bias
  1134. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_relu_fusion
  1135. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_relu_fusion_no_conv_bias
  1136. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_no_bias
  1137. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_transpose_bn
  1138. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_transpose_bn_relu
  1139. TestQuantizePT2EQAT_ConvBn1d.test_qat_inplace_add_relu
  1140. TestQuantizePT2EQAT_ConvBn1d.test_qat_preserve_source_fn_stack
  1141. TestQuantizePT2EQAT_ConvBn1d.test_qat_update_shared_qspec
  1142. TestQuantizePT2EQAT_ConvBn2d.test_prepare_qat_conv_bn_fusion_getitem_placeholder
  1143. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_fusion
  1144. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_fusion_literal_args
  1145. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_fusion_no_conv_bias
  1146. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_relu_fusion
  1147. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_relu_fusion_no_conv_bias
  1148. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_no_bias
  1149. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_transpose_bn
  1150. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_transpose_bn_relu
  1151. TestQuantizePT2EQAT_ConvBn2d.test_qat_inplace_add_relu
  1152. TestQuantizePT2EQAT_ConvBn2d.test_qat_preserve_source_fn_stack
  1153. TestQuantizePT2EQAT_ConvBn2d.test_qat_update_shared_qspec
  1154. TestQuantizePT2EQATModels.test_qat_mobilenet_v2
  1155. TestQuantizePT2EQATModels.test_qat_resnet18
  1156. TestRecordFunction.test_datapipe_with_record_function
  1157. TestRecordFunction.test_datapipe_with_record_function_fork
  1158. TestRecordFunction.test_record_function
  1159. TestReductionsCPU.test_warn_invalid_degrees_of_freedom_cpu_complex128
  1160. TestReductionsCPU.test_warn_invalid_degrees_of_freedom_cpu_complex64
  1161. TestReductionsCPU.test_warn_invalid_degrees_of_freedom_cpu_float32
  1162. TestReductionsCPU.test_warn_invalid_degrees_of_freedom_cpu_float64
  1163. TestRequire.test_C_and_F_simul
  1164. TestRequire.test_require_each
  1165. TestRequire.test_unknown_requirement
  1166. TestRollaxis.test_exceptions
  1167. TestRot90.test_basic
  1168. TestScalarIndexing.test_invalid_newaxis
  1169. TestScalarIndexing.test_invalid_subscript
  1170. TestScalarIndexing.test_invalid_subscript_assignment
  1171. TestScalarTypeNames.test_names_reflect_attributes_t0
  1172. TestScalarTypeNames.test_names_reflect_attributes_t1
  1173. TestScalarTypeNames.test_names_reflect_attributes_t2
  1174. TestScalarTypeNames.test_names_reflect_attributes_t3
  1175. TestScalarTypeNames.test_names_reflect_attributes_t4
  1176. TestScalarTypeNames.test_names_reflect_attributes_t5
  1177. TestScalarTypeNames.test_names_reflect_attributes_t6
  1178. TestScalarTypeNames.test_names_reflect_attributes_t7
  1179. TestScalarTypeNames.test_names_reflect_attributes_t8
  1180. TestScalarTypeNames.test_names_reflect_attributes_t9
  1181. TestScript.test_conv_error
  1182. TestScript.test_dict_str
  1183. TestScript.test_empty_tuple_str
  1184. TestScript.test_error_stacktrace_interface
  1185. TestScript.test_function_overloading_isinstance
  1186. TestScript.test_function_overloads
  1187. TestScript.test_ignored_as_value
  1188. TestScript.test_infer_size
  1189. TestScript.test_is_after_use
  1190. TestScript.test_is_scripting
  1191. TestScript.test_isinstance_dynamic
  1192. TestScript.test_method_overloading
  1193. TestScript.test_namedtuple_default_values_using_factory_constructor
  1194. TestScript.test_namedtuple_python
  1195. TestScript.test_nested_breaks
  1196. TestScript.test_no_self_arg_ignore_function
  1197. TestScript.test_none_type_str
  1198. TestScript.test_parse_nested_names
  1199. TestScript.test_parse_tensor_constants
  1200. TestScript.test_python_call
  1201. TestScript.test_python_call_non_tensor_wrong
  1202. TestScript.test_python_op_builtins
  1203. TestScript.test_script_optional_none
  1204. TestScript.test_string_device_implicit_conversion
  1205. TestScript.test_torch_functional_tensordot_int
  1206. TestScript.test_torchscript_multi_head_attn
  1207. TestScript.test_tuple_str
  1208. TestScript.test_type_annotation_module
  1209. TestScript.test_unspecialized_any_binding
  1210. TestScript.test_unused_decorator
  1211. TestScript.test_wrong_return_type
  1212. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_bfloat16_cpu_bfloat16
  1213. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_float16_cpu_float16
  1214. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_float32_cpu_float32
  1215. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_float64_cpu_float64
  1216. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_bfloat16_cpu_bfloat16
  1217. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_float16_cpu_float16
  1218. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_float32_cpu_float32
  1219. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_float64_cpu_float64
  1220. TestShapeOpsCUDA.test_flip_cuda_float32
  1221. TestSortAndSelectCPU.test_sort_overflow_cpu_int16
  1222. TestSortComplex.test_sort_real_type_in_g_type_out_G
  1223. TestSortComplex.test_sort_real_type_in_H_type_out_F
  1224. TestSplit.test_unequal_split
  1225. TestStackMisc.test_stack
  1226. TestStats.test_dtype_from_dtype
  1227. TestStats.test_out
  1228. TestTake.test_raise
  1229. TestTakeAlongAxis.test_invalid
  1230. TestTEFuserDynamic.test_profiler
  1231. TestTEFuserDynamic.test_remove_output_used_only_in_size
  1232. TestTEFuserDynamic.test_to_dtype
  1233. TestTEFuserDynamic.test_torch_to
  1234. TestTEFuserStatic.test_inlined_optimized_graph
  1235. TestTEFuserStatic.test_profiler
  1236. TestTEFuserStatic.test_remove_output_used_only_in_size
  1237. TestTEFuserStatic.test_skip_grad_in_check
  1238. TestTEFuserStatic.test_to_dtype
  1239. TestTEFuserStatic.test_torch_to
  1240. TestTensorBoardEmbedding.test_embedding
  1241. TestTensorBoardEmbedding.test_embedding_64
  1242. TestTensorBoardSummary.test_image_with_one_channel
  1243. TestTensorBoardSummary.test_image_without_channel
  1244. TestTensorBoardUtils.test_numpy_vid_uint8
  1245. TestTensorCreationCPU.test_block_diag_cpu
  1246. TestTensorCreationCPU.test_constructor_dtypes_cpu
  1247. TestTorch.test_cuda_not_built
  1248. TestTorch.test_map
  1249. TestTorch.test_new
  1250. TestTorch.test_parsing_intlist
  1251. TestTorch.test_pin_memory
  1252. TestTorch.test_type
  1253. TestTorch.test_upsample_nearest2d_meta
  1254. TestTorchDeviceTypeCPU.test_broadcast_fn_add_cpu
  1255. TestTorchDeviceTypeCPU.test_broadcast_fn_addcdiv_cpu
  1256. TestTorchDeviceTypeCPU.test_broadcast_fn_addcmul_cpu
  1257. TestTorchDeviceTypeCPU.test_broadcast_fn_atan2_cpu
  1258. TestTorchDeviceTypeCPU.test_broadcast_fn_copy_cpu
  1259. TestTorchDeviceTypeCPU.test_broadcast_fn_div_cpu
  1260. TestTorchDeviceTypeCPU.test_broadcast_fn_eq_cpu
  1261. TestTorchDeviceTypeCPU.test_broadcast_fn_fmod_cpu
  1262. TestTorchDeviceTypeCPU.test_broadcast_fn_ge_cpu
  1263. TestTorchDeviceTypeCPU.test_broadcast_fn_gt_cpu
  1264. TestTorchDeviceTypeCPU.test_broadcast_fn_le_cpu
  1265. TestTorchDeviceTypeCPU.test_broadcast_fn_lerp_cpu
  1266. TestTorchDeviceTypeCPU.test_broadcast_fn_lt_cpu
  1267. TestTorchDeviceTypeCPU.test_broadcast_fn_map2_cpu
  1268. TestTorchDeviceTypeCPU.test_broadcast_fn_map_cpu
  1269. TestTorchDeviceTypeCPU.test_broadcast_fn_masked_fill_cpu
  1270. TestTorchDeviceTypeCPU.test_broadcast_fn_masked_scatter_cpu
  1271. TestTorchDeviceTypeCPU.test_broadcast_fn_mul_cpu
  1272. TestTorchDeviceTypeCPU.test_broadcast_fn_ne_cpu
  1273. TestTorchDeviceTypeCPU.test_broadcast_fn_pow_cpu
  1274. TestTorchDeviceTypeCPU.test_broadcast_fn_remainder_cpu
  1275. TestTorchDeviceTypeCPU.test_broadcast_fn_sub_cpu
  1276. TestTorchDeviceTypeCPU.test_deterministic_empty_cpu_uint64
  1277. TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool1d_cpu_float32
  1278. TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool1d_cpu_float64
  1279. TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool2d_cpu_float32
  1280. TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool2d_cpu_float64
  1281. TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool3d_cpu_float32
  1282. TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool3d_cpu_float64
  1283. TestTorchFunctionMode.test_mode_notimplemented_loop
  1284. TestTorchFunctionMode.test_modes_return_notimplemented
  1285. TestTorchFunctionMode.test_nested_modes_with_python_has_torch_function
  1286. TestTorchFunctionOverride.test_Tensor___cuda_array_interface_____get__
  1287. TestTorchTidyProfiler.test_allocation_id_uniqueness
  1288. TestTorchTidyProfiler.test_allocation_ids
  1289. TestTorchTidyProfiler.test_allocation_ids_with_other_ops
  1290. TestTorchTidyProfiler.test_allocations
  1291. TestTorchTidyProfiler.test_extra_fields
  1292. TestTorchTidyProfiler.test_impl_reuse
  1293. TestTorchTidyProfiler.test_mkldnn_tensors
  1294. TestTorchTidyProfiler.test_module_and_optimizer_ids
  1295. TestTorchTidyProfiler.test_nnmodule_params
  1296. TestTorchTidyProfiler.test_pointers_and_ids
  1297. TestTorchTidyProfiler.test_scalar_ins
  1298. TestTorchTidyProfiler.test_tensor_lists
  1299. TestTorchTidyProfiler.test_tensor_properties
  1300. TestTorchTidyProfiler.test_tensorimpl_invalidation_full
  1301. TestTorchTidyProfiler.test_tensorimpl_invalidation_keep_alive
  1302. TestTorchTidyProfiler.test_tensorimpl_invalidation_scalar_args
  1303. TestTorchTidyProfiler.test_tensorimpl_invalidation_set
  1304. TestTrilIndicesFrom.test_exceptions
  1305. TestTriuIndicesFrom.test_exceptions
  1306. TestTypeHints.test_doc_examples
  1307. TestTypePromotionCPU.test_alpha_mismatch_cpu
  1308. TestTypePromotionCPU.test_alternate_result_cpu
  1309. TestUnflatten.test_unflatten_container_type
  1310. TestUnique.test_unique_axis
  1311. TestUnique.test_unique_axis_errors
  1312. TestUtils.test_get_fqn_to_example_inputs_complex_args
  1313. TestUtils.test_get_fqn_to_example_inputs_default_kwargs
  1314. TestUtils.test_get_fqn_to_example_inputs_simple
  1315. TestVerifyCorrectness.test_incorrect_verify_true
  1316. TestVerifyCorrectness.test_torchscript
  1317. TestVmapAPI.test_data_attribute
  1318. TestVmapAPI.test_fallback_does_not_warn_by_default
  1319. TestVmapAPI.test_fallback_warns_when_warnings_are_enabled
  1320. TestVmapAPI.test_out_dim_out_of_bounds_err_msg
  1321. TestVmapOperators.test_new_empty_strided
  1322. TestVsplit.test_0D_array
  1323. TestVsplit.test_1D_array
  1324. TestVsplit.test_non_iterable
  1325. TestVstack.test_empty_input
  1326. TestVstack.test_non_iterable
  1327. TestWhere.test_error
  1328. TestXNNPACKQuantizer.test_add_and_inplace_add
  1329. TestXNNPACKQuantizer.test_add_mul_long
  1330. TestXNNPACKQuantizer.test_add_mul_scalar
  1331. TestXNNPACKQuantizer.test_conv1d
  1332. TestXNNPACKQuantizer.test_conv1d_with_conv2d
  1333. TestXNNPACKQuantizer.test_conv2d
  1334. TestXNNPACKQuantizer.test_conv_linear
  1335. TestXNNPACKQuantizer.test_conv_linear_no_permute
  1336. TestXNNPACKQuantizer.test_dynamic_linear
  1337. TestXNNPACKQuantizer.test_dynamic_linear_int4_weight
  1338. TestXNNPACKQuantizer.test_dynamic_linear_with_conv
  1339. TestXNNPACKQuantizer.test_gru
  1340. TestXNNPACKQuantizer.test_linear
  1341. TestXNNPACKQuantizer.test_linear_gru
  1342. TestXNNPACKQuantizer.test_linear_relu
  1343. TestXNNPACKQuantizer.test_linear_with_dynamic_shape
  1344. TestXNNPACKQuantizer.test_mul_and_inplace_mul
  1345. TestXNNPACKQuantizer.test_mul_float32_max
  1346. TestXNNPACKQuantizer.test_obs_sharing_ops
  1347. TestXNNPACKQuantizer.test_propagate_annotation
  1348. TestXNNPACKQuantizer.test_qat_dynamic_linear
  1349. TestXNNPACKQuantizerModels.test_resnet18
  1350. TraceRuleTests.test_skipfiles_inlinelist
  1351. UnspecTests.test_builtin_max_min
  1352. UnspecTests.test_conv1d_symint_padding
  1353. UnspecTests.test_isinstance_symint
  1354. UnspecTests.test_mark_01_dynamic
  1355. UnspecTests.test_no_recompilations
  1356. UnspecTests.test_no_recompiles
  1357. UnspecTests.test_propagate_dynamic_dim
  1358. UnspecTests.test_use_and_specialize