saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test
History
zdevito d91b49dbaa expandable_segments <-> other allocator options (#134338) 
Previously setting  garbage_collection_threshold or max_split_size_mb along with expandable_segments:True could cause the allocator to hit assert failures when running nearly out of memory. This PR ensures garbage_collection and max_split freeing do not accidentally try to release expandable segments.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/134338
Approved by: https://github.com/ezyang
2024-08-29 18:43:59 +00:00
..
ao/sparsity Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
autograd [BE][Easy][7/19] enforce style for empty lines in import segments in test/[a-c]*/ and test/[q-z]*/ (#129758) 2024-07-31 10:54:03 +00:00
backends/xeon
benchmark_utils
bottleneck_test Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
cpp [AOTI] Fix test_aoti_inference CPU build issue (#134675) 2024-08-28 17:42:19 +00:00
cpp_api_parity Fix failures when default is flipped for weights_only (#127627) 2024-08-16 00:22:43 +00:00
cpp_extensions Add device daemon (#131814) 2024-08-27 23:32:07 +00:00
custom_backend
custom_operator Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
distributed [4/N] Test NaN checker against broadcast (#134701) 2024-08-29 17:00:07 +00:00
distributions Revert "Temp disable MKL in DistributionKernels.cpp (#132532)" 2024-08-06 20:57:09 +00:00
dynamo [inductor] calibration inductor windows uts (15/N) (#134586) 2024-08-29 16:18:40 +00:00
dynamo_expected_failures [dynamo] Support reading attributes from pybind objects (#134630) 2024-08-29 15:06:52 +00:00
dynamo_skips Add fallback() to torch.library (#131707) 2024-07-27 18:02:35 +00:00
edge [Reland] [11/N] Use std::nullopt and std::optional (#132622) 2024-08-05 20:36:33 +00:00
error_messages
expect Add mean.dtype_out (#133506) 2024-08-26 19:26:11 +00:00
export [export] use single FQN for param_buffer_mapping (#134500) 2024-08-29 17:06:31 +00:00
forward_backward_compatibility
functorch [hop] require hops to override __call__. (#134352) 2024-08-28 19:56:40 +00:00
fx [export] Temporarily bypass torch_fn in partitioner (#134292) 2024-08-24 05:50:18 +00:00
higher_order_ops [aotd] Support HOP effects in backward (#132638) 2024-08-23 15:30:58 +00:00
inductor Changed masked out rows logsumexp to be -inf and not zero (#134650) 2024-08-29 17:22:52 +00:00
jit Fix failures when default is flipped for weights_only (#127627) 2024-08-16 00:22:43 +00:00
jit_hooks [BE][Easy][13/19] enforce style for empty lines in import segments in test/j*/ (#129764) 2024-08-01 12:13:42 +00:00
lazy Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
mobile Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
nn Fix failures when default is flipped for weights_only (#127627) 2024-08-16 00:22:43 +00:00
onnx [ONNX] Update exporter logic (#134304) 2024-08-24 00:49:54 +00:00
optim Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
package Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
profiler [Profiler] Add test to make sure FunctionEvents are processed lazily (#134359) 2024-08-24 23:03:19 +00:00
quantization Update pt2e numeric debugger to use node.meta["custom"] field (#134040) 2024-08-27 19:51:03 +00:00
scripts [BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200) 2024-08-15 15:50:19 +00:00
test_img
torch_np [test/torch_np] Fix usages of deprecated NumPy 2.0 APIs in numpy_tests (#131909) 2024-08-05 16:21:08 +00:00
typing [BE][Easy][7/19] enforce style for empty lines in import segments in test/[a-c]*/ and test/[q-z]*/ (#129758) 2024-07-31 10:54:03 +00:00
xpu Populate submodules of torch._C to sys.modules recursively (#132216) 2024-08-08 10:20:25 +00:00
_test_bazel.py
allowlist_for_publicAPI.json Revert "[dtensor] move DTensor to public namespace (#133113)" 2024-08-19 05:00:19 +00:00
conftest.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
create_dummy_torchscript_model.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
delete.py
hi.py
HowToWriteTestsUsingFileCheck.md
linear.py
load_torchscript_model.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
minioptest_failures_dict.json
mkl_verbose.py
mkldnn_verbose.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
pytest_shard_custom.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
run_doctests.sh
run_test.py Exclude test_transformers and unit tests which require recent GPU arch (#132895) 2024-08-27 20:40:53 +00:00
simulate_nccl_errors.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
slow_tests.json Move slow tests to be in repo (#132379) 2024-08-07 18:42:56 +00:00
test_ao_sparsity.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_autocast.py using new device-agnostic api instead of old api like torch.cpu or torch.cuda (#134448) 2024-08-28 01:01:49 +00:00
test_autograd.py [Dynamo][autograd.Function] Support mark_non_differentiable (#134087) 2024-08-28 08:12:37 +00:00
test_autograd_fallback.py
test_autoload.py
test_binary_ufuncs.py Fix type promotion for ldexp (#133519) 2024-08-16 01:26:26 +00:00
test_bundled_images.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_bundled_inputs.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
test_ci_sanity_check_fail.py
test_comparison_utils.py
test_compile_benchmark_util.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_complex.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_content_store.py
test_cpp_api_parity.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_cpp_extensions_aot.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_cpp_extensions_jit.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_cpp_extensions_mtia_backend.py
test_cpp_extensions_open_device_registration.py Add torch.serialization.skip_data context manager (#134504) 2024-08-29 04:52:52 +00:00
test_cpp_extensions_stream_and_event.py Re-implement pin_memory to be device-agnostic by leveraging the Accelerator concept (#126376) 2024-07-23 01:44:15 +00:00
test_cuda.py expandable_segments <-> other allocator options (#134338) 2024-08-29 18:43:59 +00:00
test_cuda_expandable_segments.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_cuda_multigpu.py Allow torch.cuda.memory.mem_get_info to take a device str argument with an unspecified device index. (#132616) 2024-08-06 13:19:46 +00:00
test_cuda_nvml_based_avail.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_cuda_primary_ctx.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_cuda_sanitizer.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
test_cuda_trace.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_custom_ops.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
test_dataloader.py Support IPC for Expandable Segments (#130890) 2024-08-05 18:48:13 +00:00
test_datapipe.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_decomp.py Add a private _safe_softmax (#131060) 2024-08-08 23:09:38 +00:00
test_deploy.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_determination.py
test_dispatch.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
test_dlpack.py
test_dynamic_shapes.py remove dynamic_dim (#134211) 2024-08-23 04:13:03 +00:00
test_expanded_weights.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_fake_tensor.py [Fake tensor] Align the appearance of device_put op in fx_graph generated for CUDA and XPU, which is exposed in the issue #130823 (#132479) 2024-08-09 05:31:00 +00:00
test_flop_counter.py [NJT][flop counter] attention: if offsets are fake, use max seqlen (#132356) 2024-08-02 20:42:29 +00:00
test_foreach.py TestForeach::test_parity: Remove check for error message text (#134251) 2024-08-26 22:40:54 +00:00
test_function_schema.py
test_functional_autograd_benchmark.py
test_functional_optim.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_functionalization.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
test_functionalization_of_rng_ops.py
test_futures.py
test_fx.py [fx] Implement deepcopy for Proxy (#133706) 2024-08-22 16:37:30 +00:00
test_fx_experimental.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_fx_passes.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_fx_reinplace_pass.py Allow SymInt input for torch.fx reinplace pass (#133178) 2024-08-13 20:07:17 +00:00
test_hub.py
test_import_stats.py
test_indexing.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
test_itt.py
test_jit.py [EZ] Restore test_unicode_comments (#134589) 2024-08-27 21:51:06 +00:00
test_jit_autocast.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_jit_disabled.py
test_jit_fuser.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_jit_fuser_legacy.py
test_jit_fuser_te.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
test_jit_legacy.py
test_jit_llga_fuser.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_jit_profiling.py
test_jit_simple.py
test_jit_string.py
test_jiterator.py
test_kernel_launch_checks.py
test_legacy_vmap.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
test_license.py
test_linalg.py [reland][ROCm] TunableOp for gemm_and_bias (#128919) 2024-08-22 18:27:50 +00:00
test_logging.py
test_masked.py
test_maskedtensor.py Revert "Add MaskedTensor support to *_like API (#128637)" 2024-08-29 01:42:52 +00:00
test_matmul_cuda.py [CUTLASS][FP8] Skip scaled_mm rowwise test on sm89 (#133612) 2024-08-15 23:43:30 +00:00
test_meta.py [TEST] Fix _scaled_mm tests (#130897) 2024-07-18 02:15:00 +00:00
test_metal.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_mkl_verbose.py
test_mkldnn.py Add oneDNN support for Half LSTM on CPU (#132607) 2024-08-29 03:40:10 +00:00
test_mkldnn_fusion.py improve mkldnn_linear_pointwise_binary performance for contiguous tensor with non default contiguous strides (#132019) 2024-07-30 05:02:38 +00:00
test_mkldnn_verbose.py
test_mobile_optimizer.py [BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200) 2024-08-15 15:50:19 +00:00
test_model_dump.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_model_exports_to_core_aten.py [BE]: Update ruff to 0.6.0 (#133609) 2024-08-16 14:11:01 +00:00
test_module_tracker.py Move module_tracker to logging for confused hierarchy (#134467) 2024-08-26 19:39:08 +00:00
test_modules.py Fix failures when default is flipped for weights_only (#127627) 2024-08-16 00:22:43 +00:00
test_monitor.py [pytorch][counters] Pybind for WaitCounter (#132357) 2024-08-02 16:08:10 +00:00
test_mps.py [MPS] Fix SDP training (#134719) 2024-08-29 01:28:53 +00:00
test_multiprocessing.py Support IPC for Expandable Segments (#130890) 2024-08-05 18:48:13 +00:00
test_multiprocessing_spawn.py Create processes in parallel in mp.start_processes for forkserver (#134629) 2024-08-28 21:34:32 +00:00
test_namedtensor.py
test_namedtuple_return_api.py
test_native_functions.py
test_native_mha.py
test_nestedtensor.py [NJT] Support NJT SDPA + meta-device flop counting (#134289) 2024-08-29 03:43:42 +00:00
test_nn.py Use newer toAccumulateType signature in Normalization.cpp (#134540) 2024-08-27 18:09:20 +00:00
test_nnapi.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_numba_integration.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_numpy_interop.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
test_openmp.py
test_ops.py Move attention kernels back from fake_impls to meta_registrations (#134288) 2024-08-27 21:10:36 +00:00
test_ops_fwd_gradients.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_ops_gradients.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_ops_jit.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_optim.py move param's device check to _init_group for fused (#131153) 2024-08-17 04:49:47 +00:00
test_out_dtype_op.py
test_overrides.py Fix test flakiness due to not resetting state (#134058) 2024-08-21 11:54:08 +00:00
test_package.py
test_per_overload_api.py
test_prims.py properly register conjugate/neg fallthroughs to prim ops (#132699) 2024-08-06 17:57:04 +00:00
test_proxy_tensor.py Only thunkify proxies in some situations (#132421) 2024-08-08 12:03:06 +00:00
test_pruning_op.py
test_public_bindings.py [ONNX] New export logic leveraging ExportedProgram and ONNX IR (#132530) 2024-08-21 01:08:42 +00:00
test_python_dispatch.py Fix failures when default is flipped for weights_only (#127627) 2024-08-16 00:22:43 +00:00
test_pytree.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
test_quantization.py Rename generate_numeric_debug_handle to numeric_debugger (#130590) 2024-07-15 22:42:27 +00:00
test_reductions.py fix torch.prod vectorized path for bool (#128009) 2024-08-28 05:27:50 +00:00
test_scatter_gather_ops.py
test_schema_check.py
test_segment_reductions.py
test_serialization.py Add torch.serialization.skip_data context manager (#134504) 2024-08-29 04:52:52 +00:00
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_show_pickle.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_sort_and_select.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
test_sparse.py [sparse] enable meta tests (#133379) 2024-08-14 21:58:23 +00:00
test_sparse_csr.py SparseCsrCUDA: cuDSS backend for linalg.solve (#129856) 2024-08-22 07:57:30 +00:00
test_sparse_semi_structured.py [sparse] Update cuSPARSELt to v0.6.2 (#134022) 2024-08-23 19:34:53 +00:00
test_spectral_ops.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_stateless.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_static_runtime.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_subclass.py Fix failures when default is flipped for weights_only (#127627) 2024-08-16 00:22:43 +00:00
test_sympy_utils.py Keep zero check be compatible with different sympy versions (#130729) 2024-07-16 08:39:00 +00:00
test_tensor_creation_ops.py
test_tensorboard.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_tensorexpr.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_tensorexpr_pybind.py [BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200) 2024-08-15 15:50:19 +00:00
test_testing.py [BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200) 2024-08-15 15:50:19 +00:00
test_throughput_benchmark.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_torch.py [CUDA] is_bf16_supported() should not crash if there are no GPUs (#132313) 2024-08-02 02:50:43 +00:00
test_transformers.py [CPU][flash attention] make the stride of output align with input (#134656) 2024-08-29 16:04:25 +00:00
test_type_hints.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_type_info.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_type_promotion.py
test_typing.py [BE][Easy][6/19] enforce style for empty lines in import segments in test/ (#129757) 2024-07-17 06:42:37 +00:00
test_unary_ufuncs.py
test_utils.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_utils_internal.py Create a JustknobConfig for use in config (#134161) 2024-08-27 16:07:33 +00:00
test_view_ops.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
test_vulkan.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_weak.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
test_xnnpack_integration.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_xpu.py Add new prop to _XpuDevicePropertie for triton gemm optimization (#131738) 2024-08-18 08:32:30 +00:00