pytorch/test
Peter Bell 9cdc80d581 [inductor] Fix torch.bernoulli decomposition return type (#115699)
Strangely enough, `torch.bernoulli` doesn't return a boolean and instead
it matches the output type of the inplace bernoulli.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/115699
Approved by: https://github.com/lezcano
ghstack dependencies: #115677
2023-12-13 14:16:31 +00:00
..
ao/sparsity Fix typo under test directory (#111304) 2023-10-16 23:06:06 +00:00
autograd Fix typo under test directory (#111304) 2023-10-16 23:06:06 +00:00
backends/xeon
benchmark_utils
bottleneck_test
cpp Add python and C++ support for LPPool3d (#114199) 2023-12-08 18:18:44 +00:00
cpp_api_parity Add python and C++ support for LPPool3d (#114199) 2023-12-08 18:18:44 +00:00
cpp_extensions Revert "Add function to materialize COW storages (#113396)" 2023-11-20 10:26:01 +00:00
custom_backend
custom_operator Update impl_abstract_pystub to be less boilerplatey (#113182) 2023-11-08 00:39:00 +00:00
distributed [DCP][BE] Apply ufmt to DCP and turn on lintrunner for DCP (#115302) 2023-12-13 10:32:36 +00:00
distributions Add Half support for poisson and use float for Half cumulative distribution on CPU (#112124) 2023-11-05 16:10:27 +00:00
dynamo [dynamo][BE] Move torchvision import inside of test_multi_import (#115677) 2023-12-13 14:16:31 +00:00
edge Adding event_tracer evalue logging calls in codegen (#114584) 2023-11-28 18:32:05 +00:00
error_messages
expect [sparse][semi-structured] add alg_id to _cslt_sparse_mm and _cslt_sparse_mm_search (#115178) 2023-12-11 23:08:51 +00:00
export Revert "[Export] Test non-strict mode on existing test cases (#115399)" 2023-12-13 12:59:09 +00:00
forward_backward_compatibility Fix backward for SDPA NT jagged layout (#115576) 2023-12-12 18:35:40 +00:00
functorch Migrated loss functions to ModuleInfos (#115584) 2023-12-12 22:20:20 +00:00
fx Fix typo under test directory (#112346) 2023-11-03 07:53:33 +00:00
inductor [inductor] Fix torch.bernoulli decomposition return type (#115699) 2023-12-13 14:16:31 +00:00
jit Add support for torch.Generator type in TorchScript (#110413) 2023-11-21 23:07:21 +00:00
jit_hooks
lazy [BE]: Enable RUF015 codebase wide (#115507) 2023-12-11 15:51:01 +00:00
mobile [TorchFix] Update old pretrained TorchVision API in tests (#111708) 2023-10-21 07:05:33 +00:00
nn Add Half support for AvgPool2d on CPU (#109578) 2023-12-12 12:59:47 +00:00
onnx Enable builtin tests for ONNX Export with ExportedProgram models (#114762) 2023-12-12 19:50:06 +00:00
onnx_caffe2
optim Move test_multi_tensor_optimizers to use OptimizerInfos (#114797) 2023-12-07 19:37:56 +00:00
package
profiler [PyTorch][ET] Collect Execution Traces in Chakra schema (#114753) 2023-11-30 04:07:11 +00:00
quantization [export] Preserve FQN in export_to_torch_ir (#115462) 2023-12-13 04:58:47 +00:00
scripts
test_img
torch_np [Dynamo][6.1/N] Refactor out TorchInGraphFunctionVariable and improve heuristic (#113432) 2023-12-09 05:11:44 +00:00
typing Add regression test for cuda_stream type checks (#111430) 2023-10-18 07:24:01 +00:00
_test_bazel.py
allowlist_for_publicAPI.json Remove dead tensor_has_hints. (#114071) 2023-11-20 16:02:24 +00:00
conftest.py
create_dummy_torchscript_model.py
delete.py
HowToWriteTestsUsingFileCheck.md
linear.py
load_torchscript_model.py
minioptest_failures_dict.json
mkl_verbose.py
mkldnn_verbose.py
run_doctests.sh
run_test.py [ez] Remove some args from run_test.py (#115459) 2023-12-11 19:56:37 +00:00
simulate_nccl_errors.py
test_ao_sparsity.py
test_autocast.py Add Half support for CPU autocast on eager mode (#112484) 2023-11-21 20:08:28 +00:00
test_autograd.py [BE]: Enable RUF015 codebase wide (#115507) 2023-12-11 15:51:01 +00:00
test_autograd_fallback.py
test_binary_ufuncs.py Check for boolean values as argument on pow function. (#114133) 2023-11-22 22:57:36 +00:00
test_bundled_images.py
test_bundled_inputs.py
test_comparison_utils.py
test_compile_benchmark_util.py
test_complex.py
test_content_store.py Fix recompilation issue with content store (#113533) 2023-11-13 23:58:13 +00:00
test_cpp_api_parity.py
test_cpp_extensions_aot.py
test_cpp_extensions_jit.py [BE] Remove test corner cases for CUDA older than supported 11.8 (#114989) 2023-12-04 21:41:03 +00:00
test_cpp_extensions_open_device_registration.py new_qtensor support privateuseone allocator. (#111464) 2023-11-01 05:16:58 +00:00
test_cuda.py [BE] Remove test corner cases for CUDA older than supported 11.8 (#114989) 2023-12-04 21:41:03 +00:00
test_cuda_expandable_segments.py
test_cuda_multigpu.py
test_cuda_nvml_based_avail.py
test_cuda_primary_ctx.py
test_cuda_sanitizer.py
test_cuda_trace.py
test_custom_ops.py fix error message in pytorch (#115349) 2023-12-07 19:27:29 +00:00
test_dataloader.py [BE]: Apply FURB145 to make code more readable and idiomatic. (#112990) 2023-11-06 13:15:04 +00:00
test_datapipe.py
test_decomp.py Deprecate primTorch module, replace it with decompositions in module Owners (#114754) 2023-11-29 18:27:20 +00:00
test_deploy.py
test_determination.py
test_dispatch.py
test_dlpack.py
test_dynamic_shapes.py Add is_integer to SymFloat (#114703) 2023-12-07 23:23:53 +00:00
test_expanded_weights.py
test_fake_tensor.py Add Stateful/Stateless symbolic contexts, use fresh fake mode for dynamo backends (#113926) (#114526) 2023-11-26 23:40:32 +00:00
test_flop_counter.py Updated flop counter to accept pytree inputs/outputs (#111990) 2023-10-26 01:25:27 +00:00
test_foreach.py [BE]: Enable RUF015 codebase wide (#115507) 2023-12-11 15:51:01 +00:00
test_function_schema.py
test_functional_autograd_benchmark.py
test_functional_optim.py
test_functionalization.py Skip aliasing correction for lift_fresh. (#112202) 2023-11-03 20:46:30 +00:00
test_functionalization_of_rng_ops.py
test_futures.py
test_fx.py [export] make UnflattenedModule not inherit from GraphModule (#115408) 2023-12-11 22:15:21 +00:00
test_fx_experimental.py
test_fx_passes.py
test_fx_reinplace_pass.py
test_hub.py
test_import_stats.py
test_indexing.py
test_itt.py
test_jit.py fix error message in pytorch (#115349) 2023-12-07 19:27:29 +00:00
test_jit_autocast.py
test_jit_disabled.py
test_jit_fuser.py
test_jit_fuser_legacy.py
test_jit_fuser_te.py
test_jit_legacy.py
test_jit_llga_fuser.py
test_jit_profiling.py
test_jit_simple.py
test_jit_string.py
test_jiterator.py
test_kernel_launch_checks.py
test_legacy_vmap.py
test_license.py
test_linalg.py [BE] Remove test corner cases for CUDA older than supported 11.8 (#114989) 2023-12-04 21:41:03 +00:00
test_logging.py
test_masked.py
test_maskedtensor.py
test_matmul_cuda.py Skip unit tests that fail on MI210 runners (#114613) 2023-11-27 22:25:35 +00:00
test_meta.py Deprecate primTorch module, replace it with decompositions in module Owners (#114754) 2023-11-29 18:27:20 +00:00
test_metal.py
test_mkl_verbose.py
test_mkldnn.py [TorchFix] Update old pretrained TorchVision API in tests (#111708) 2023-10-21 07:05:33 +00:00
test_mkldnn_fusion.py improve mkldnn_linear_pointwise performance for contiguous tensor with non default contiguous strides (#114939) 2023-12-01 23:30:07 +00:00
test_mkldnn_verbose.py
test_mobile_optimizer.py
test_model_dump.py
test_model_exports_to_core_aten.py Build and test ExecuTorch on PyTorch (#113364) 2023-11-16 02:19:58 +00:00
test_module_init.py Add python and C++ support for LPPool3d (#114199) 2023-12-08 18:18:44 +00:00
test_modules.py Migrated loss functions to ModuleInfos (#115584) 2023-12-12 22:20:20 +00:00
test_monitor.py
test_mps.py Add Half support for AvgPool2d on CPU (#109578) 2023-12-12 12:59:47 +00:00
test_multiprocessing.py
test_multiprocessing_spawn.py
test_namedtensor.py Add some checks about Device and Layout when create/convert named tensor (#113628) 2023-11-14 20:40:27 +00:00
test_namedtuple_return_api.py
test_native_functions.py
test_native_mha.py
test_nestedtensor.py Fix SDPA for SAM (#115636) 2023-12-12 18:52:38 +00:00
test_nn.py Fix _load_from_state_dict for num_batches_tracked in batchnorm (#115285) 2023-12-07 22:48:26 +00:00
test_nnapi.py
test_numba_integration.py
test_numpy_interop.py Fix NULL dereference in binary CPU ops (#115183) 2023-12-06 03:37:47 +00:00
test_openmp.py
test_ops.py Add decomposition for torch.block_diag (#115096) 2023-12-11 20:04:22 +00:00
test_ops_fwd_gradients.py s390x: skip tests relying on specific openblas precision (#112843) 2023-11-07 17:18:20 +00:00
test_ops_gradients.py
test_ops_jit.py
test_optim.py Move test_multi_tensor_optimizers to use OptimizerInfos (#114797) 2023-12-07 19:37:56 +00:00
test_out_dtype_op.py
test_overrides.py
test_package.py
test_per_overload_api.py
test_prims.py Deprecate primTorch module, replace it with decompositions in module Owners (#114754) 2023-11-29 18:27:20 +00:00
test_proxy_tensor.py Expand dynamic dims support for traceable subclasses (#114311) 2023-12-05 21:09:25 +00:00
test_pruning_op.py [dynamo] Eagerly install guards (#111415) 2023-11-07 19:55:19 +00:00
test_public_bindings.py [DeviceMesh] Rename _device_mesh.py to device_mesh.py to prepare for beta (#115099) (#115193) 2023-12-08 08:44:32 +00:00
test_python_dispatch.py Allow functionalization to work with optional mutable (#114803) 2023-11-30 23:48:03 +00:00
test_pytree.py [pytree] hardcode values for none_is_leaf and namespace in C++ pytree (#114858) 2023-12-01 15:01:33 +00:00
test_quantization.py enable cat for cuda bits types (#115044) 2023-12-06 00:05:18 +00:00
test_reductions.py Add Half support for cummax, cummin, cumprod, logcumsumexp, and prod on CPU (#112132) 2023-11-05 12:31:38 +00:00
test_scatter_gather_ops.py
test_schema_check.py
test_segment_reductions.py
test_serialization.py Make Float8 types serializeable (#114662) 2023-11-29 23:23:23 +00:00
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py
test_show_pickle.py
test_sort_and_select.py Fix buffer overflow in torch.sort (#111672) 2023-10-21 02:30:11 +00:00
test_sparse.py Enable test_sparse_addmm for crossref tests (#115536) 2023-12-12 17:26:40 +00:00
test_sparse_csr.py Add tune_bsr_dense_addmm as an API to find optimal triton kernel parameters for bsr_dense_addmm (#115499) 2023-12-12 16:44:51 +00:00
test_sparse_semi_structured.py [sparse][semi-structured] add alg_id to _cslt_sparse_mm and _cslt_sparse_mm_search (#115178) 2023-12-11 23:08:51 +00:00
test_spectral_ops.py
test_stateless.py
test_static_runtime.py Skip test_fork_wait_4 and test_fork_wait_4_async (#112743) 2023-11-02 20:46:29 +00:00
test_subclass.py
test_sympy_utils.py Run sympy expressions with Python values / FX tracing (#113978) 2023-11-20 21:25:11 +00:00
test_tensor_creation_ops.py Remove unecessary warning when getting storage.filename (#113212) 2023-11-08 02:09:59 +00:00
test_tensorboard.py
test_tensorexpr.py
test_tensorexpr_pybind.py
test_testing.py Print the index and summary of the SampleInput that failed an OpInfo test (#99444) 2023-11-21 23:08:35 +00:00
test_throughput_benchmark.py
test_torch.py Terminate handler (#101332) 2023-12-12 17:55:27 +00:00
test_transformers.py [cpu] flash attention optimization (#115151) 2023-12-12 01:09:55 +00:00
test_type_hints.py
test_type_info.py
test_type_promotion.py Disallow fp8 type promotion (#113975) 2023-11-20 19:47:43 +00:00
test_typing.py [BE] Revive test_typing (#111428) 2023-10-18 02:19:49 +00:00
test_unary_ufuncs.py
test_utils.py Move test_utils.py back to MYPY (#113745) 2023-11-16 01:57:58 +00:00
test_view_ops.py
test_vulkan.py
test_weak.py [BE]: Enable RUF015 codebase wide (#115507) 2023-12-11 15:51:01 +00:00
test_xnnpack_integration.py