| .. |
|
ActivationCheckpointingTests.test_cond_with_kwargs
|
|
|
|
AotAutogradFallbackTests.test_aot_sequence_nr
|
|
|
|
AutogradFunctionTests.test_graph_break_if_lifted_free_variable
|
|
|
|
AutogradFunctionTests.test_print_in_bwd
|
|
|
|
AutogradFunctionTests.test_stride_in_bwd
|
|
|
|
ComptimeTests.test_graph_break
|
|
|
|
CtxManagerTests.test_autograd_profiler_enabled
|
|
|
|
CtxManagerTests.test_disable_saved_tensors_hooks
|
|
|
|
CtxManagerTests.test_disable_saved_tensors_hooks_prev_disabled
|
|
|
|
CtxManagerTests.test_disable_saved_tensors_hooks_prev_disabled_nested
|
|
|
|
DecoratorTests.test_mark_static_address_guarded
|
|
|
|
DecoratorTests.test_mark_static_address_unguarded
|
|
|
|
DefaultsTests.test_cast_tensor_single_elem
|
|
|
|
DefaultsTests.test_compare_constant_and_tensor
|
|
|
|
DefaultsTests.test_func_default_torch_args
|
|
|
|
DefaultsTests.test_in_set_inplace
|
|
|
|
DefaultsTests.test_in_set_would_fail_broadcast
|
|
|
|
DynamoProfilerTests.test_dynamo_timed_profiling_isolated
|
|
|
|
DynamoProfilerTests.test_profiler_cache_lookup
|
|
|
|
DynamoProfilerTests.test_profiler_dynamo_compiled_region
|
|
|
|
End2EndTests.test_init_group
|
|
|
|
ExampleTests.test_exportdb_supported_case_assume_constant_result
|
|
|
|
ExampleTests.test_exportdb_supported_case_constrain_as_size_example
|
|
|
|
ExcTests.test_backend_suppress_line
|
|
|
|
ExcTests.test_graph_break_log
|
|
|
|
ExcTests.test_internal_error_no_suppress
|
|
|
|
ExcTests.test_internal_error_suppress_errors
|
|
|
|
ExcTests.test_not_implemented_error
|
|
|
|
ExcTests.test_trigger_bisect_on_error
|
|
|
|
ExcTests.test_trigger_on_error
|
|
|
|
ExportTests.test_cond_raise_user_error_on_non_list_operands
|
|
|
|
ExportTests.test_cond_raise_user_error_on_non_tensor_operands
|
|
|
|
ExportTests.test_cond_raise_user_error_on_unsupported_pred
|
|
|
|
ExportTests.test_export_decomp
|
|
|
|
ExportTests.test_export_dynamic_dim_cleanup
|
|
|
|
ExportTests.test_export_dynamic_dim_range_constraint
|
|
|
|
ExportTests.test_export_mark_dynamic_conflict_dynamic_dim
|
|
|
|
ExportTests.test_export_multi_dynamic_dim_constraint
|
|
|
|
ExportTests.test_export_multi_dynamic_dim_unsafe_relationship
|
|
|
|
ExportTests.test_export_no_raise
|
|
|
|
ExportTests.test_export_preserve_constraints_as_metadata_scalar
|
|
|
|
ExportTests.test_export_raise_on_relationship
|
|
|
|
ExportTests.test_export_with_constant_dict_values
|
|
|
|
ExportTests.test_export_with_constant_free_function
|
|
|
|
ExportTests.test_export_with_constant_free_function_and_class_method
|
|
|
|
ExportTests.test_export_with_constant_free_function_and_class_method_multiarg
|
|
|
|
ExportTests.test_export_with_constant_free_function_and_class_method_multiarg_diff
|
|
|
|
ExportTests.test_export_with_constant_list_nonzero
|
|
|
|
ExportTests.test_export_with_constant_list_nonzero_free_function
|
|
|
|
ExportTests.test_export_with_constant_method_on_module
|
|
|
|
ExportTests.test_export_with_constant_method_on_module_invoke_twice
|
|
|
|
ExportTests.test_export_with_constant_none_control_flow
|
|
|
|
ExportTests.test_export_with_constant_none_control_flow_free_func
|
|
|
|
ExportTests.test_export_with_constant_not_none_control_flow
|
|
|
|
ExportTests.test_export_with_constant_not_none_control_flow_free_func
|
|
|
|
ExportTests.test_export_with_constant_not_none_control_flow_pos
|
|
|
|
ExportTests.test_export_with_constant_not_return_const
|
|
|
|
ExportTests.test_export_with_constant_tuple_nonzero
|
|
|
|
ExportTests.test_export_with_map_cond
|
|
|
|
ExportTests.test_exported_graph_serialization
|
|
|
|
ExportTests.test_map_cond_param_buffer_lifted
|
|
|
|
ExportTests.test_torch_inference_mode_ctx
|
|
|
|
ExportTests.test_untracked_inputs_in_constraints
|
|
|
|
FakeTensorOperatorInvariants.test_like_ops
|
|
|
|
FakeTensorOperatorInvariants.test_non_kwarg_only_device
|
|
|
|
FakeTensorOperatorInvariants.test_tensor_constructors_all_have_kwarg_device
|
|
|
|
FunctionTests.test_default_dict
|
|
|
|
FunctionTests.test_default_dict_closure
|
|
|
|
FunctionTests.test_default_dict_lambda
|
|
|
|
FunctionTests.test_is_contiguous_frame_counts
|
|
|
|
FunctionTests.test_math_radians
|
|
|
|
FunctionTests.test_partials_as_input_partials_lambda
|
|
|
|
FunctionTests.test_partials_as_input_partials_mod
|
|
|
|
FuncTorchHigherOrderOpTests.test_grad
|
|
|
|
FuncTorchHigherOrderOpTests.test_grad_disable_capture
|
|
|
|
FuncTorchHigherOrderOpTests.test_grad_fn_with_kwargs
|
|
|
|
FuncTorchHigherOrderOpTests.test_grad_freevar_python_scalar
|
|
|
|
FuncTorchHigherOrderOpTests.test_grad_has_aux
|
|
|
|
FuncTorchHigherOrderOpTests.test_grad_non_tensor_input
|
|
|
|
FuncTorchHigherOrderOpTests.test_grad_pytree
|
|
|
|
FuncTorchHigherOrderOpTests.test_grad_two_tensor_all_grad_has_aux
|
|
|
|
FuncTorchHigherOrderOpTests.test_grad_two_tensor_has_aux
|
|
|
|
FuncTorchHigherOrderOpTests.test_grad_with_graph_break
|
|
|
|
FuncTorchHigherOrderOpTests.test_grad_with_side_effect
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_disable_capture
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_free_const
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_free_tensor
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_kwargs
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_multiple_invocation_in_dims
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_multiple_invocation_out_dims
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_multiple_outputs
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_multiple_outputs_diff_dims
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_multiple_outputs_out_dims_tuple
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_over_vmap_captured
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_over_vmap_two_inputs
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_previous_illegal_op_no_graph_break
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_pytree_inputs
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_recompile_same_config
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
FuncTorchHigherOrderOpTests.test_vmap_recompile_with_randomness
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
FuncTorchHigherOrderOpTests.test_vmap_side_effects
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_two_inputs
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_two_inputs_tuple_in_dims
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_with_conditional_graph_break
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_with_graph_break
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_with_graph_break_2
|
|
|
|
FuncTorchHigherOrderOpTests.test_vmap_with_graph_break_lambda
|
|
|
|
HigherOrderOpTests.test_capture_untracked_global_nested
|
|
|
|
HigherOrderOpTests.test_capture_value_created_in_subgraph
|
|
|
|
HigherOrderOpTests.test_cond_branches_no_arguments
|
|
|
|
HigherOrderOpTests.test_cond_branches_no_arguments_no_closure
|
|
|
|
HigherOrderOpTests.test_cond_pytree_operands
|
|
|
|
HigherOrderOpTests.test_cond_pytree_operands_with_non_tensor_leaves
|
|
|
|
HigherOrderOpTests.test_cond_source_fn_stack
|
|
|
|
HigherOrderOpTests.test_cond_subgraph_name_is_valid
|
|
|
|
HigherOrderOpTests.test_fallback_on_python_primitives_output
|
|
|
|
HigherOrderOpTests.test_map_lowers_to_graph
|
|
|
|
HigherOrderOpTests.test_map_multi_return
|
|
|
|
HigherOrderOpTests.test_map_pytree_return
|
|
|
|
HigherOrderOpTests.test_map_source_fn_stack
|
|
|
|
HigherOrderOpTests.test_map_subgraph_name_is_valid
|
|
|
|
HigherOrderOpTests.test_map_symint_input
|
|
|
|
HigherOrderOpTests.test_modules
|
|
|
|
HigherOrderOpTests.test_nested_tuple_output
|
|
|
|
HigherOrderOpTests.test_output_with_dict
|
|
|
|
HigherOrderOpTests.test_side_effect_in_body
|
|
|
|
HigherOrderOpTests.test_vmap_source_fn_stack
|
|
|
|
HigherOrderOpTests.test_wrap_kwarg_recompile
|
|
|
|
HigherOrderOpTests.test_wrap_subgraph_name_is_valid
|
|
|
|
HooksTests.test_functools_arg_vary
|
|
|
|
HooksTests.test_post_acc_grad_hook
|
|
|
|
HooksTests.test_tensor_register_global_hook
|
|
|
|
HooksTests.test_tensor_register_hook_in_graph_local
|
|
|
|
HooksTests.test_tensor_register_multiple_hooks
|
|
|
|
HooksTests.test_tensor_register_multiple_hooks_handles_in_list
|
|
|
|
InteropTests.test_vmap_in_graph
|
|
|
|
LoggingTests.test_aot
|
|
|
|
LoggingTests.test_aot_graphs
|
|
|
|
LoggingTests.test_aot_joint_graph
|
|
|
|
LoggingTests.test_bytecode
|
|
|
|
LoggingTests.test_custom_format
|
|
|
|
LoggingTests.test_custom_format_exc
|
|
|
|
LoggingTests.test_dynamo_debug
|
|
|
|
LoggingTests.test_dynamo_error
|
|
|
|
LoggingTests.test_dynamo_info
|
|
|
|
LoggingTests.test_graph
|
|
|
|
LoggingTests.test_graph_breaks
|
|
|
|
LoggingTests.test_graph_code
|
|
|
|
LoggingTests.test_graph_sizes
|
|
|
|
LoggingTests.test_graph_sizes_dynamic
|
|
|
|
LoggingTests.test_guards_recompiles
|
|
|
|
LoggingTests.test_inductor_debug
|
|
|
|
LoggingTests.test_inductor_error
|
|
|
|
LoggingTests.test_inductor_info
|
|
|
|
LoggingTests.test_multiline_format
|
|
|
|
LoggingTests.test_output_code
|
|
|
|
LoggingTests.test_recompiles
|
|
|
|
LoggingTests.test_trace_source_cond
|
|
|
|
LoggingTests.test_trace_source_nested
|
|
|
|
MiscTests.test_add_to_set
|
|
|
|
MiscTests.test_any_all_symnode
|
|
|
|
MiscTests.test_backend_match_guard
|
|
|
|
MiscTests.test_boolarg
|
|
|
|
MiscTests.test_callpacked
|
|
|
|
MiscTests.test_closure_out_of_scope_cell
|
|
|
|
MiscTests.test_compare_shapes_with_constant
|
|
|
|
MiscTests.test_cond_nested
|
|
|
|
MiscTests.test_cond_side_effects
|
|
|
|
MiscTests.test_dataclass_fields
|
|
|
|
MiscTests.test_deque_append_left
|
|
|
|
MiscTests.test_deque_input
|
|
|
|
MiscTests.test_deterministic_algorithms_mutated
|
|
|
|
MiscTests.test_dict_order_keys
|
|
|
|
MiscTests.test_dict_order_keys_modules
|
|
|
|
MiscTests.test_dict_order_keys_tensors
|
|
|
|
MiscTests.test_dictcomp
|
|
|
|
MiscTests.test_dtypes_no_graphbreaks
|
|
|
|
MiscTests.test_dunder_new_function_inlining
|
|
|
|
MiscTests.test_frozenset_torch_func_contains
|
|
|
|
MiscTests.test_get_cache_entry
|
|
|
|
MiscTests.test_guard_failure_fn
|
|
|
|
MiscTests.test_guard_failure_fn_shape_control
|
|
|
|
MiscTests.test_guard_failure_fn_tensor_iter
|
|
|
|
MiscTests.test_guard_failure_fn2
|
|
|
|
MiscTests.test_inference_mode
|
|
|
|
MiscTests.test_inline_closure_not_loaded_by_parent
|
|
|
|
MiscTests.test_inplace_param_update
|
|
|
|
MiscTests.test_inplace_view_on_graph_input
|
|
|
|
MiscTests.test_intermediary_tensor_grad_access
|
|
|
|
MiscTests.test_is_compiling
|
|
|
|
MiscTests.test_iter_set
|
|
|
|
MiscTests.test_itertools_accumulate_symint_default_sum
|
|
|
|
MiscTests.test_itertools_accumulate_tensors_default_sum
|
|
|
|
MiscTests.test_itertools_groupby_pure_python_default_identify_func
|
|
|
|
MiscTests.test_itertools_groupby_pure_python_key_func
|
|
|
|
MiscTests.test_itertools_infinite_cycle
|
|
|
|
MiscTests.test_itertools_infinite_repeat
|
|
|
|
MiscTests.test_itertools_infinite_repeat_mutation
|
|
|
|
MiscTests.test_itertools_repeat
|
|
|
|
MiscTests.test_listcomp
|
|
|
|
MiscTests.test_mandelbrot_numpy
|
|
|
|
MiscTests.test_mark_static
|
|
|
|
MiscTests.test_namedtuple1
|
|
|
|
MiscTests.test_namedtuple2
|
|
|
|
MiscTests.test_nan
|
|
|
|
MiscTests.test_nested_closure
|
|
|
|
MiscTests.test_nested_closure_mutation
|
|
|
|
MiscTests.test_nested_optimize
|
|
|
|
MiscTests.test_nested_optimize_decorator
|
|
|
|
MiscTests.test_nested_optimize_run
|
|
|
|
MiscTests.test_nn_module_getattr
|
|
|
|
MiscTests.test_no_raise_guard_partial_constraint
|
|
|
|
MiscTests.test_no_raise_guard_partial_constraint_across_break
|
|
|
|
MiscTests.test_numpy_array_of_arrays
|
|
|
|
MiscTests.test_numpy_force
|
|
|
|
MiscTests.test_numpy_int_constant
|
|
|
|
MiscTests.test_numpy_iter
|
|
|
|
MiscTests.test_numpy_non_torch_dtype
|
|
|
|
MiscTests.test_numpy_readonly
|
|
|
|
MiscTests.test_numpy_recompilation_scalar
|
|
|
|
MiscTests.test_numpy_size_attr
|
|
|
|
MiscTests.test_numpy_subdtype
|
|
|
|
MiscTests.test_numpy_tolist
|
|
|
|
MiscTests.test_numpy_torch_operators
|
|
|
|
MiscTests.test_numpy_with_builtin_type
|
|
|
|
MiscTests.test_out_variants_with_resizing_on_graph_inputs
|
|
|
|
MiscTests.test_pure_python_accumulate
|
|
|
|
MiscTests.test_py_guards_mark_dynamic
|
|
|
|
MiscTests.test_raise_guard_full_constraint
|
|
|
|
MiscTests.test_raise_guard_partial_constraint_no_graph_break
|
|
|
|
MiscTests.test_raise_on_backend_error
|
|
|
|
MiscTests.test_recompile_on_global_state_change
|
|
|
|
MiscTests.test_release_input_memory
|
|
|
|
MiscTests.test_release_module_memory
|
|
|
|
MiscTests.test_release_scope_memory
|
|
|
|
MiscTests.test_repeat_interleave_graphbreaks
|
|
|
|
MiscTests.test_return_nested_function
|
|
|
|
MiscTests.test_set_aliasing_recompiles
|
|
|
|
MiscTests.test_setattr_mutation1
|
|
|
|
MiscTests.test_simple_set_usage
|
|
|
|
MiscTests.test_size_input
|
|
|
|
MiscTests.test_str_format_assert2
|
|
|
|
MiscTests.test_tensor_build_list_unpack
|
|
|
|
MiscTests.test_tensor_dict1
|
|
|
|
MiscTests.test_tensor_dict2
|
|
|
|
MiscTests.test_tensor_dict3
|
|
|
|
MiscTests.test_tensor_item_capture
|
|
|
|
MiscTests.test_tolist_1d
|
|
|
|
MiscTests.test_tolist_0d
|
|
|
|
MiscTests.test_tolist_kd
|
|
|
|
MiscTests.test_tolist_kd_dynamic
|
|
|
|
MiscTests.test_tolist_scalar
|
|
|
|
MiscTests.test_torch_seed
|
|
|
|
MiscTests.test_tracing_nested_py_tree
|
|
|
|
MiscTests.test_tracing_nested_py_tree_dicts
|
|
|
|
MiscTests.test_tracing_nested_py_tree_mixed_all
|
|
|
|
MiscTests.test_tracing_nested_py_tree_tuples
|
|
|
|
MiscTests.test_tracing_py_tree
|
|
|
|
MiscTests.test_tracing_py_tree_tensor_subclass
|
|
|
|
MiscTests.test_tracing_tree_map_only
|
|
|
|
MiscTests.test_type_copy
|
|
|
|
MiscTests.test_typing_typevar
|
|
|
|
MiscTests.test_user_getattribute
|
|
|
|
MiscTests.test_with_builtin_type
|
|
|
|
MiscTests.test_yield_from
|
|
|
|
MiscTests.test_yield_gen_and_from
|
|
|
|
MiscTests.test_yield_send_to_subgenerator_graph_break
|
|
|
|
NNModuleTests.test_lazy_module_no_cls_to_become
|
|
|
|
NNModuleTests.test_lazy_module1
|
|
|
|
NNModuleTests.test_lazy_module2
|
|
|
|
NNModuleTests.test_lazy_module4
|
|
|
|
NNModuleTests.test_lazy_module5
|
|
|
|
NNModuleTests.test_lazy_module6
|
|
|
|
NNModuleTests.test_self_mutating1
|
|
|
|
NNModuleTests.test_unsupportedmethod
|
|
|
|
NNModuleTests.test_unsupportedmodule
|
|
|
|
NonStrictExportTestExport.test__scaled_dot_product_flash_attention_non_strict
|
|
|
|
NonStrictExportTestExport.test_basic_non_strict_fake_tensor_non_strict
|
|
|
|
NonStrictExportTestExport.test_basic_non_strict_real_tensor_non_strict
|
|
|
|
NonStrictExportTestExport.test_buffer_util_non_strict
|
|
|
|
NonStrictExportTestExport.test_cond_with_module_stack_export_with_non_strict
|
|
|
|
NonStrictExportTestExport.test_export_decomps_dynamic_non_strict
|
|
|
|
NonStrictExportTestExport.test_export_decomps_simple_non_strict
|
|
|
|
NonStrictExportTestExport.test_export_with_wrong_inputs_non_strict
|
|
|
|
NonStrictExportTestExport.test_external_call_non_strict_real_tensor_non_strict
|
|
|
|
NonStrictExportTestExport.test_fqn_non_strict
|
|
|
|
NonStrictExportTestExport.test_nn_module_stack_non_strict
|
|
|
|
NonStrictExportTestExport.test_nn_module_stack_shared_submodule_non_strict
|
|
|
|
NonStrictExportTestExport.test_non_strict_dynamic_shapes_non_strict
|
|
|
|
NonStrictExportTestExport.test_non_strict_dynamic_shapes_suggested_fixes_non_strict
|
|
|
|
NonStrictExportTestExport.test_param_util_non_strict
|
|
|
|
NonStrictExportTestExport.test_raise_user_error_when_guard_on_data_dependent_operation_non_strict
|
|
|
|
NonStrictExportTestExport.test_sym_sqrt_non_strict
|
|
|
|
NonStrictExportTestExport.test_to_module_with_mutated_buffer_multiple_non_strict
|
|
|
|
NonStrictExportTestExport.test_to_module_with_mutated_buffer_multiple_update_sub_later_non_strict
|
|
|
|
NonStrictExportTestExport.test_to_module_with_mutated_buffer_non_strict
|
|
|
|
NumpyTestsCPU.test_boolean_indexing_weirdness_cpu
|
|
|
|
NumpyTestsCPU.test_boolean_shape_mismatch_cpu
|
|
|
|
NumpyTestsCPU.test_empty_fancy_index_cpu
|
|
|
|
NumpyTestsCPU.test_index_no_floats_cpu
|
|
|
|
OptimizedModuleTest.test_backward_hooks
|
|
|
|
OptimizedModuleTest.test_cache_size_limit_on_guarded_nn_modules
|
|
|
|
OptimizedModuleTest.test_composition_with_opt_mod
|
|
|
|
OptimizedModuleTest.test_hooks_inner
|
|
|
|
OptimizedModuleTest.test_hooks_outer
|
|
|
|
OptimizedModuleTest.test_hooks_skip_guards
|
|
|
|
OptimizedModuleTest.test_module_dict_iter_name
|
|
|
|
OptimizedModuleTest.test_nn_module
|
|
|
|
OptimizedModuleTest.test_no_recompile_on_nn_guarded_modules
|
|
|
|
OptimizedModuleTest.test_to
|
|
|
|
OptimizerTests.test_adagrad
|
|
|
|
OptimizerTests.test_adam
|
|
|
|
OptimizerTests.test_adamax
|
|
|
|
OptimizerTests.test_adamw
|
|
|
|
OptimizerTests.test_asgd
|
|
|
|
OptimizerTests.test_nadam
|
|
|
|
OptimizerTests.test_rmsprop
|
|
|
|
OptimizerTests.test_rprop
|
|
|
|
OptimizerTests.test_sgd
|
|
|
|
PackedSequenceTest.test_pack_sequence
|
|
|
|
PackedSequenceTest.test_to
|
|
|
|
PackedSequenceTest.test_total_length
|
|
|
|
PackedSequenceTest.test_type_casts
|
|
|
|
RecompileTests.test_aliasing_guard_failures_with_globals
|
|
|
|
RecompileTests.test_automatic_dynamic_reduce_recompiles
|
|
|
|
RecompileTests.test_automatic_dynamic_tensor_scalar_change
|
|
|
|
RecompileTests.test_dynamic_shape_parameter_recompile
|
|
|
|
RecompileTests.test_recompiles_true_false_flop
|
|
|
|
RecompileUxTests.test_mismatched_type
|
|
|
|
RecompileUxTests.test_verbose_tensor_check
|
|
|
|
ReproTests.test_add_sub_alpha_out
|
|
|
|
ReproTests.test_addr_alpha_beta_out
|
|
|
|
ReproTests.test_boxes_len
|
|
|
|
ReproTests.test_chunk_reformer_ff
|
|
|
|
ReproTests.test_convert_boxes_to_pooler_format
|
|
|
|
ReproTests.test_create_rand_mask_from_inputs
|
|
|
|
ReproTests.test_do_paste_mask
|
|
|
|
ReproTests.test_dynamic_shapes_float_guard
|
|
|
|
ReproTests.test_dynamic_shapes_implicit_guard
|
|
|
|
ReproTests.test_empty_list_contains_with_jump
|
|
|
|
ReproTests.test_function_in_skipfiles
|
|
|
|
ReproTests.test_functools_wraps
|
|
|
|
ReproTests.test_hf_t5_forward
|
|
|
|
ReproTests.test_hf_xsoftmax_training
|
|
|
|
ReproTests.test_issue175
|
|
|
|
ReproTests.test_jit_trace_errors
|
|
|
|
ReproTests.test_list_aliasing
|
|
|
|
ReproTests.test_list_self_reference
|
|
|
|
ReproTests.test_longformer_chunk
|
|
|
|
ReproTests.test_merge_criteria_processor_list1
|
|
|
|
ReproTests.test_merge_criteria_processor_list2
|
|
|
|
ReproTests.test_module_in_skipfiles
|
|
|
|
ReproTests.test_multi_import
|
|
|
|
ReproTests.test_negative_shape_guard
|
|
|
|
ReproTests.test_numpy_not_ndarray_recompiles
|
|
|
|
ReproTests.test_recursive_map
|
|
|
|
ReproTests.test_reformer_eval
|
|
|
|
ReproTests.test_reformer_min_chunk_len
|
|
|
|
ReproTests.test_reformer_sorting
|
|
|
|
ReproTests.test_relative_import
|
|
|
|
ReproTests.test_relative_import_no_modulename
|
|
|
|
ReproTests.test_restricted_list_subclass1
|
|
|
|
ReproTests.test_rewrite_assert_noop
|
|
|
|
ReproTests.test_rewrite_assert_with_msg
|
|
|
|
ReproTests.test_rewrite_assert_with_non_string_msg
|
|
|
|
ReproTests.test_seq_append_list
|
|
|
|
ReproTests.test_size_typematch
|
|
|
|
ReproTests.test_tensor_data_kwarg
|
|
|
|
ReproTests.test_threading_local
|
|
|
|
ReproTests.test_validate_model_kwargs
|
|
|
|
RetraceExportTestDynamismExpression.test_export_inline_constraints_retraceability
|
|
|
|
RetraceExportTestExport.test_cond_with_module_stack_export_with_retraceability
|
|
|
|
RetraceExportTestExport.test_constrain_size_in_eager_retraceability
|
|
|
|
RetraceExportTestExport.test_constrain_size_with_constrain_value_retraceability
|
|
|
|
RetraceExportTestExport.test_constrain_size_with_various_cases_retraceability
|
|
|
|
RetraceExportTestExport.test_nn_module_stack_retraceability
|
|
|
|
RetraceExportTestExport.test_nn_module_stack_shared_submodule_retraceability
|
|
|
|
RetraceExportTestExport.test_non_strict_dynamic_shapes_retraceability
|
|
|
|
RetraceExportTestExport.test_non_strict_dynamic_shapes_suggested_fixes_retraceability
|
|
|
|
SerDesExportTestDynamismExpression.test_export_inline_constraints_serdes
|
|
|
|
SerDesExportTestExport.test_basic_non_strict_fake_tensor_serdes
|
|
|
|
SerDesExportTestExport.test_basic_non_strict_real_tensor_serdes
|
|
|
|
SerDesExportTestExport.test_cond_with_module_stack_export_with_serdes
|
|
|
|
SerDesExportTestExport.test_constrain_size_in_eager_serdes
|
|
|
|
SerDesExportTestExport.test_constrain_size_with_constrain_value_serdes
|
|
|
|
SerDesExportTestExport.test_constrain_size_with_various_cases_serdes
|
|
|
|
SerDesExportTestExport.test_external_call_non_strict_real_tensor_serdes
|
|
|
|
SerDesExportTestExport.test_nn_module_stack_serdes
|
|
|
|
SerDesExportTestExport.test_nn_module_stack_shared_submodule_serdes
|
|
|
|
SerDesExportTestExport.test_non_strict_dynamic_shapes_serdes
|
|
|
|
SerDesExportTestExport.test_non_strict_dynamic_shapes_suggested_fixes_serdes
|
|
|
|
SkipNonTensorTests.test_add_tensor_dict
|
|
|
|
SkipNonTensorTests.test_add_tensor_list
|
|
|
|
SkipNonTensorTests.test_add_tensor1
|
|
|
|
SkipNonTensorTests.test_add_tensor2
|
|
|
|
SkipNonTensorTests.test_do_not_skip_side_effects
|
|
|
|
SkipNonTensorTests.test_recursive_list
|
|
|
|
SubclassTests.test_compile_higher_order_with_functionalization
|
|
|
|
SubclassTests.test_compile_with_functionalization
|
|
|
|
SubclassTests.test_torch_function_state_guards
|
|
|
|
SubclassTests.test_wrapper_subclass_guards_on_inner_tensor
|
|
|
|
SubclassTests.test_wrapper_subclass_with_differently_sized_inner_tensor
|
|
|
|
SubclassTests.test_wrapper_subclass_with_same_sized_inner_tensor
|
|
|
|
SubGraphTests.test_dynamic_duck_size
|
|
|
|
SubGraphTests.test_dynamic_getitem
|
|
|
|
SubGraphTests.test_dynamic_kwarg
|
|
|
|
SubGraphTests.test_dynamic_order_dependence
|
|
|
|
SubGraphTests.test_dynamic_zero_inference
|
|
|
|
SubGraphTests.test_enumerate_not_break_graph
|
|
|
|
SubGraphTests.test_no_graph_break_on_item
|
|
|
|
TestAnalyze.test_trace_dependencies
|
|
|
|
TestAOTAutograd.test_default_partitioner_saves_symints_not_tensors_for_bw
|
|
|
|
TestAOTAutograd.test_dupe_arg_torture
|
|
|
|
TestAOTAutograd.test_input_aliased_with_mutation_output_alias
|
|
|
|
TestAOTAutograd.test_input_data_and_metadata_mutation
|
|
|
|
TestAOTAutograd.test_input_data_and_metadata_mutation_aliases_other_input
|
|
|
|
TestAOTAutograd.test_input_mutation_alias_everything
|
|
|
|
TestAOTAutograd.test_input_mutation_aliases_and_none_require_gradients
|
|
|
|
TestAOTAutograd.test_input_mutation_aliases_and_output_alias
|
|
|
|
TestAOTAutograd.test_input_mutation_aliases_bases_out_of_order
|
|
|
|
TestAOTAutograd.test_input_mutation_and_output_view
|
|
|
|
TestAOTAutograd.test_input_mutation_false_aliasing
|
|
|
|
TestAOTAutograd.test_input_mutation_is_output
|
|
|
|
TestAOTAutograd.test_input_mutation_metadata
|
|
|
|
TestAOTAutograd.test_input_mutation_noncontiguous
|
|
|
|
TestAOTAutograd.test_input_mutation_output_view_multiple
|
|
|
|
TestAOTAutograd.test_input_mutation_set__input_mutation
|
|
|
|
TestAOTAutograd.test_input_mutation_set__nop
|
|
|
|
TestAOTAutograd.test_input_output_view_metadata_mutate_multiple
|
|
|
|
TestAOTAutograd.test_input_output_view_mutate_multiple
|
|
|
|
TestAOTAutograd.test_input_output_view_simple
|
|
|
|
TestAOTAutograd.test_invalid_dupe
|
|
|
|
TestAOTAutograd.test_invalid_dupe_fake
|
|
|
|
TestAOTAutograd.test_invalid_dupe_left_bias
|
|
|
|
TestAOTAutograd.test_new_inp_requires_grad_now
|
|
|
|
TestAOTAutograd.test_output_aliases_input_multi_output_view
|
|
|
|
TestAOTAutograd.test_output_aliases_intermediate_and_returned
|
|
|
|
TestAOTAutograd.test_output_aliases_intermediate_and_returned_different_grad
|
|
|
|
TestAOTAutograd.test_output_aliases_intermediate_and_returned_flipped
|
|
|
|
TestAOTAutograd.test_output_aliases_intermediate_multi_output_view
|
|
|
|
TestAOTAutograd.test_output_aliases_intermediate_multiple
|
|
|
|
TestAOTAutograd.test_output_aliases_intermediate_multiple_mixed
|
|
|
|
TestAOTAutograd.test_output_aliases_intermediate_returned_multiple_times
|
|
|
|
TestAOTAutograd.test_output_aliases_multiple_inputs_get_correct_one
|
|
|
|
TestAOTAutograd.test_output_all_alias_types
|
|
|
|
TestAOTAutograd.test_output_dict
|
|
|
|
TestAOTAutograd.test_output_op_depending_on_symint
|
|
|
|
TestAOTAutograd.test_outputs_are_aliased
|
|
|
|
TestAOTAutograd.test_set__and_data_mutation_bad
|
|
|
|
TestAOTAutograd.test_set__and_data_mutation_good
|
|
|
|
TestAOTAutograd.test_some_output_requires_grad_input_doesnt
|
|
|
|
TestAOTAutograd.test_some_outputs_dont_require_grad_non_view
|
|
|
|
TestAOTAutograd.test_some_outputs_dont_require_grad_view
|
|
|
|
TestAOTAutograd.test_view_and_inplace_view
|
|
|
|
TestAOTAutograd.test_view_detach
|
|
|
|
TestAOTDispatch.test_aot_dispatch_input_mutation
|
|
|
|
TestAOTDispatch.test_aot_dispatch_input_mutation_and_output_alias
|
|
|
|
TestAOTDispatch.test_aot_dispatch_output_alias
|
|
|
|
TestAOTDispatch.test_aot_dispatch_simple
|
|
|
|
TestApplyAlongAxis.test_0d_array
|
|
|
|
TestApplyAlongAxis.test_axis_insertion
|
|
|
|
TestApplyAlongAxis.test_scalar_array
|
|
|
|
TestArange.test_infinite
|
|
|
|
TestArange.test_nan_step
|
|
|
|
TestArange.test_require_range
|
|
|
|
TestArange.test_zero_step
|
|
|
|
TestArgmax.test_combinations_data62
|
|
|
|
TestArgmax.test_combinations_data63
|
|
|
|
TestArgmax.test_combinations_data65
|
|
|
|
TestArgmax.test_combinations_data66
|
|
|
|
TestArrayAttributeDeletion.test_multiarray_not_writable_attributes_deletion
|
|
|
|
TestArrayAttributeDeletion.test_multiarray_writable_attributes_deletion
|
|
|
|
TestArrayConstruction.test_array_empty
|
|
|
|
TestArrayCreationCopyArgument.test_scalars
|
|
|
|
TestArrayCreationCopyArgument.test_striding_not_ok
|
|
|
|
TestArrayFromScalar.test_integers_np_byte_np_longlong
|
|
|
|
TestArrayFromScalar.test_integers_np_int__np_longlong
|
|
|
|
TestArrayFromScalar.test_integers_np_intc_np_longlong
|
|
|
|
TestArrayFromScalar.test_integers_np_longlong_np_longlong
|
|
|
|
TestArrayFromScalar.test_integers_np_longlong_t26
|
|
|
|
TestArrayFromScalar.test_integers_np_short_np_longlong
|
|
|
|
TestArrayFromScalar.test_integers_t15_np_longlong
|
|
|
|
TestArraySplit.test_integer_0_split
|
|
|
|
TestAsArrayCPU.test_copy_list_cpu_bfloat16
|
|
|
|
TestAsArrayCPU.test_copy_list_cpu_bool
|
|
|
|
TestAsArrayCPU.test_copy_list_cpu_complex64
|
|
|
|
TestAsArrayCPU.test_copy_list_cpu_complex128
|
|
|
|
TestAsArrayCPU.test_copy_list_cpu_float16
|
|
|
|
TestAsArrayCPU.test_copy_list_cpu_float32
|
|
|
|
TestAsArrayCPU.test_copy_list_cpu_float64
|
|
|
|
TestAsArrayCPU.test_copy_list_cpu_int8
|
|
|
|
TestAsArrayCPU.test_copy_list_cpu_int16
|
|
|
|
TestAsArrayCPU.test_copy_list_cpu_int32
|
|
|
|
TestAsArrayCPU.test_copy_list_cpu_int64
|
|
|
|
TestAsArrayCPU.test_copy_list_cpu_uint8
|
|
|
|
TestAssignment.test_assignment_broadcasting
|
|
|
|
TestAttributes.test_fill_max_uint64
|
|
|
|
TestAttributes.test_fill_readonly
|
|
|
|
TestAttributes.test_fill_struct_array
|
|
|
|
TestAttributes.test_set_stridesattr
|
|
|
|
TestAttributes.test_stridesattr
|
|
|
|
TestAutograd.test_access_saved_tensor_twice_without_recomputation_works
|
|
|
|
TestAutograd.test_anomaly_mode_no_check_nan
|
|
|
|
TestAutograd.test_autograd_simple_views_python
|
|
|
|
TestAutograd.test_autograd_views_codegen
|
|
|
|
TestAutograd.test_backward_with_inputs
|
|
|
|
TestAutograd.test_checkpoint_detects_non_determinism
|
|
|
|
TestAutograd.test_checkpointing_non_reentrant_autocast_cpu
|
|
|
|
TestAutograd.test_checkpointing_without_reentrant_custom_function_works
|
|
|
|
TestAutograd.test_checkpointing_without_reentrant_with_context_fn
|
|
|
|
TestAutograd.test_custom_autograd_no_early_free
|
|
|
|
TestAutograd.test_custom_autograd_repeated_grad_grad
|
|
|
|
TestAutograd.test_custom_function_cycle
|
|
|
|
TestAutograd.test_custom_function_forward_mode_forward_is_no_op
|
|
|
|
TestAutograd.test_custom_function_forward_mode_inplace_checks
|
|
|
|
TestAutograd.test_custom_function_forward_mode_view_checks
|
|
|
|
TestAutograd.test_custom_function_forward_mode_wrong_formula
|
|
|
|
TestAutograd.test_custom_function_non_tensor_inputs_outputs
|
|
|
|
TestAutograd.test_custom_function_save_for_forward
|
|
|
|
TestAutograd.test_custom_function_saved_tensors
|
|
|
|
TestAutograd.test_gradcheck_backward_mul_by_grad_output
|
|
|
|
TestAutograd.test_gradcheck_check_forward_or_backward_only
|
|
|
|
TestAutograd.test_gradcheck_forward_ad
|
|
|
|
TestAutograd.test_gradcheck_forward_ad_batched_grad
|
|
|
|
TestAutograd.test_gradcheck_nondeterministic
|
|
|
|
TestAutograd.test_hook_closure_cycle_use_custom_function_True_use_tensor_hook_False
|
|
|
|
TestAutograd.test_hook_none
|
|
|
|
TestAutograd.test_nested_anomaly_detect_nan
|
|
|
|
TestAutograd.test_nested_anomaly_printstack_cleanup
|
|
|
|
TestAutograd.test_profiler
|
|
|
|
TestAutograd.test_profiler_aggregation_table
|
|
|
|
TestAutograd.test_profiler_propagation
|
|
|
|
TestAutograd.test_profiler_seq_nr
|
|
|
|
TestAutograd.test_profiler_shapes
|
|
|
|
TestAutograd.test_record_function
|
|
|
|
TestAutograd.test_record_function_callbacks
|
|
|
|
TestAutograd.test_record_function_legacy
|
|
|
|
TestAutograd.test_return_duplicate
|
|
|
|
TestAutograd.test_return_duplicate_inplace
|
|
|
|
TestAutograd.test_save_on_cpu_and_checkpoint
|
|
|
|
TestAutograd.test_saved_tensor_hooks_custom_function_intermediates
|
|
|
|
TestAutograd.test_set_grad_coroutines
|
|
|
|
TestAutograd.test_set_grad_coroutines_exit
|
|
|
|
TestAutograd.test_set_grad_generator_functions
|
|
|
|
TestAutograd.test_set_grad_generator_functions_recursive
|
|
|
|
TestAutograd.test_sparse_mm_backward
|
|
|
|
TestAutogradDeviceTypeCPU.test_inplace_on_view_modify_base_cpu
|
|
|
|
TestAutogradDeviceTypeCPU.test_inplace_on_view_python_cpu
|
|
|
|
TestAutogradForwardMode.test_create_new_zeros_with_same_meta
|
|
|
|
TestAutogradForwardMode.test_detach_view_tracking
|
|
|
|
TestAutogradForwardMode.test_forward_level_cleanup
|
|
|
|
TestAutogradFunctionCPU.test_once_differentiable_autograd_vjp_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestAutogradInferenceMode.test_inference_mode_decorator
|
|
|
|
TestAutogradInferenceMode.test_inference_mode_inf_tensor_in_inf_mode_inplace_op
|
|
|
|
TestAutogradLogging.test_logging
|
|
|
|
TestBaseMath.test_lower_align
|
|
|
|
TestBaseSparsifier.test_state_dict
|
|
|
|
TestBaseStructuredSparsifier.test_constructor
|
|
|
|
TestBaseStructuredSparsifier.test_prepare_conv2d
|
|
|
|
TestBaseStructuredSparsifier.test_prepare_linear
|
|
|
|
TestBaseStructuredSparsifier.test_step_linear
|
|
|
|
TestBincount.test_with_incorrect_minlength
|
|
|
|
TestBitsCPU.test_cat_cpu
|
|
|
|
TestBool.test_sum_2
|
|
|
|
TestBooleanIndexing.test_bool_as_int_argument_errors
|
|
|
|
TestBooleanIndexing.test_boolean_indexing_weirdness
|
|
|
|
TestBroadcast.test_broadcast_error_kwargs
|
|
|
|
TestBroadcast.test_broadcast_single_arg
|
|
|
|
TestBufferProtocolCPU.test_byte_to_int_cpu
|
|
|
|
TestClip.test_clip_func_takes_out
|
|
|
|
TestClip.test_simple_complex
|
|
|
|
TestClip.test_simple_int32_inout_casting_unsafe
|
|
|
|
TestColumnStack.test_non_iterable
|
|
|
|
TestCommonPass.test_correctness_CSEPass_Mutation_cpu
|
|
|
|
TestCommonPass.test_correctness_CSEPass_MutationInput_cpu
|
|
|
|
TestCommonPass.test_correctness_CSEPass_MutationMetadata_cpu
|
|
|
|
TestCommonPass.test_correctness_CSEPass_MutationTorchTensorCall_cpu
|
|
|
|
TestCommonPass.test_correctness_CSEPass_ReturnList_cpu
|
|
|
|
TestCommonPass.test_correctness_CSEPass_TakeList_cpu
|
|
|
|
TestCommonPass.test_correctness_factory_CSEPass_FactoryFunctionCall_cpu
|
|
|
|
TestCommonPass.test_correctness_factory_CSEPass_MutationFactory_cpu
|
|
|
|
TestCompileTransformsCPU.test_compile_vmap_hessian_cpu
|
|
|
|
TestComposability.test_convert_without_squash_mask
|
|
|
|
TestComposability.test_fusion_before_s_prep
|
|
|
|
TestComposability.test_q_prep_before_s_prep
|
|
|
|
TestComposability.test_qat_prep_before_s_prep
|
|
|
|
TestComposability.test_s_prep_before_fusion
|
|
|
|
TestComposability.test_s_prep_before_q_prep
|
|
|
|
TestComposability.test_s_prep_before_qat_prep
|
|
|
|
TestComposabilityCPU.test_make_fx_vjp_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestComposabilityCPU.test_vjp_grad_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestComposabilityCPU.test_vjp_vjp_cpu
|
|
|
|
TestComposabilityCPU.test_vjp_vmap_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestComposabilityCPU.test_vmap_vjp_cpu
|
|
|
|
TestConcatenate.test_bad_out_shape
|
|
|
|
TestConcatenate.test_concatenate
|
|
|
|
TestConcatenate.test_exceptions
|
|
|
|
TestConcatenate.test_large_concatenate_axis_None
|
|
|
|
TestCond.test_empty_sq_cases
|
|
|
|
TestCond.test_sq_cases
|
|
|
|
TestConstant.test_check_constant_float3
|
|
|
|
TestConstant.test_check_constant_pad_2d
|
|
|
|
TestContentStoreCPU.test_repeated_hash_cpu
|
|
|
|
TestControlFlow.test_map_autograd_nested_list
|
|
|
|
TestControlFlow.test_map_autograd_no_grad_output
|
|
|
|
TestControlFlow.test_map_dict_in_out
|
|
|
|
TestControlFlow.test_map_list_in_out
|
|
|
|
TestControlFlowTraced.test_map_functionalized
|
|
|
|
TestControlFlowTraced.test_map_functionalized_aot_func
|
|
|
|
TestControlFlowTraced.test_nested_cond_map_cond_symbolic
|
|
|
|
TestControlFlowTraced.test_nested_map_cond_real
|
|
|
|
TestControlFlowTraced.test_nested_map_cond_symbolic
|
|
|
|
TestControlFlowTraced.test_tracing_map_autograd_aot_functionalized
|
|
|
|
TestControlFlowTraced.test_tracing_map_autograd_symbolic_dict
|
|
|
|
TestControlFlowTraced.test_tracing_map_autograd_symbolic_list
|
|
|
|
TestControlFlowTraced.test_tracing_map_autograd_symbolic_simple
|
|
|
|
TestControlFlowTraced.test_tracing_map_real
|
|
|
|
TestControlFlowTraced.test_tracing_map_symbolic_dict
|
|
|
|
TestControlFlowTraced.test_tracing_map_symbolic_list
|
|
|
|
TestControlFlowTraced.test_tracing_map_symbolic_simple
|
|
|
|
TestConversion.test_to_int_scalar
|
|
|
|
TestConvolutionNN.test_Conv2d_missing_argument
|
|
|
|
TestCov.test_aweights
|
|
|
|
TestCov.test_fweights
|
|
|
|
TestCppExtensionJIT.test_cpp_frontend_module_has_same_output_as_python
|
|
|
|
TestCppExtensionJIT.test_cpp_frontend_module_has_up_to_date_attribute
|
|
|
|
TestCppExtensionJIT.test_cpp_frontend_module_has_up_to_date_attributes
|
|
|
|
TestCppExtensionOpenRgistration.test_open_device_registration
|
|
|
|
TestCross.test_broadcasting_shapes
|
|
|
|
TestCustomBackendAPI.test_aot_autograd_api
|
|
|
|
TestCustomBackendAPI.test_lookup_backend
|
|
|
|
TestCustomBackendAPI.test_register_backend_api
|
|
|
|
TestCustomOp.test_impl_device_cpu
|
|
|
|
TestCustomOp.test_impl_device_function
|
|
|
|
TestCustomOp.test_legacy_define
|
|
|
|
TestCustomOpTestingCPU.test_missing_functionalization_cpu
|
|
|
|
TestDelete.test_slices
|
|
|
|
TestDeserialize.test_tensor_tensor_list
|
|
|
|
TestDiag.test_failure
|
|
|
|
TestDiff.test_append
|
|
|
|
TestDiff.test_axis
|
|
|
|
TestDiff.test_n
|
|
|
|
TestDiff.test_prepend
|
|
|
|
TestDistributionShapes.test_mixture_same_family_shape
|
|
|
|
TestDLPack.test_dlpack_device
|
|
|
|
TestDLPack.test_dtype_passthrough_dtype0
|
|
|
|
TestDLPack.test_dtype_passthrough_dtype1
|
|
|
|
TestDLPack.test_dtype_passthrough_dtype2
|
|
|
|
TestDLPack.test_dtype_passthrough_dtype3
|
|
|
|
TestDLPack.test_dtype_passthrough_dtype4
|
|
|
|
TestDLPack.test_dtype_passthrough_dtype5
|
|
|
|
TestDLPack.test_dtype_passthrough_dtype6
|
|
|
|
TestDLPack.test_dtype_passthrough_dtype7
|
|
|
|
TestDLPack.test_dtype_passthrough_dtype8
|
|
|
|
TestDLPack.test_dtype_passthrough_dtype9
|
|
|
|
TestDLPack.test_dunder_dlpack_refcount
|
|
|
|
TestDLPack.test_from_dlpack_refcount
|
|
|
|
TestDLPack.test_from_torch
|
|
|
|
TestDLPack.test_higher_dims_ndim_0
|
|
|
|
TestDLPack.test_higher_dims_ndim_1
|
|
|
|
TestDLPack.test_higher_dims_ndim_2
|
|
|
|
TestDLPack.test_higher_dims_ndim_3
|
|
|
|
TestDLPack.test_higher_dims_ndim_4
|
|
|
|
TestDLPack.test_higher_dims_ndim_5
|
|
|
|
TestDLPack.test_higher_dims_ndim_6
|
|
|
|
TestDLPack.test_higher_dims_ndim_7
|
|
|
|
TestDLPack.test_higher_dims_ndim_8
|
|
|
|
TestDLPack.test_higher_dims_ndim_9
|
|
|
|
TestDLPack.test_higher_dims_ndim_10
|
|
|
|
TestDLPack.test_higher_dims_ndim_11
|
|
|
|
TestDLPack.test_higher_dims_ndim_12
|
|
|
|
TestDLPack.test_higher_dims_ndim_13
|
|
|
|
TestDLPack.test_higher_dims_ndim_14
|
|
|
|
TestDLPack.test_higher_dims_ndim_15
|
|
|
|
TestDLPack.test_higher_dims_ndim_16
|
|
|
|
TestDLPack.test_higher_dims_ndim_17
|
|
|
|
TestDLPack.test_higher_dims_ndim_18
|
|
|
|
TestDLPack.test_higher_dims_ndim_19
|
|
|
|
TestDLPack.test_higher_dims_ndim_20
|
|
|
|
TestDLPack.test_higher_dims_ndim_21
|
|
|
|
TestDLPack.test_higher_dims_ndim_22
|
|
|
|
TestDLPack.test_higher_dims_ndim_23
|
|
|
|
TestDLPack.test_higher_dims_ndim_24
|
|
|
|
TestDLPack.test_higher_dims_ndim_25
|
|
|
|
TestDLPack.test_higher_dims_ndim_26
|
|
|
|
TestDLPack.test_higher_dims_ndim_27
|
|
|
|
TestDLPack.test_higher_dims_ndim_28
|
|
|
|
TestDLPack.test_higher_dims_ndim_29
|
|
|
|
TestDLPack.test_higher_dims_ndim_30
|
|
|
|
TestDLPack.test_higher_dims_ndim_31
|
|
|
|
TestDLPack.test_higher_dims_ndim_32
|
|
|
|
TestDLPack.test_ndim0
|
|
|
|
TestDLPack.test_non_contiguous
|
|
|
|
TestDropoutNN.test_invalid_dropout_p
|
|
|
|
TestDsplit.test_1D_array
|
|
|
|
TestDsplit.test_2D_array
|
|
|
|
TestDsplit.test_0D_array
|
|
|
|
TestDsplit.test_non_iterable
|
|
|
|
TestDstack.test_non_iterable
|
|
|
|
TestDynamicQuantizedOps.test_qlinear
|
|
|
|
TestDynamismExpression.test_export_inline_constraints
|
|
|
|
TestEigh.test_invalid
|
|
|
|
TestEigvalsh.test_invalid
|
|
|
|
TestExamplesCorrectnessCPU.test_resnet18_per_sample_grads_mechanism_functional_call_cpu
|
|
|
|
TestExamplesCorrectnessCPU.test_resnet18_per_sample_grads_mechanism_make_functional_cpu
|
|
|
|
TestExpandDims.test_axis_out_of_range
|
|
|
|
TestExpandDims.test_repeated_axis
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_conv1d_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_conv2d_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_conv3d_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_group_norm_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_instance_norm_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_layer_norm_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_conv1d_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_conv2d_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_conv3d_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_group_norm_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_instance_norm_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_layer_norm_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_conv1d_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_conv2d_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_conv3d_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_group_norm_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_instance_norm_cpu_float64
|
|
|
|
TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_layer_norm_cpu_float64
|
|
|
|
TestExperiment.test_with_buffer_as_submodule
|
|
|
|
TestExperimentalUtils.test_profiler_name_pattern
|
|
|
|
TestExperimentalUtils.test_profiler_optimizer_single_tensor_pattern
|
|
|
|
TestExperimentalUtils.test_profiler_pattern_match_helper
|
|
|
|
TestExperimentalUtils.test_profiler_pattern_matcher_json_report
|
|
|
|
TestExperimentalUtils.test_profiler_synchronized_dataloader_pattern
|
|
|
|
TestExperimentalUtils.test_utils_compute_queue_depth_when_no_cuda_events
|
|
|
|
TestExperimentalUtils.test_utils_compute_self_time
|
|
|
|
TestExport.test_basic_non_strict_fake_tensor
|
|
|
|
TestExport.test_basic_non_strict_real_tensor
|
|
|
|
TestExport.test_cond_with_module_stack_export_with
|
|
|
|
TestExport.test_constrain_size_in_eager
|
|
|
|
TestExport.test_constrain_size_with_constrain_value
|
|
|
|
TestExport.test_constrain_size_with_various_cases
|
|
|
|
TestExport.test_external_call_non_strict_real_tensor
|
|
|
|
TestExport.test_nn_module_stack
|
|
|
|
TestExport.test_nn_module_stack_shared_submodule
|
|
|
|
TestExport.test_non_strict_dynamic_shapes
|
|
|
|
TestExport.test_non_strict_dynamic_shapes_suggested_fixes
|
|
|
|
TestFakeQuantizeOps.test_backward_per_channel
|
|
|
|
TestFakeQuantizeOps.test_backward_per_tensor
|
|
|
|
TestFakeQuantizeOps.test_learnable_backward_per_channel_cpu
|
|
|
|
TestFakeQuantizeOps.test_learnable_backward_per_channel_cuda
|
|
|
|
TestFakeQuantizeOps.test_learnable_backward_per_tensor_cuda
|
|
|
|
TestFakeQuantizeOps.test_learnable_forward_per_channel_cuda
|
|
|
|
TestFakeQuantizeOps.test_learnable_forward_per_tensor_cpu
|
|
|
|
TestFakeQuantizeOps.test_learnable_forward_per_tensor_cuda
|
|
|
|
TestFakeSparsity.test_jit_trace
|
|
|
|
TestFakeSparsity.test_masking_logic
|
|
|
|
TestFlag.test_writeable_from_readonly
|
|
|
|
TestFlip.test_axes
|
|
|
|
TestFliplr.test_basic
|
|
|
|
TestFloatNonIntegerArgument.test_non_integer_argument_errors
|
|
|
|
TestFloatNonIntegerArgument.test_reduce_axis_float_index
|
|
|
|
TestFromBuffer.test_basic_little_dtype0
|
|
|
|
TestFromBuffer.test_basic_little_dtype1
|
|
|
|
TestFromBuffer.test_basic_little_dtype2
|
|
|
|
TestFromBuffer.test_empty
|
|
|
|
TestFunctionalizeCPU.test_multioutput_view_cpu
|
|
|
|
TestFunctionalizeCPU.test_simple_view_cpu
|
|
|
|
TestFunctors.test_cat_transform
|
|
|
|
TestFunctors.test_cat_transform_non_uniform
|
|
|
|
TestFX.test_custom_traceback_raised_when_exception_source_is_graphmodule
|
|
|
|
TestFX.test_immutable_dict_pytree_ops
|
|
|
|
TestFX.test_immutable_list_pytree_ops
|
|
|
|
TestFX.test_pytree_concrete
|
|
|
|
TestFXAPIBackwardCompatibility.test_public_api_surface
|
|
|
|
TestFxComposability.test_q_prep_fx_before_s_prep
|
|
|
|
TestFxComposability.test_q_prep_fx_s_prep_ref_conv
|
|
|
|
TestFxComposability.test_s_prep_before_q_prep_fx
|
|
|
|
TestFxComposability.test_s_prep_before_qat_prep_fx
|
|
|
|
TestFxComposability.test_s_prep_q_prep_fx_ref
|
|
|
|
TestFXGraphMatcherModels.test_mobilenet_v2_qat
|
|
|
|
TestGenerateNumericDebugHandle.test_quantize_pt2e_preserve_handle
|
|
|
|
TestGenerateOpcheckTests.test_opcheck_bad_op
|
|
|
|
TestGenericCumSumProd.test_bad_axis_func0
|
|
|
|
TestGenericCumSumProd.test_bad_axis_func1
|
|
|
|
TestGenericPytree.test_flatten_unflatten_deque_cxx
|
|
|
|
TestGenericPytree.test_flatten_unflatten_deque_py
|
|
|
|
TestGenericReductions.test_bad_axis_func0
|
|
|
|
TestGenericReductions.test_bad_axis_func1
|
|
|
|
TestGenericReductions.test_bad_axis_func2
|
|
|
|
TestGenericReductions.test_bad_axis_func3
|
|
|
|
TestGenericReductions.test_bad_axis_func4
|
|
|
|
TestGenericReductions.test_bad_axis_func5
|
|
|
|
TestGenericReductions.test_bad_axis_func6
|
|
|
|
TestGenericReductions.test_bad_axis_func7
|
|
|
|
TestGenericReductions.test_bad_axis_func8
|
|
|
|
TestGenericReductions.test_bad_axis_func9
|
|
|
|
TestGenericReductions.test_bad_axis_func10
|
|
|
|
TestGenericReductions.test_bad_axis_func11
|
|
|
|
TestGradient.test_badargs
|
|
|
|
TestGradient.test_second_order_accurate
|
|
|
|
TestGradient.test_specific_axes
|
|
|
|
TestGradient.test_values
|
|
|
|
TestGradNewOnesOverride.test_newones
|
|
|
|
TestGradTransformCPU.test_grad_of_vjp_composition_cpu
|
|
|
|
TestGradTransformCPU.test_no_grad_outside_vjp_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestGradTransformCPU.test_no_grad_outside_vjp_fn_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestGradTransformCPU.test_no_grad_outside_vjp_only_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestGradTransformCPU.test_unrelated_vjp_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestGradTransformCPU.test_unrelated_vjp_multiple_inputs_outputs_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestGradTransformCPU.test_vjp_aux_pytree_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestGradTransformCPU.test_vjp_cpu
|
|
|
|
TestGradTransformCPU.test_vjp_of_grad_composition_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestGradTransformCPU.test_vjp_pytree_input_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestGradTransformCPU.test_vjp_pytree_output_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestGradTransformCPU.test_vjp_two_outputs_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestGraph.test_traverse_circular_datapipe
|
|
|
|
TestGraph.test_traverse_unhashable_datapipe
|
|
|
|
TestHistogram.test_error_binnum_type
|
|
|
|
TestHistogram.test_finite_range
|
|
|
|
TestHistogram.test_one_bin
|
|
|
|
TestHistogram.test_precision
|
|
|
|
TestHistogram.test_unsigned_monotonicity_check
|
|
|
|
TestHistogram2d.test_all_outliers
|
|
|
|
TestHistogram2d.test_asym
|
|
|
|
TestHistogram2d.test_binparameter_combination
|
|
|
|
TestHistogramdd.test_bins_error_2
|
|
|
|
TestHistogramdd.test_bins_errors
|
|
|
|
TestHistogramdd.test_equal_edges
|
|
|
|
TestHistogramdd.test_finite_range
|
|
|
|
TestHistogramdd.test_inf_edges
|
|
|
|
TestHistogramdd.test_simple
|
|
|
|
TestHistogramdd.test_weights
|
|
|
|
TestHsplit.test_0D_array
|
|
|
|
TestHsplit.test_non_iterable
|
|
|
|
TestHstack.test_empty_input
|
|
|
|
TestHstack.test_non_iterable
|
|
|
|
TestImports.test_circular_dependencies
|
|
|
|
TestIndexing.test_index_no_floats
|
|
|
|
TestIndexing.test_slicing_no_floats
|
|
|
|
TestIndexingCPU.test_empty_ndim_index_bool_cpu
|
|
|
|
TestIndexingCPU.test_index_cpu
|
|
|
|
TestIndexingCPU.test_index_limits_cpu
|
|
|
|
TestIndexingCPU.test_invalid_index_cpu
|
|
|
|
TestIndexingCPU.test_out_of_bound_index_cpu
|
|
|
|
TestIndexingCPU.test_zero_dim_index_cpu
|
|
|
|
TestInputAttrTracking.test_complex_attr_access_without_graph_breaks
|
|
|
|
TestInputAttrTracking.test_const_property_on_tensor
|
|
|
|
TestInputAttrTracking.test_set_data_on_input_tensor
|
|
|
|
TestInputAttrTracking.test_tensor_property_assigned_on_tensor
|
|
|
|
TestInputAttrTracking.test_tensor_property_on_tensor
|
|
|
|
TestIsScalar.test_is_not_scalar_value6
|
|
|
|
TestIterator.test_iterator
|
|
|
|
TestIterDataPipeGraphFastForward.test_simple_snapshot_custom_non_generator
|
|
|
|
TestIterDataPipeGraphFastForward.test_simple_snapshot_custom_self_next
|
|
|
|
TestJacCPU.test_against_reference_correctness_different_devices_cpu
|
|
|
|
TestJacCPU.test_against_reference_default_arg_cpu
|
|
|
|
TestJacCPU.test_against_reference_multi_input_cpu
|
|
|
|
TestJacCPU.test_against_reference_multi_input_multi_output_cpu
|
|
|
|
TestJacCPU.test_against_reference_simple_cpu
|
|
|
|
TestJacCPU.test_against_reference_unrelated_outputs_cpu
|
|
|
|
TestJacCPU.test_against_reference_zero_dim_cpu
|
|
|
|
TestJacCPU.test_argnums_defaults_to_zero_cpu
|
|
|
|
TestJacCPU.test_argnums_effect_on_return_jacfwd_cpu
|
|
|
|
TestJacCPU.test_argnums_effect_on_return_jacrev_cpu
|
|
|
|
TestJacCPU.test_argnums_tuple_jacfwd_cpu
|
|
|
|
TestJacCPU.test_argnums_tuple_jacrev_cpu
|
|
|
|
TestJacCPU.test_aux_pytree_cpu
|
|
|
|
TestJacCPU.test_dimensionality_cpu
|
|
|
|
TestJacCPU.test_empty_argnums_jacfwd_cpu
|
|
|
|
TestJacCPU.test_empty_output_cpu
|
|
|
|
TestJacCPU.test_float_argnums_jacfwd_cpu
|
|
|
|
TestJacCPU.test_inplace_cpu
|
|
|
|
TestJacCPU.test_jac_with_non_tensor_args_cpu
|
|
|
|
TestJacCPU.test_multiple_args_jacfwd_cpu
|
|
|
|
TestJacCPU.test_multiple_args_jacrev_cpu
|
|
|
|
TestJacCPU.test_multiple_inputs_outputs_pytree_cpu
|
|
|
|
TestJacCPU.test_multiple_inputs_pytree_cpu
|
|
|
|
TestJacCPU.test_multiple_outputs_multiple_argnums_cpu
|
|
|
|
TestJacCPU.test_multiple_outputs_single_argnums_cpu
|
|
|
|
TestJacCPU.test_negative_argnums_jacfwd_cpu
|
|
|
|
TestJacCPU.test_out_of_bounds_argnums_jacfwd_cpu
|
|
|
|
TestJacCPU.test_outputs_can_any_pytree_cpu
|
|
|
|
TestJacCPU.test_repeated_argnums_jacfwd_cpu
|
|
|
|
TestJacCPU.test_simple_jacfwd_cpu
|
|
|
|
TestJacCPU.test_simple_jacrev_cpu
|
|
|
|
TestJacCPU.test_simple_not_flat_jacfwd_cpu
|
|
|
|
TestJacCPU.test_simple_not_flat_jacrev_cpu
|
|
|
|
TestJacCPU.test_unrelated_input_cpu
|
|
|
|
TestJacCPU.test_unrelated_output_cpu
|
|
|
|
TestJacCPU.test_vmap_on_jac_simple_jacfwd_cpu
|
Let torch dynamo inline torch.func.grad (#118407)
|
2024-02-28 20:05:00 +00:00 |
|
TestJacCPU.test_vmap_on_jac_simple_jacrev_cpu
|
|
|
|
TestJit.test_batchnorm
|
|
|
|
TestJit.test_function_default_values
|
|
|
|
TestJvpCPU.test_disable_fwd_grad_outside_cpu
|
|
|
|
TestJvpCPU.test_inputs_are_tuples_of_tensors_cpu
|
|
|
|
TestJvpCPU.test_jvp_inside_autograd_function_cpu
|
|
|
|
TestJvpCPU.test_nonempty_primals_and_tangents_cpu
|
|
|
|
TestJvpCPU.test_primals_tangents_length_mismatch_cpu
|
|
|
|
TestJvpCPU.test_simple_cpu
|
|
|
|
TestLazyModules.test_lazy_batchnorm_with_dict_input
|
|
|
|
TestLazyModules.test_lazy_conv_transpose1d_pickle
|
|
|
|
TestLazyModules.test_lazy_conv_transpose2d
|
|
|
|
TestLazyModules.test_lazy_conv_transpose2d_pickle
|
|
|
|
TestLazyModules.test_lazy_conv_transpose3d
|
|
|
|
TestLazyModules.test_lazy_conv_transpose3d_pickle
|
|
|
|
TestLazyModules.test_lazy_conv_transposed1d
|
|
|
|
TestLazyModules.test_lazy_conv1d
|
|
|
|
TestLazyModules.test_lazy_conv1d_pickle
|
|
|
|
TestLazyModules.test_lazy_conv2d
|
|
|
|
TestLazyModules.test_lazy_conv2d_pickle
|
|
|
|
TestLazyModules.test_lazy_conv3d
|
|
|
|
TestLazyModules.test_lazy_conv3d_pickle
|
|
|
|
TestLazyModules.test_lazy_linear_pickle
|
|
|
|
TestLazyModules.test_linear
|
|
|
|
TestLexsort.test_datetime
|
|
|
|
TestLexsort.test_mixed
|
|
|
|
TestLinalgCPU.test_addmm_sizes_cpu_float32
|
|
|
|
TestLinalgCPU.test_addmm_sizes_cpu_float64
|
|
|
|
TestLinalgCPU.test_addr_integral_cpu_int8
|
|
|
|
TestLinalgCPU.test_addr_integral_cpu_int16
|
|
|
|
TestLinalgCPU.test_addr_integral_cpu_int32
|
|
|
|
TestLinalgCPU.test_addr_integral_cpu_int64
|
|
|
|
TestLinalgCPU.test_addr_integral_cpu_uint8
|
|
|
|
TestLinalgCPU.test_geqrf_cpu_complex64
|
|
|
|
TestLinalgCPU.test_geqrf_cpu_complex128
|
|
|
|
TestLinalgCPU.test_geqrf_cpu_float32
|
|
|
|
TestLinalgCPU.test_geqrf_cpu_float64
|
|
|
|
TestLinalgCPU.test_householder_product_cpu_complex64
|
|
|
|
TestLinalgCPU.test_householder_product_cpu_complex128
|
|
|
|
TestLinalgCPU.test_householder_product_cpu_float32
|
|
|
|
TestLinalgCPU.test_householder_product_cpu_float64
|
|
|
|
TestLinalgCPU.test_inverse_cpu_complex64
|
|
|
|
TestLinalgCPU.test_inverse_cpu_complex128
|
|
|
|
TestLinalgCPU.test_inverse_cpu_float32
|
|
|
|
TestLinalgCPU.test_inverse_cpu_float64
|
|
|
|
TestLinalgCPU.test_lobpcg_torchscript_cpu_float64
|
|
|
|
TestLinalgCPU.test_matmul_small_brute_force_3d_Nd_cpu_complex64
|
|
|
|
TestLinalgCPU.test_matmul_small_brute_force_3d_Nd_cpu_float32
|
|
|
|
TestLinalgCPU.test_matmul_small_brute_force_3d_Nd_cpu_int64
|
|
|
|
TestLinalgCPU.test_pinv_cpu_complex64
|
|
|
|
TestLinalgCPU.test_pinv_cpu_complex128
|
|
|
|
TestLinalgCPU.test_pinv_cpu_float32
|
|
|
|
TestLinalgCPU.test_pinv_cpu_float64
|
|
|
|
TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_complex64
|
|
|
|
TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_complex128
|
|
|
|
TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_float32
|
|
|
|
TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_float64
|
|
|
|
TestLinalgCPU.test_solve_cpu_complex64
|
|
|
|
TestLinalgCPU.test_solve_cpu_complex128
|
|
|
|
TestLinalgCPU.test_solve_cpu_float32
|
|
|
|
TestLinalgCPU.test_solve_cpu_float64
|
|
|
|
TestMatmul.test_exceptions
|
|
|
|
TestMedian.test_extended_axis_invalid
|
|
|
|
TestMeshgrid.test_indexing
|
|
|
|
TestMeshgrid.test_invalid_arguments
|
|
|
|
TestMetaKernel.test_addmm_invalid_dtype
|
|
|
|
TestMin.test_adapt
|
|
|
|
TestMin.test_attn
|
|
|
|
TestMin.test_big_split
|
|
|
|
TestMin.test_c
|
|
|
|
TestMin.test_compare_dims
|
|
|
|
TestMin.test_diag
|
|
|
|
TestMin.test_dim_args
|
|
|
|
TestMin.test_dims_with_size
|
|
|
|
TestMin.test_dir
|
|
|
|
TestMin.test_doc
|
|
|
|
TestMin.test_embed
|
|
|
|
TestMin.test_eq
|
|
|
|
TestMin.test_expand
|
|
|
|
TestMin.test_functorch
|
|
|
|
TestMin.test_hello
|
|
|
|
TestMin.test_index
|
|
|
|
TestMin.test_index_placement
|
|
|
|
TestMin.test_inplace
|
|
|
|
TestMin.test_manual_stuff
|
|
|
|
TestMin.test_mask
|
|
|
|
TestMin.test_max
|
|
|
|
TestMin.test_mm
|
|
|
|
TestMin.test_mm_fuse
|
|
|
|
TestMin.test_monkey
|
|
|
|
TestMin.test_network
|
|
|
|
TestMin.test_order
|
|
|
|
TestMin.test_order_keyword
|
|
|
|
TestMin.test_parse
|
|
|
|
TestMin.test_permute_orig
|
|
|
|
TestMin.test_seg
|
|
|
|
TestMin.test_simple
|
|
|
|
TestMin.test_softmax_split
|
|
|
|
TestMin.test_stack
|
|
|
|
TestMin.test_time_mm_fuse
|
|
|
|
TestMin.test_with_dims_split
|
|
|
|
TestMinFunctorchOnly.test_adapt
|
|
|
|
TestMinFunctorchOnly.test_attn
|
|
|
|
TestMinFunctorchOnly.test_big_split
|
|
|
|
TestMinFunctorchOnly.test_c
|
|
|
|
TestMinFunctorchOnly.test_compare_dims
|
|
|
|
TestMinFunctorchOnly.test_diag
|
|
|
|
TestMinFunctorchOnly.test_dim_args
|
|
|
|
TestMinFunctorchOnly.test_dims_with_size
|
|
|
|
TestMinFunctorchOnly.test_dir
|
|
|
|
TestMinFunctorchOnly.test_doc
|
|
|
|
TestMinFunctorchOnly.test_embed
|
|
|
|
TestMinFunctorchOnly.test_eq
|
|
|
|
TestMinFunctorchOnly.test_expand
|
|
|
|
TestMinFunctorchOnly.test_functorch
|
|
|
|
TestMinFunctorchOnly.test_hello
|
|
|
|
TestMinFunctorchOnly.test_index
|
|
|
|
TestMinFunctorchOnly.test_index_placement
|
|
|
|
TestMinFunctorchOnly.test_inplace
|
|
|
|
TestMinFunctorchOnly.test_manual_stuff
|
|
|
|
TestMinFunctorchOnly.test_mask
|
|
|
|
TestMinFunctorchOnly.test_max
|
|
|
|
TestMinFunctorchOnly.test_mm
|
|
|
|
TestMinFunctorchOnly.test_mm_fuse
|
|
|
|
TestMinFunctorchOnly.test_monkey
|
|
|
|
TestMinFunctorchOnly.test_network
|
|
|
|
TestMinFunctorchOnly.test_order
|
|
|
|
TestMinFunctorchOnly.test_order_keyword
|
|
|
|
TestMinFunctorchOnly.test_parse
|
|
|
|
TestMinFunctorchOnly.test_permute_orig
|
|
|
|
TestMinFunctorchOnly.test_seg
|
|
|
|
TestMinFunctorchOnly.test_simple
|
|
|
|
TestMinFunctorchOnly.test_softmax_split
|
|
|
|
TestMinFunctorchOnly.test_stack
|
|
|
|
TestMinFunctorchOnly.test_with_dims_split
|
|
|
|
TestMinMax.test_axis
|
|
|
|
TestMinMax.test_scalar
|
|
|
|
TestMisc.test_byteorder_check
|
|
|
|
TestMisc.test_generalized_raise_multiloop
|
|
|
|
TestMkldnnCPU.test_add_cpu
|
|
|
|
TestMkldnnCPU.test_batch_norm_2d_cpu
|
|
|
|
TestMkldnnCPU.test_batch_norm_3d_cpu
|
|
|
|
TestMkldnnCPU.test_conv1d_cpu
|
|
|
|
TestMkldnnCPU.test_conv2d_cpu
|
|
|
|
TestMkldnnCPU.test_conv3d_cpu
|
|
|
|
TestMkldnnCPU.test_copy_cpu
|
|
|
|
TestMkldnnCPU.test_linear_cpu
|
|
|
|
TestMkldnnCPU.test_linear_lowp_cpu_float16
|
|
|
|
TestMkldnnCPU.test_mul_cpu
|
|
|
|
TestMkldnnCPU.test_prelu_bf16_cpu
|
|
|
|
TestMkldnnCPU.test_prelu_cpu
|
|
|
|
TestMkldnnCPU.test_reshape_blocked_format_cpu
|
|
|
|
TestMkldnnCPU.test_resnet18_cpu
|
|
|
|
TestMkldnnCPU.test_resnext50_32x4d_cpu
|
|
|
|
TestMkldnnCPU.test_sigmoid_cpu
|
|
|
|
TestMkldnnCPU.test_tanh_cpu
|
|
|
|
TestModuleHookNN.test_hook_inplace
|
|
|
|
TestMoveaxis.test_errors
|
|
|
|
TestMultiDot.test_dynamic_programming_optimization_and_out
|
|
|
|
TestMultiDot.test_three_arguments_and_out
|
|
|
|
TestMultiDot.test_too_few_input_arrays
|
|
|
|
TestMultiDot.test_two_arguments_and_out
|
|
|
|
TestMultiprocessing.test_empty_shared
|
|
|
|
TestMultiprocessing.test_fs_is_shared
|
|
|
|
TestMultiprocessing.test_inherit_tensor
|
|
|
|
TestMultiprocessing.test_is_shared
|
|
|
|
TestNamedTensor.test_addmm
|
|
|
|
TestNamedTensor.test_bmm
|
|
|
|
TestNamedTensor.test_cat
|
|
|
|
TestNamedTensor.test_flatten
|
|
|
|
TestNamedTensor.test_info_smoke
|
|
|
|
TestNamedTensor.test_logical_ops
|
|
|
|
TestNamedTensor.test_matmul
|
|
|
|
TestNamedTensor.test_mm
|
|
|
|
TestNamedTensor.test_reduction_fns
|
|
|
|
TestNamedTensor.test_set_names_property
|
|
|
|
TestNamedTensor.test_transpose_variants
|
|
|
|
TestNamedTensor.test_unary_propagate_names_fns
|
|
|
|
TestNamedTensor.test_unflatten
|
|
|
|
TestNamedTensor.test_using_unseen_interned_string_bumps_refcount_permanently
|
|
|
|
TestNamedTensor.test_using_unseen_uninterned_string_refcounts
|
|
|
|
TestNamedTuple.test_max
|
|
|
|
TestNativeFunctions.test_optional_floatlist_invalid
|
|
|
|
TestNativeFunctions.test_optional_intlist_invalid
|
|
|
|
TestNativeFunctions.test_symintlist_error
|
|
|
|
TestNativeFunctions.test_symintlist_error_with_overload
|
|
|
|
TestNativeFunctions.test_vararg_symintlist_error
|
|
|
|
TestNegative.test_exceptions
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_early_stop_False
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_early_stop_True
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_kwargs_early_stop_False
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_kwargs_early_stop_True
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_non_tensor_inputs_and_outputs_early_stop_False
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_non_tensor_inputs_and_outputs_early_stop_True
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_reentrant_backwards_early_stop_False
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_reentrant_backwards_early_stop_True
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_same_graph_early_stop_False
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_same_graph_early_stop_True
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_set_early_stop
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_set_early_stop_no_recompution_needed
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_two_children_early_stop_False
|
|
|
|
TestNestedCheckpoint.test_nested_checkpoint_two_children_early_stop_True
|
|
|
|
TestNestedTensor.test_binary_recompiles
|
|
|
|
TestNestedTensor.test_unbind
|
|
|
|
TestNN.test_fb_fc_packed
|
|
|
|
TestNN.test_ParameterDict_replication
|
|
|
|
TestNN.test_ParameterList_replication
|
|
|
|
TestNN.test_Sequential_iadd
|
|
|
|
TestNNDeviceTypeCPU.test_invalid_reduction_strings_cpu
|
|
|
|
TestNNDeviceTypeCPU.test_module_to_empty_cpu_float32
|
|
|
|
TestNNDeviceTypeCPU.test_module_to_empty_cpu_float64
|
|
|
|
TestNNDeviceTypeCPU.test_nll_loss_byte_target_matches_long_cpu
|
|
|
|
TestNNDeviceTypeCPU.test_threshold_inplace_overlap_cpu
|
|
|
|
TestNNDeviceTypeCPU.test_triplet_margin_with_distance_loss_cpu
|
|
|
|
TestNNDeviceTypeCUDA.test_nll_loss_byte_target_matches_long_cuda
|
Fix nll loss dynamo failure (#120805)
|
2024-02-29 22:34:49 +00:00 |
|
TestNNParametrization.test_deepcopy_after_parametrization
|
|
|
|
TestNNParametrization.test_new_spectral_norm
|
|
|
|
TestNNParametrization.test_new_spectral_norm_dim
|
|
|
|
TestNNParametrization.test_new_spectral_norm_forward
|
|
|
|
TestNNParametrization.test_weight_norm_deepcopy
|
|
|
|
TestNNParametrization.test_weight_norm_state_dict_compat
|
|
|
|
TestNonarrayArgs.test_dunder_round_edgecases_val_2147483647_ndigits_-1
|
|
|
|
TestNonarrayArgs.test_dunder_round_edgecases_val_2147483647_ndigits_-9
|
|
|
|
TestNonarrayArgs.test_dunder_round_edgecases_val_2147483647_ndigits_-10
|
|
|
|
TestNonzeroAndCountNonzero.test_count_nonzero_axis
|
|
|
|
TestNormDouble.test_axis
|
|
|
|
TestNormDouble.test_bad_args
|
|
|
|
TestNormDouble.test_matrix_2x2
|
|
|
|
TestNormInt64.test_axis
|
|
|
|
TestNormInt64.test_bad_args
|
|
|
|
TestNormInt64.test_matrix_2x2
|
|
|
|
TestNormSingle.test_axis
|
|
|
|
TestNormSingle.test_bad_args
|
|
|
|
TestNormSingle.test_matrix_2x2
|
|
|
|
TestNumPyInteropCPU.test_numpy_non_writeable_cpu
|
|
|
|
TestObserver.test_per_channel_observers
|
|
|
|
TestObserver.test_per_tensor_observers
|
|
|
|
TestOpCPU.test_cat_cpu_bfloat16
|
|
|
|
TestOpCPU.test_cat_cpu_float32
|
|
|
|
TestOptimizations.test_example_inputs
|
|
|
|
TestOptimizations.test_example_inputs_runtime_use
|
|
|
|
TestOutDtypeOp.test_out_dtype_non_op_overload
|
|
|
|
TestOutDtypeOp.test_out_dtype_wrong_output
|
|
|
|
TestPadding.test_constant_padNd
|
|
|
|
TestPartitioning.test_contiguous
|
|
|
|
TestPartitioning.test_default_partitioner_output_tensor_shape_tensor
|
|
|
|
TestPartitioning.test_min_cut_partitioner_output_tensor_shape_tensor
|
|
|
|
TestPasses.test_functionalize_inline_contraints
|
|
|
|
TestPasses.test_views_op_having_view_copy
|
|
|
|
TestPassManager.test_pass_manager
|
|
|
|
TestPercentile.test_extended_axis_invalid
|
|
|
|
TestPercentile.test_scalar_q
|
|
|
|
TestPickle.test_pickle
|
|
|
|
TestPoolingNN.test_MaxUnpool2d_output_size
|
|
|
|
TestPower.test_modular_power
|
|
|
|
TestProfiler.test_concrete_inputs_profiling
|
|
|
|
TestProfiler.test_export_stacks
|
|
|
|
TestProfiler.test_flops
|
|
|
|
TestProfiler.test_guarded_record_function_fast
|
|
|
|
TestProfiler.test_high_level_trace
|
|
|
|
TestProfiler.test_is_profiler_enabled
|
|
|
|
TestProfiler.test_profiler_correlation_id
|
|
|
|
TestProfiler.test_profiler_fwd_bwd_link
|
|
|
|
TestProfiler.test_profiler_op_event_args
|
|
|
|
TestProfiler.test_profiler_tracing
|
|
|
|
TestProfiler.test_source
|
|
|
|
TestProfiler.test_source_multithreaded_basic_work_in_main_thread_True
|
|
|
|
TestProfiler.test_source_multithreaded_close_in_scope_work_in_main_thread_True
|
|
|
|
TestProfiler.test_source_multithreaded_complex_work_in_main_thread_True
|
|
|
|
TestProfiler.test_source_multithreaded_multiple_preexisting_work_in_main_thread_True
|
|
|
|
TestProfiler.test_source_multithreaded_open_in_scope_work_in_main_thread_True
|
|
|
|
TestProfilerTree.test_profiler_experimental_tree_with_memory
|
|
|
|
TestProfilerTree.test_profiler_experimental_tree_with_memory_and_stack
|
|
|
|
TestProfilerTree.test_profiler_experimental_tree_with_record_function
|
|
|
|
TestProfilerTree.test_profiler_experimental_tree_with_stack_and_torch_dispatch
|
|
|
|
TestProfilerTree.test_profiler_experimental_tree_with_stack_and_torch_function
|
|
|
|
TestPT2ERepresentation.test_add
|
|
|
|
TestPT2ERepresentation.test_add_relu
|
|
|
|
TestPT2ERepresentation.test_conv2d
|
|
|
|
TestPT2ERepresentation.test_dynamic_linear
|
|
|
|
TestPT2ERepresentation.test_maxpool2d
|
|
|
|
TestPT2ERepresentation.test_qdq
|
|
|
|
TestPT2ERepresentation.test_qdq_per_channel
|
|
|
|
TestPT2ERepresentation.test_static_linear
|
|
|
|
TestPutAlongAxis.test_broadcast
|
|
|
|
TestPutmask.test_byteorder_greater_False
|
|
|
|
TestPutmask.test_byteorder_greater_True
|
|
|
|
TestPutmask.test_record_array
|
|
|
|
TestPythonAutograd.test_backwards2
|
|
|
|
TestPythonAutograd.test_forwards1
|
|
|
|
TestPythonAutograd.test_forwards2
|
|
|
|
TestPythonAutograd.test_split
|
|
|
|
TestPythonDispatch.test_make_subclass_with_modes
|
|
|
|
TestPythonPytree.test_treespec_equality
|
|
|
|
TestPythonRegistration.test_alias_analysis
|
|
|
|
TestPythonRegistration.test_finalizer
|
|
|
|
TestQR.test_mode_raw
|
|
|
|
TestQR.test_qr_empty_m_3_n_0
|
|
|
|
TestQR.test_qr_empty_m_0_n_0
|
|
|
|
TestQR.test_qr_empty_m_0_n_3
|
|
|
|
TestQuantile.test_quantile_monotonic_method_averaged_inverted_cdf
|
|
|
|
TestQuantile.test_quantile_monotonic_method_closest_observation
|
|
|
|
TestQuantile.test_quantile_monotonic_method_hazen
|
|
|
|
TestQuantile.test_quantile_monotonic_method_interpolated_inverted_cdf
|
|
|
|
TestQuantile.test_quantile_monotonic_method_inverted_cdf
|
|
|
|
TestQuantile.test_quantile_monotonic_method_median_unbiased
|
|
|
|
TestQuantile.test_quantile_monotonic_method_normal_unbiased
|
|
|
|
TestQuantile.test_quantile_monotonic_method_weibull
|
|
|
|
TestQuantizedEmbeddingOps.test_embedding_bag_2bit
|
|
|
|
TestQuantizedEmbeddingOps.test_embedding_bag_2d_indices
|
|
|
|
TestQuantizedEmbeddingOps.test_embedding_bag_4bit
|
|
|
|
TestQuantizedEmbeddingOps.test_embedding_bag_byte
|
|
|
|
TestQuantizedOps.test_avg_pool2d_nhwc
|
|
|
|
TestQuantizedOps.test_avg_pool3d_nhwc
|
|
|
|
TestQuantizedOps.test_cat_nhwc
|
|
|
|
TestQuantizedOps.test_custom_module_lstm
|
|
|
|
TestQuantizedOps.test_interpolate
|
|
|
|
TestQuantizedOps.test_interpolate3d
|
|
|
|
TestQuantizedOps.test_max_pool2d_cudnn
|
|
|
|
TestQuantizedOps.test_max_pool2d_nhwc
|
|
|
|
TestQuantizedOps.test_qtanh
|
|
|
|
TestQuantizedTensor.test_choose_qparams
|
|
|
|
TestQuantizedTensor.test_choose_qparams_optimized
|
|
|
|
TestQuantizedTensor.test_decomposed_dequantize_per_tensor
|
|
|
|
TestQuantizedTensor.test_decomposed_quantize_per_tensor
|
|
|
|
TestQuantizedTensor.test_decomposed_quantize_per_tensor_bfloat16_input
|
|
|
|
TestQuantizedTensor.test_fp16_saturate_op
|
|
|
|
TestQuantizedTensor.test_per_channel_qtensor_to_memory_format
|
|
|
|
TestQuantizedTensor.test_per_tensor_qtensor_to_memory_format
|
|
|
|
TestQuantizedTensor.test_qtensor_cpu
|
|
|
|
TestQuantizedTensor.test_quantize_per_channel_sub_byte
|
|
|
|
TestQuantizeEagerQATNumerics.test_conv_bn_folded_vs_unfolded
|
|
|
|
TestQuantizeFx.test_prepare_custom_config_set_standalone_module_class
|
|
|
|
TestQuantizeFx.test_static_lstm_with_custom_fixed_qparams
|
|
|
|
TestQuantizePT2E.test_composable_quantizer_linear_conv
|
|
|
|
TestQuantizePT2E.test_constant_prop_preserve_metadata
|
|
|
|
TestQuantizePT2E.test_embedding_conv_linear_quantization
|
|
|
|
TestQuantizePT2E.test_fold_all_ops_before_quantize
|
|
|
|
TestQuantizePT2E.test_fold_quantize
|
|
|
|
TestQuantizePT2E.test_fold_quantize_per_channel
|
|
|
|
TestQuantizePT2E.test_groupwise_per_channel_quant
|
|
|
|
TestQuantizePT2E.test_reentrant
|
|
|
|
TestQuantizePT2E.test_save_load
|
|
|
|
TestQuantizePT2E.test_speed
|
|
|
|
TestQuantizePT2EQAT_ConvBn1d.test_prepare_qat_conv_bn_fusion_getitem_placeholder
|
|
|
|
TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_fusion
|
|
|
|
TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_fusion_literal_args
|
|
|
|
TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_fusion_no_conv_bias
|
|
|
|
TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_relu_fusion
|
|
|
|
TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_relu_fusion_no_conv_bias
|
|
|
|
TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_no_bias
|
|
|
|
TestQuantizePT2EQAT_ConvBn1d.test_qat_inplace_add_relu
|
|
|
|
TestQuantizePT2EQAT_ConvBn1d.test_qat_preserve_source_fn_stack
|
|
|
|
TestQuantizePT2EQAT_ConvBn1d.test_qat_update_shared_qspec
|
|
|
|
TestQuantizePT2EQAT_ConvBn2d.test_prepare_qat_conv_bn_fusion_getitem_placeholder
|
|
|
|
TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_fusion
|
|
|
|
TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_fusion_literal_args
|
|
|
|
TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_fusion_no_conv_bias
|
|
|
|
TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_relu_fusion
|
|
|
|
TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_relu_fusion_no_conv_bias
|
|
|
|
TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_no_bias
|
|
|
|
TestQuantizePT2EQAT_ConvBn2d.test_qat_inplace_add_relu
|
|
|
|
TestQuantizePT2EQAT_ConvBn2d.test_qat_preserve_source_fn_stack
|
|
|
|
TestQuantizePT2EQAT_ConvBn2d.test_qat_update_shared_qspec
|
|
|
|
TestQuantizePT2EQATModels.test_qat_mobilenet_v2
|
|
|
|
TestQuantizePT2EQATModels.test_qat_resnet18
|
|
|
|
TestRecordFunction.test_datapipe_with_record_function
|
|
|
|
TestRecordFunction.test_datapipe_with_record_function_fork
|
|
|
|
TestRecordFunction.test_record_function
|
|
|
|
TestRequire.test_C_and_F_simul
|
|
|
|
TestRequire.test_require_each
|
|
|
|
TestRequire.test_unknown_requirement
|
|
|
|
TestRollaxis.test_exceptions
|
|
|
|
TestRot90.test_basic
|
|
|
|
TestScalarIndexing.test_invalid_newaxis
|
|
|
|
TestScalarIndexing.test_invalid_subscript
|
|
|
|
TestScalarIndexing.test_invalid_subscript_assignment
|
|
|
|
TestScalarTypeNames.test_names_reflect_attributes_t0
|
|
|
|
TestScalarTypeNames.test_names_reflect_attributes_t1
|
|
|
|
TestScalarTypeNames.test_names_reflect_attributes_t2
|
|
|
|
TestScalarTypeNames.test_names_reflect_attributes_t3
|
|
|
|
TestScalarTypeNames.test_names_reflect_attributes_t4
|
|
|
|
TestScalarTypeNames.test_names_reflect_attributes_t5
|
|
|
|
TestScalarTypeNames.test_names_reflect_attributes_t6
|
|
|
|
TestScalarTypeNames.test_names_reflect_attributes_t7
|
|
|
|
TestScalarTypeNames.test_names_reflect_attributes_t8
|
|
|
|
TestScalarTypeNames.test_names_reflect_attributes_t9
|
|
|
|
TestScript.test_conv_error
|
|
|
|
TestScript.test_dict_str
|
|
|
|
TestScript.test_empty_tuple_str
|
|
|
|
TestScript.test_error_stacktrace_interface
|
|
|
|
TestScript.test_function_overloading_isinstance
|
|
|
|
TestScript.test_function_overloads
|
|
|
|
TestScript.test_ignored_as_value
|
|
|
|
TestScript.test_infer_size
|
|
|
|
TestScript.test_is_after_use
|
|
|
|
TestScript.test_is_scripting
|
|
|
|
TestScript.test_isinstance_dynamic
|
|
|
|
TestScript.test_method_overloading
|
|
|
|
TestScript.test_namedtuple_default_values_using_factory_constructor
|
|
|
|
TestScript.test_namedtuple_python
|
|
|
|
TestScript.test_nested_breaks
|
|
|
|
TestScript.test_nn_GRU
|
|
|
|
TestScript.test_nn_LSTM_with_layers
|
|
|
|
TestScript.test_no_self_arg_ignore_function
|
|
|
|
TestScript.test_none_type_str
|
|
|
|
TestScript.test_pack_unpack_state
|
|
|
|
TestScript.test_parse_nested_names
|
|
|
|
TestScript.test_parse_tensor_constants
|
|
|
|
TestScript.test_python_call
|
|
|
|
TestScript.test_python_call_non_tensor_wrong
|
|
|
|
TestScript.test_python_op_builtins
|
|
|
|
TestScript.test_script_optional_none
|
|
|
|
TestScript.test_string_device_implicit_conversion
|
|
|
|
TestScript.test_torch_functional_tensordot_int
|
|
|
|
TestScript.test_torchscript_multi_head_attn
|
|
|
|
TestScript.test_tuple_str
|
|
|
|
TestScript.test_type_annotation_module
|
|
|
|
TestScript.test_unspecialized_any_binding
|
|
|
|
TestScript.test_unused_decorator
|
|
|
|
TestScript.test_wrong_return_type
|
|
|
|
TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_bfloat16_cpu_bfloat16
|
|
|
|
TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_float16_cpu_float16
|
|
|
|
TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_float32_cpu_float32
|
|
|
|
TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_float64_cpu_float64
|
|
|
|
TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_bfloat16_cpu_bfloat16
|
|
|
|
TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_float16_cpu_float16
|
|
|
|
TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_float32_cpu_float32
|
|
|
|
TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_float64_cpu_float64
|
|
|
|
TestSetOps.test_in1d_table_timedelta_fails
|
|
|
|
TestSetOps.test_in1d_timedelta_kind_sort
|
|
|
|
TestSetOps.test_in1d_timedelta_kind0
|
|
|
|
TestSetOps.test_setdiff1d
|
|
|
|
TestShapeOpsCUDA.test_flip_cuda_float32
|
|
|
|
TestSortAndSelectCPU.test_sort_overflow_cpu_int16
|
|
|
|
TestSortAndSelectCPU.test_topk_arguments_cpu
|
|
|
|
TestSortComplex.test_sort_real_type_in_g_type_out_G
|
|
|
|
TestSortComplex.test_sort_real_type_in_H_type_out_F
|
|
|
|
TestSplit.test_unequal_split
|
|
|
|
TestStackMisc.test_stack
|
|
|
|
TestStatelessFunctionalAPI.test_reparametrize_module_fail_reset_to_original_stateless
|
|
|
|
TestStatelessFunctionalAPI.test_reparametrize_module_fail_reset_to_original_torch_func
|
|
|
|
TestStatelessFunctionalAPI.test_reparametrized_module_change_parametrization_original_stateless
|
|
|
|
TestStatelessFunctionalAPI.test_reparametrized_module_change_parametrization_original_torch_func
|
|
|
|
TestStaticQuantizedModule.test_embedding_api
|
|
|
|
TestStaticQuantizedModule.test_embedding_bag_api
|
|
|
|
TestStats.test_dtype_from_dtype
|
|
|
|
TestStats.test_out
|
|
|
|
TestSubclass.test_parametrization_base_tensor_leave_parametrized_False
|
|
|
|
TestSubclass.test_parametrization_base_tensor_leave_parametrized_True
|
|
|
|
TestTake.test_raise
|
|
|
|
TestTakeAlongAxis.test_invalid
|
|
|
|
TestTEFuserDynamic.test_gelu
|
|
|
|
TestTEFuserDynamic.test_profiler
|
|
|
|
TestTEFuserDynamic.test_remove_output_used_only_in_size
|
|
|
|
TestTEFuserDynamic.test_to_dtype
|
|
|
|
TestTEFuserDynamic.test_torch_to
|
|
|
|
TestTEFuserStatic.test_gelu
|
|
|
|
TestTEFuserStatic.test_inlined_optimized_graph
|
|
|
|
TestTEFuserStatic.test_profiler
|
|
|
|
TestTEFuserStatic.test_remove_output_used_only_in_size
|
|
|
|
TestTEFuserStatic.test_skip_grad_in_check
|
|
|
|
TestTEFuserStatic.test_to_dtype
|
|
|
|
TestTEFuserStatic.test_torch_to
|
|
|
|
TestTensorBoardEmbedding.test_embedding
|
|
|
|
TestTensorBoardEmbedding.test_embedding_64
|
|
|
|
TestTensorBoardSummary.test_image_with_one_channel
|
|
|
|
TestTensorBoardSummary.test_image_without_channel
|
|
|
|
TestTensorBoardSummary.test_uint8_image
|
|
|
|
TestTensorBoardSummary.test_video
|
|
|
|
TestTensorBoardUtils.test_numpy_vid_uint8
|
|
|
|
TestTensorBoardUtils.test_to_HWC
|
|
|
|
TestTensorCreationCPU.test_block_diag_cpu
|
|
|
|
TestTensorCreationCPU.test_constructor_dtypes_cpu
|
|
|
|
TestTorch.test_contains
|
|
|
|
TestTorch.test_cuda_not_built
|
|
|
|
TestTorch.test_map
|
|
|
|
TestTorch.test_new
|
|
|
|
TestTorch.test_parsing_double
|
|
|
|
TestTorch.test_parsing_int64
|
|
|
|
TestTorch.test_parsing_intlist
|
|
|
|
TestTorch.test_pin_memory
|
|
|
|
TestTorch.test_type
|
|
|
|
TestTorch.test_upsample_nearest2d_meta
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_add_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_addcdiv_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_addcmul_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_atan2_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_copy_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_div_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_eq_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_fmod_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_ge_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_gt_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_le_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_lerp_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_lt_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_map_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_map2_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_masked_fill_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_masked_scatter_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_mul_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_ne_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_pow_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_remainder_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_broadcast_fn_sub_cpu
|
|
|
|
TestTorchDeviceTypeCPU.test_deterministic_empty_cpu_uint64
|
|
|
|
TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool1d_cpu_float32
|
|
|
|
TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool1d_cpu_float64
|
|
|
|
TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool2d_cpu_float32
|
|
|
|
TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool2d_cpu_float64
|
|
|
|
TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool3d_cpu_float32
|
|
|
|
TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool3d_cpu_float64
|
|
|
|
TestTorchFunctionMode.test_mode_notimplemented_loop
|
|
|
|
TestTorchFunctionMode.test_modes_return_notimplemented
|
|
|
|
TestTorchFunctionMode.test_nested_modes_with_python_has_torch_function
|
|
|
|
TestTorchFunctionOverride.test_Tensor___cuda_array_interface_____get__
|
|
|
|
TestTorchTidyProfiler.test_allocation_id_uniqueness
|
|
|
|
TestTorchTidyProfiler.test_allocation_ids
|
|
|
|
TestTorchTidyProfiler.test_allocation_ids_with_other_ops
|
|
|
|
TestTorchTidyProfiler.test_allocations
|
|
|
|
TestTorchTidyProfiler.test_extra_fields
|
|
|
|
TestTorchTidyProfiler.test_impl_reuse
|
|
|
|
TestTorchTidyProfiler.test_mkldnn_tensors
|
|
|
|
TestTorchTidyProfiler.test_module_and_optimizer_ids
|
|
|
|
TestTorchTidyProfiler.test_nnmodule_params
|
|
|
|
TestTorchTidyProfiler.test_optimizer
|
|
|
|
TestTorchTidyProfiler.test_optimizer_parameters_adam
|
|
|
|
TestTorchTidyProfiler.test_optimizer_parameters_sgd
|
|
|
|
TestTorchTidyProfiler.test_pointers_and_ids
|
|
|
|
TestTorchTidyProfiler.test_scalar_ins
|
|
|
|
TestTorchTidyProfiler.test_tensor_lists
|
|
|
|
TestTorchTidyProfiler.test_tensor_properties
|
|
|
|
TestTorchTidyProfiler.test_tensorimpl_invalidation_full
|
|
|
|
TestTorchTidyProfiler.test_tensorimpl_invalidation_keep_alive
|
|
|
|
TestTorchTidyProfiler.test_tensorimpl_invalidation_scalar_args
|
|
|
|
TestTorchTidyProfiler.test_tensorimpl_invalidation_set
|
|
|
|
TestTrilIndicesFrom.test_exceptions
|
|
|
|
TestTriuIndicesFrom.test_exceptions
|
|
|
|
TestTypeHints.test_doc_examples
|
|
|
|
TestTypePromotionCPU.test_alpha_mismatch_cpu
|
|
|
|
TestTypePromotionCPU.test_alternate_result_cpu
|
|
|
|
TestUnflatten.test_unflatten_container_type
|
|
|
|
TestUnique.test_unique_axis
|
|
|
|
TestUnique.test_unique_axis_errors
|
|
|
|
TestUtils.test_get_fqn_to_example_inputs_complex_args
|
|
|
|
TestUtils.test_get_fqn_to_example_inputs_default_kwargs
|
|
|
|
TestUtils.test_get_fqn_to_example_inputs_simple
|
|
|
|
TestVerifyCorrectness.test_incorrect_verify_true
|
|
|
|
TestVerifyCorrectness.test_torchscript
|
|
|
|
TestVmapAPILegacy.test_accepts_nested_inputs
|
|
|
|
TestVmapAPILegacy.test_backward_unsupported_interaction
|
|
|
|
TestVmapAPILegacy.test_batched_gradient_basic
|
|
|
|
TestVmapAPILegacy.test_fallback_with_undefined_grad
|
|
|
|
TestVmapAPILegacy.test_grad_unsupported_interaction
|
|
|
|
TestVmapAPILegacy.test_in_dim_not_in_tensor_err_msg
|
|
|
|
TestVmapAPILegacy.test_multiple_out_dims
|
|
|
|
TestVmapAPILegacy.test_multiple_outputs
|
|
|
|
TestVmapAPILegacy.test_multiple_outputs_error_cases
|
|
|
|
TestVmapAPILegacy.test_nested_out_dims
|
|
|
|
TestVmapAPILegacy.test_nested_with_different_map_dim
|
|
|
|
TestVmapAPILegacy.test_non_default_in_dims_out_dims
|
|
|
|
TestVmapAPILegacy.test_non_tensor_output_raises
|
|
|
|
TestVmapAPILegacy.test_non_zero_in_dims
|
|
|
|
TestVmapAPILegacy.test_nonzero_out_dims
|
|
|
|
TestVmapAPILegacy.test_noop_in_inner_vmap
|
|
|
|
TestVmapAPILegacy.test_out_dim_out_of_bounds_err_msg
|
|
|
|
TestVmapAPILegacy.test_out_dims_and_num_outputs_mismatch_err_msg
|
|
|
|
TestVmapAPILegacy.test_out_dims_edge_case
|
|
|
|
TestVmapAPILegacy.test_single_input
|
|
|
|
TestVmapAPILegacy.test_unsupported_op_err_msg
|
|
|
|
TestVmapBatchedGradientLegacyCPU.test_binary_cross_entropy_cpu
|
|
|
|
TestVmapBatchedGradientLegacyCPU.test_div_cpu
|
|
|
|
TestVmapBatchedGradientLegacyCPU.test_index_cpu
|
|
|
|
TestVmapBatchedGradientLegacyCPU.test_inplace_manyview_cpu
|
|
|
|
TestVmapBatchedGradientLegacyCPU.test_inplace_on_view_cpu
|
|
|
|
TestVmapBatchedGradientLegacyCPU.test_lgamma_cpu
|
|
|
|
TestVmapBatchedGradientLegacyCPU.test_log_cpu
|
|
|
|
TestVmapBatchedGradientLegacyCPU.test_log1p_cpu
|
|
|
|
TestVmapBatchedGradientLegacyCPU.test_logsumexp_cpu
|
|
|
|
TestVmapBatchedGradientLegacyCPU.test_mul_cpu
|
|
|
|
TestVmapBatchedGradientLegacyCPU.test_sigmoid_cpu
|
|
|
|
TestVmapBatchedGradientLegacyCPU.test_unrelated_output_cpu
|
|
|
|
TestVmapBatchedGradientLegacyCPU.test_unrelated_output_multiple_grad_cpu
|
|
|
|
TestVmapOperatorsLegacy.test_as_strided
|
|
|
|
TestVmapOperatorsLegacy.test_binary_pointwise_ops
|
|
|
|
TestVmapOperatorsLegacy.test_chunk
|
|
|
|
TestVmapOperatorsLegacy.test_clamp
|
|
|
|
TestVmapOperatorsLegacy.test_clone
|
|
|
|
TestVmapOperatorsLegacy.test_comparison_ops
|
|
|
|
TestVmapOperatorsLegacy.test_diagonal
|
|
|
|
TestVmapOperatorsLegacy.test_fill_and_zero_inplace
|
|
|
|
TestVmapOperatorsLegacy.test_is_complex
|
|
|
|
TestVmapOperatorsLegacy.test_is_contiguous
|
|
|
|
TestVmapOperatorsLegacy.test_is_floating_point
|
|
|
|
TestVmapOperatorsLegacy.test_new_empty
|
|
|
|
TestVmapOperatorsLegacy.test_new_empty_strided
|
|
|
|
TestVmapOperatorsLegacy.test_new_zeros
|
|
|
|
TestVmapOperatorsLegacy.test_no_random_op_support
|
|
|
|
TestVmapOperatorsLegacy.test_reshape
|
|
|
|
TestVmapOperatorsLegacy.test_result_type
|
|
|
|
TestVmapOperatorsLegacy.test_select
|
|
|
|
TestVmapOperatorsLegacy.test_slice
|
|
|
|
TestVmapOperatorsLegacy.test_split
|
|
|
|
TestVmapOperatorsLegacy.test_stride
|
|
|
|
TestVmapOperatorsLegacy.test_sum_dim
|
|
|
|
TestVmapOperatorsLegacy.test_T_numpy
|
|
|
|
TestVmapOperatorsLegacy.test_to
|
|
|
|
TestVmapOperatorsLegacy.test_transpose
|
|
|
|
TestVmapOperatorsLegacy.test_unbind
|
|
|
|
TestVmapOperatorsLegacy.test_view
|
|
|
|
TestVsplit.test_1D_array
|
|
|
|
TestVsplit.test_0D_array
|
|
|
|
TestVsplit.test_non_iterable
|
|
|
|
TestVstack.test_empty_input
|
|
|
|
TestVstack.test_non_iterable
|
|
|
|
TestWhere.test_error
|
|
|
|
TestXNNPACKQuantizer.test_add_and_inplace_add
|
|
|
|
TestXNNPACKQuantizer.test_add_mul_long
|
|
|
|
TestXNNPACKQuantizer.test_add_mul_scalar
|
|
|
|
TestXNNPACKQuantizer.test_conv_linear
|
|
|
|
TestXNNPACKQuantizer.test_conv_linear_no_permute
|
|
|
|
TestXNNPACKQuantizer.test_conv1d
|
|
|
|
TestXNNPACKQuantizer.test_conv1d_with_conv2d
|
|
|
|
TestXNNPACKQuantizer.test_conv2d
|
|
|
|
TestXNNPACKQuantizer.test_dynamic_linear
|
|
|
|
TestXNNPACKQuantizer.test_dynamic_linear_int4_weight
|
|
|
|
TestXNNPACKQuantizer.test_dynamic_linear_with_conv
|
|
|
|
TestXNNPACKQuantizer.test_gru
|
|
|
|
TestXNNPACKQuantizer.test_linear
|
|
|
|
TestXNNPACKQuantizer.test_linear_gru
|
|
|
|
TestXNNPACKQuantizer.test_linear_relu
|
|
|
|
TestXNNPACKQuantizer.test_linear_with_dynamic_shape
|
|
|
|
TestXNNPACKQuantizer.test_mul_and_inplace_mul
|
|
|
|
TestXNNPACKQuantizer.test_mul_float32_max
|
|
|
|
TestXNNPACKQuantizer.test_obs_sharing_ops
|
|
|
|
TestXNNPACKQuantizer.test_propagate_annotation
|
|
|
|
TestXNNPACKQuantizer.test_qat_dynamic_linear
|
|
|
|
TestXNNPACKQuantizerModels.test_resnet18
|
|
|
|
TraceRuleTests.test_skipfiles_inlinelist
|
|
|
|
UnspecTests.test_builtin_max_min
|
|
|
|
UnspecTests.test_conv1d_symint_padding
|
|
|
|
UnspecTests.test_isinstance_symint
|
|
|
|
UnspecTests.test_mark_01_dynamic
|
|
|
|
UnspecTests.test_no_recompilations
|
|
|
|
UnspecTests.test_no_recompiles
|
|
|
|
UnspecTests.test_propagate_dynamic_dim
|
|
|
|
UnspecTests.test_use_and_specialize
|
|
|
