Daniele Trifirò 3183d65ac0 use shutil.which in _find_cuda_home (#126060) ... Replace `subprocess.check_output` call with `shutil.which`, similarly to how this is done in `_find_rocm_home` Pull Request resolved: https://github.com/pytorch/pytorch/pull/126060 Approved by: https://github.com/r-barnes		2024-05-13 17:38:17 +00:00
..
_sympy	Compute bounds for the variables created during codegen (#123100)	2024-05-08 08:14:06 +00:00
backcompat
benchmark	Adding Compare in torch.utils.benchmark documentation (#125009)	2024-05-03 00:50:54 +00:00
bottleneck
data	Fix broken docs (#124940)	2024-04-26 19:24:52 +00:00
hipify	[ROCm] enforce ROCM_VERSION >= 6.0 (#125646)	2024-05-12 18:01:28 +00:00
jit	Fix global flake8 issues (#124771)	2024-04-26 15:35:53 +00:00
model_dump
tensorboard	[BE]: Update ruff to v0.4.4 (#125031)	2024-05-12 20:02:37 +00:00
viz
__init__.py
_config_module.py
_config_typing.pyi
_content_store.py
_contextlib.py
_cpp_extension_versioner.py
_cxx_pytree.py
_device.py
_exposed_in.py
_foreach_utils.py
_freeze.py
_get_clean_triton.py	Add script for removing Inductor dependencies from Inductor generated code (#125811)	2024-05-10 00:00:25 +00:00
_import_utils.py
_mode_utils.py
_python_dispatch.py	[RELAND] Switch default behavoir of export IR to be predispatch (#125860)	2024-05-10 17:36:53 +00:00
_pytree.py
_stats.py
_traceback.py
_triton.py
_typing_utils.py
_zip.py
backend_registration.py	add methods for open device in PackedSequence module. (#124923)	2024-04-26 15:26:20 +00:00
bundled_inputs.py
checkpoint.py	make torch.amp.autocast more generic (#125103)	2024-05-08 12:13:26 +00:00
collect_env.py
cpp_backtrace.py
cpp_extension.py	use shutil.which in _find_cuda_home (#126060)	2024-05-13 17:38:17 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py	Add "accurate" FlopCounter implementations for NestedTensor SDPA kernels (#125776)	2024-05-10 19:49:37 +00:00
hooks.py	[FSDP] mark pre_backward_hook unserializable (#125464)	2024-05-06 20:20:31 +00:00
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py	More accurate is_bw and prompt parents cleanup for ModuleTracker utils (#125634)	2024-05-07 20:57:36 +00:00
show_pickle.py
throughput_benchmark.py
weak.py