pytorch/torch/utils
eellison 16aaff7783 Fix mm pad regresion - more conservative estimation of plannable inputs (#128909)
- More conservative estimation of plannable inputs
- Consider constant_pad_nd as pointwise node in concat lowering
- Use aten.cat instead of constant pad ndwhen padding just a single dimension because it can be memory-planned away

Pull Request resolved: https://github.com/pytorch/pytorch/pull/128909
Approved by: https://github.com/Chillee
2024-07-18 16:42:30 +00:00
..
_strobelight
_sympy Keep zero check be compatible with different sympy versions (#130729) 2024-07-16 08:39:00 +00:00
backcompat
benchmark [BE] enable UFMT for torch/storage.py (#127706) 2024-06-27 23:16:24 +00:00
bottleneck
data [BE][Easy] fix ruff rule needless-bool (SIM103) (#130206) 2024-07-14 08:17:52 +00:00
hipify [ROCm] Add int4 support (#129710) 2024-07-09 19:49:12 +00:00
jit
model_dump [BE][Easy] replace import pathlib with from pathlib import Path (#129426) 2024-06-30 01:36:07 +00:00
tensorboard
viz
__init__.py [BE] enable UFMT in torch.utils.data (#127705) 2024-06-27 23:16:24 +00:00
_config_module.py Fix mm pad regresion - more conservative estimation of plannable inputs (#128909) 2024-07-18 16:42:30 +00:00
_config_typing.pyi Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419) 2024-06-29 09:23:39 +00:00
_content_store.py
_contextlib.py
_cpp_extension_versioner.py
_cxx_pytree.py Revert "[BE] bump optree version to 0.12.1 (#130139)" 2024-07-15 19:42:11 +00:00
_device.py
_exposed_in.py
_foreach_utils.py
_freeze.py
_get_clean_triton.py
_import_utils.py
_mode_utils.py
_python_dispatch.py typing fake_tensor.py (#128041) 2024-07-13 06:07:40 +00:00
_pytree.py
_stats.py
_traceback.py
_triton.py [inductor] Better messaging when triton version is too old (#130403) 2024-07-10 23:45:50 +00:00
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
collect_env.py
cpp_backtrace.py
cpp_extension.py [BE][Easy] fix ruff rule needless-bool (SIM103) (#130206) 2024-07-14 08:17:52 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py [cuDNN][SDPA] Remove TORCH_CUDNN_SDPA_ENABLED=1, enable cuDNN SDPA by default on H100 and 2nd on other archs >= sm80 (#125343) 2024-06-30 19:22:16 +00:00
hooks.py
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py
show_pickle.py
throughput_benchmark.py
weak.py