Lines Matching defs:device
127 def test_cuda_vitals_gpu_only(self, device): argument
147 def test_constants(self, device): argument
164 def test_bytes_to_scalar(self, device, dtype): argument
181 def test_storage(self, device, dtype): argument
211 def test_storage_setitem(self, device, dtype): argument
243 def test_tensor_storage_type(self, device, dtype): argument
253 def test_tensor_from_storage(self, device, dtype): argument
270 def test_set_storage(self, device, dtype): argument
308 def test_typed_storage_meta(self, device, dtype): argument
321 def test_untyped_storage_meta(self, device): argument
335 def test_storage_meta_from_tensor(self, device, dtype): argument
344 def test_storage_meta_errors(self, device, dtype): argument
384 def test_storage_meta_ok(self, device, dtype): argument
399 def test_deepcopy(self, device, dtype): argument
426 def test_deepcopy_scalar(self, device, dtype): argument
433 dtype, device, argument
480 def ternary_check_input_output_mem_overlap(self, op, device, argument
535 def test_scalar_check(self, device): argument
732 def test_check_tensor_all(self, device): argument
797 def test_check_tensor_internal(self, device): argument
833 def test_cpp_warnings_have_python_context(self, device): argument
895 def test_warn_always_caught(self, device): argument
915 def test_complex_half_experimental_warning(self, device): argument
952 def test_dtypetensor_warnings(self, device): argument
960 def test_set_default_tensor_type_warnings(self, device): argument
974 def test_conv_transposed_backward_agnostic_to_memory_format(self, device): argument
999 def test_conv_transposed_large(self, device): argument
1012 def test_is_set_to(self, device): argument
1053 def test_broadcast(self, fn, device): argument
1202 def test_cublas_config_nondeterministic_alert(self, device): argument
1273 def test_nondeterministic_resize_quantized(self, device, dtype): argument
1283 def test_deterministic_resize(self, device, dtype): argument
1341 def test_deterministic_empty(self, device, dtype): argument
1368 def test_nondeterministic_alert_AvgPool3d(self, device): argument
1381 def test_nondeterministic_alert_AdaptiveAvgPool2d(self, device): argument
1394 def test_nondeterministic_alert_AdaptiveAvgPool3d(self, device): argument
1407 def test_nondeterministic_alert_MaxPool3d(self, device): argument
1420 def test_nondeterministic_alert_AdaptiveMaxPool2d(self, device): argument
1433 def test_nondeterministic_alert_FractionalMaxPool2d(self, device): argument
1446 def test_nondeterministic_alert_FractionalMaxPool3d(self, device): argument
1459 def test_nondeterministic_alert_MaxUnpool1d(self, device, dtype): argument
1473 def test_nondeterministic_alert_MaxUnpool2d(self, device, dtype): argument
1487 def test_nondeterministic_alert_MaxUnpool3d(self, device, dtype): argument
1501 def test_nondeterministic_alert_interpolate_linear(self, device): argument
1516 def test_nondeterministic_alert_interpolate_bilinear(self, device): argument
1531 def test_deterministic_replication_pad2d(self, device): argument
1562 def test_deterministic_interpolate_bilinear(self, device): argument
1581 def test_nondeterministic_alert_interpolate_bicubic(self, device): argument
1597 def test_nondeterministic_alert_interpolate_trilinear(self, device): argument
1613 def test_nondeterministic_alert_ReflectionPad1d(self, device): argument
1625 def test_nondeterministic_alert_ReflectionPad2d(self, device): argument
1638 def test_nondeterministic_alert_ReflectionPad3d(self, device): argument
1651 def test_nondeterministic_alert_ReplicationPad1d(self, device): argument
1663 def test_nondeterministic_alert_ReplicationPad2d(self, device): argument
1690 def test_nondeterministic_alert_ReplicationPad3d(self, device): argument
1702 def test_nondeterministic_alert_NLLLoss(self, device): argument
1714 def test_nondeterministic_alert_CTCLoss(self, device): argument
1729 def test_nondeterministic_alert_EmbeddingBag_max(self, device): argument
1744 def test_nondeterministic_alert_cumsum(self, device, dtype): argument
1756 def test_nondeterministic_alert_put(self, device): argument
1770 def test_nondeterministic_alert_put_accumulate(self, device): argument
1782 def test_nondeterministic_alert_histc(self, device): argument
1791 def test_nondeterministic_alert_bincount(self, device): argument
1810 def test_nondeterministic_alert_kthvalue(self, device, dtype): argument
1834 def test_nondeterministic_alert_grid_sample_2d(self, device): argument
1847 def test_nondeterministic_alert_grid_sample_3d(self, device): argument
1858 def test_invalid_shapes_grid_sampler(self, device): argument
1908 def test_dist(self, device): argument
1924 def test_nondeterministic_alert_median(self, device, dtype): argument
1958 def _test_gather_backward_one_dim(self, device, deterministic: bool = False) -> None: argument
1985 def test_gather_backward_deterministic_path(self, device) -> None: argument
1990 def test_gather_backward_one_dim(self, device) -> None: argument
1995 def test_scatter_add_one_dim_deterministic(self, device) -> None: argument
2020 def test_scatter_zero_size_index(self, device) -> None: argument
2029 def test_sync_warning(self, device): argument
2089 def test_log_normal(self, device, dtype): argument
2096 def test_geometric(self, device, dtype): argument
2102 def test_repeat_interleave(self, device): argument
2131 def test_bernoulli_p(self, device, dtype): argument
2153 def test_bernoulli_self(self, device, dtype): argument
2181 def test_bernoulli_edge_cases(self, device, dtype): argument
2193 def test_exponential(self, device, dtype): argument
2208 def test_exponential_no_zero(self, device, dtype): argument
2216 def _generate_correlation_tensors(self, device, dtype): argument
2231 def test_corrcoef(self, device, dtype): argument
2239 def test_cov(self, device, dtype): argument
2259 def test_uniform_kstest(self, device, dtype): argument
2272 def test_normal_kstest(self, device, dtype): argument
2285 def test_lognormal_kstest(self, device, dtype): argument
2300 def test_exponential_kstest(self, device, dtype): argument
2312 def test_cauchy_kstest(self, device, dtype): argument
2324 def test_cauchy_no_inf(self, device, dtype): argument
2332 def test_cauchy(self, device, dtype): argument
2349 def test_geometric_kstest(self, device, dtype): argument
2360 def test_pairwise_distance_empty(self, device): argument
2374 def test_pdist_empty(self, device): argument
2387 def test_cdist_empty(self, device): argument
2412 def test_cdist_norm(self, device): argument
2430 def test_cdist_norm_batch(self, device): argument
2448 def test_cdist_cuda_backward(self, device): argument
2472 def test_cdist_large(self, device): argument
2483 def test_cdist_large_batch(self, device): argument
2493 def test_cdist_non_contiguous(self, device): argument
2521 def test_cdist_non_contiguous_batch(self, device): argument
2548 def test_cdist_euclidean_large(self, device): argument
2569 def test_cdist_grad_p_lt_1_no_nan(self, device): argument
2580 def test_cdist_same_inputs(self, device): argument
2597 def test_cumsum(self, device): argument
2648 def test_cumprod(self, device): argument
2700 def test_cummax_cummin(self, device): argument
2767 def test_logcumsumexp(self, device): argument
2841 def test_diff_noncontig(self, device, dtype): argument
2863 def test_diff(self, device, dtype): argument
2916 def test_gradient_all(self, device, dtype): argument
2970 def test_gradient_extreme_cases(self, device, dtype): argument
2991 def test_gradient_type_promotion(self, device): argument
3037 def test_gradient_spacing_list_length_error(self, device, dtype): argument
3074 def test_large_cumsum(self, device, dtype): argument
3087 def test_large_cumprod(self, device, dtype): argument
3097 def test_discontiguous_out_cumsum(self, device): argument
3122 def test_cummax_discontiguous(self, device): argument
3129 def test_cummin_discontiguous(self, device): argument
3135 def test_bool_tensor_value_change(self, device): argument
3142 def test_unfold_all_devices_and_dtypes(self, device): argument
3153 def test_unfold_scalars(self, device): argument
3163 def test_copy_all_dtypes_and_devices(self, device): argument
3175 def test_bfloat16_neg_abs(self, device): argument
3187 def test_reduced_type_float_copy(self, device, dtype): argument
3203 def test_copy_math_view(self, device): argument
3245 def test_copy_transpose_math_view(self, device, dtype): argument
3264 def test_clone_all_dtypes_and_devices(self, device): argument
3270 def test_clone_zero_stride_dim(self, device): argument
3286 def test_addcmul(self, device, dtype): argument
3296 def rand_tensor(size, dtype, device): argument
3327 def test_narrow_empty(self, device): argument
3335 def test_narrow_copy_non_contiguous(self, device): argument
3345 def test_index_reduce(self, device, dtype, reduce): argument
3394 def test_index_copy(self, device, dtype): argument
3429 def test_index_copy_scalars(self, device, dtype): argument
3441 def test_errors_index_copy(self, device): argument
3466 self, dim: int, device: torch.device argument
3481 def test_index_copy_deterministic(self, device: torch.device) -> None: argument
3501 def test_index_add_deterministic(self, device: torch.device) -> None: argument
3519 def test_index_put_non_accumulate_deterministic(self, device) -> None: argument
3540 def test_index_fill(self, device, dtype): argument
3558 def test_index_select(self, device, dtype): argument
3604 def test_take(self, device, dtype): argument
3639 def test_put(self, device, dtype): argument
3710 def test_put_accumulate(self, device, dtype): argument
3732 def test_take_empty(self, device): argument
3740 def test_put_empty(self, device): argument
3750 def scatter_allow_reduce(self, device, dtype, reduceop): argument
3757 def test_scatter_reduce_operations_to_large_input(self, device, dtype): argument
3784 def test_scatter_reduce_scalar(self, device, dtype): argument
3808 def test_scatter_add_non_unique_index(self, device): argument
3823 def test_scatter_reduce_non_unique_index(self, device, dtype): argument
3845 def test_scatter_reduce_multiply_unsupported_dtypes(self, device, dtype): argument
3855 def test_scatter_to_large_input(self, device): argument
3866 def test_scatter_add_to_large_input(self, device): argument
3877 def test_scatter_bool(self, device): argument
3886 def test_scatter_add_bool(self, device): argument
3897 def test_masked_scatter(self, device, dtype): argument
3943 def test_masked_scatter_bool_tensor(self, device): argument
3959 def test_masked_scatter_large_tensor(self, device): argument
3968 def test_masked_select(self, device, dtype): argument
4017 def test_masked_select_discontiguous(self, device): argument
4037 def test_masked_fill(self, device, dtypes): argument
4072 def test_masked_fill_bool_tensor(self, device): argument
4082 def test_tensor_shape_empty(self, device): argument
4141 def test_dim_function_empty(self, device): argument
4302 def test_pdist_norm_large(self, device): argument
4317 def test_addcdiv(self, device, dtype): argument
4327 def non_zero_rand(size, dtype, device): argument
4364 def test_nullary_op_mem_overlap(self, device): argument
4385 def test_ternary_op_mem_overlap(self, device, dtype): argument
4413 def test_copy_mem_overlap(self, device, dtype): argument
4424 def test_index_add_mem_overlap(self, device): argument
4441 def test_index_copy_mem_overlap(self, device): argument
4459 def test_index_fill_mem_overlap(self, device): argument
4473 def test_shift_mem_overlap(self, device): argument
4484 def test_bernoulli_mem_overlap(self, device): argument
4499 def test_put_mem_overlap(self, device): argument
4521 def test_index_put_mem_overlap(self, device): argument
4543 def test_masked_fill_mem_overlap(self, device): argument
4560 def test_masked_scatter_mem_overlap(self, device): argument
4571 def test_scatter_mem_overlap(self, device): argument
4585 def test_multinomial_device_constrain(self, device): argument
4692 def test_is_signed(self, device): argument
4746 def test_memory_format_preserved_after_permute(self, device): argument
4757 def test_memory_format_propagation_rules(self, device): argument
4800 def test_memory_format_empty_like(self, device): argument
4831 def test_memory_format_consistency(self, device): argument
4842 def test_memory_format_operators(self, device): argument
5007 def test_strides_propagation(self, device): argument
5054 def test_pin_memory_from_constructor(self, device): argument
5124 def test_lazy_clone(self, device, dtype): argument
5146 def test_lazy_clone_view(self, device, dtype): argument
5173 def test_lazy_clone_view_materialize(self, device, dtype): argument
5219 def test_lazy_clone_binary_op_no_materialize(self, device, dtype): argument
5233 def test_parallel_cow_materialize_error(self, device, dtype): argument
5272 def test_multinomial(self, device, dtype): argument
5365 def test_multinomial_deterministic(self, device, dtype): argument
5387 def test_multinomial_rng_state_advance(self, device, dtype): argument
5403 def _test_memory_format_transformations(self, device, input_generator_fn, transformation_fn, argument
5457 def test_memory_format_to(self, device): argument
5459 def input_generator_fn(device): argument
5474 def test_memory_format_type(self, device): argument
5476 def input_generator_fn(device): argument
5491 def test_memory_format_clone(self, device): argument
5493 def input_generator_fn(device): argument
5508 def test_memory_format_factory_like_functions_preserve(self, device): argument
5510 def input_generator_fn(device): argument
5533 def test_memory_format_type_shortcuts(self, device): argument
5535 def input_generator_fn(device): argument
5566 def test_memory_format_cpu_and_cuda_ops(self, device): argument
5568 def input_generator_fn(device): argument
5590 def test_pickle_gradscaler(self, device): argument
5629 def _test_multinomial_empty(self, device, replacement, num_samples): argument
5636 def test_multinomial_empty_w_replacement(self, device): argument
5641 def test_multinomial_empty_wo_replacement(self, device): argument
5647 def test_grad_scaling_unscale(self, device, dtype): argument
5743 def test_grad_scaling_update_scale(self, device, dtype): argument
5768 def test_grad_scaling_unscale_sparse(self, device, dtype): argument
5825 def test_grad_scaling_state_dict(self, device): argument
5852 …def _run_scaling_case(self, device, run, unskipped, skipped, atol=1e-7, optimizer_ctor=torch.optim… argument
5894 def test_grad_scaling_autocast(self, device, dtype, optim_info, foreach, fused): argument
5897 def run(device, data, model, optimizer, scaler, loss_fn, skip_iter, try_scaling_api): argument
5948 …def test_params_invalidated_with_grads_invalidated_between_unscale_and_step(self, device, dtype, o… argument
5977 def test_grad_scale_will_not_overflow(self, device): argument
5992 def test_grad_scaling_clipping(self, device): argument
5995 def run(device, data, model, optimizer, scaler, loss_fn, skip_iter, try_scaling_api): argument
6017 def test_grad_scaling_clipping_separate_unscale(self, device): argument
6020 def run(device, data, model, optimizer, scaler, loss_fn, skip_iter, try_scaling_api): argument
6044 def test_grad_scaling_penalty(self, device): argument
6047 def run(device, data, model, optimizer, scaler, loss_fn, skip_iter, try_scaling_api): argument
6081 def test_grad_scaling_accumulation(self, device): argument
6084 def run(device, data, model, optimizer, scaler, loss_fn, skip_iter, try_scaling_api): argument
6106 def test_grad_scaling_multiple(self, device): argument
6159 def test_grad_scaler_pass_itself(self, device): argument
6199 def test_grad_scaler_deprecated_warning(self, device): argument
6212 def test_multinomial_cpu(self, device, dtype): argument
6237 def test_where_scalar_handcrafted_values(self, device): argument
6285 def test_hook_remove(self, device): argument
6319 def test_skip_xla(self, device): argument
6328 def test_expected_failure_xla(self, device): argument
6335 def test_assertRaisesRegex_ignore_msg_non_native_device(self, device): argument
6348 def test_copy_(self, device, dtype): argument
6387 def test_item(self, device, dtype): argument
6394 def test_masked_scatter_inplace_noncontiguous(self, device): argument
6434 def test_index_add_bfloat16(self, device): argument
6448 def test_device_serialization(self, device): argument
6506 def test_from_sequence(self, device, dtype): argument
6578 def test_copy_broadcast(self, device) -> None: argument
6591 def test_clamp(self, device, dtype): argument
6632 def test_cuda_device_idx(self, device): argument
6877 def helper(dim, dtype, device, size_result, size_source): argument
9580 def get_expected_device_repr(device): argument