OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
473b17c4c1
pytorch/torch/utils
History
Edward Z. Yang 473b17c4c1 Run sympy expressions with Python values / FX tracing (#113978) 
To codegen deferred runtime asserts, I need to be able to convert sympy expressions back into regular Python expressions that I can put in FX graphs. This PR adds some of the machinery to do this: it adds a new sympy analysis that runs operations on all FX traceable operations that can also be run with plain Python int/float/bool/etc. It's tested by symbolic tracing through the analysis, and then testing that this traced graph gives the same result as running the Python analysis directly.

Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/113978
Approved by: https://github.com/aakhundov, https://github.com/lezcano
2023-11-20 21:25:11 +00:00
..
_sympy Run sympy expressions with Python values / FX tracing (#113978) 2023-11-20 21:25:11 +00:00
backcompat
benchmark [BE]: ruff apply rule PLW1510 to find silent subprocess errors (#113644) 2023-11-14 20:59:40 +00:00
bottleneck
data Add sparse tensors support to dataloader. (#112842) 2023-11-19 16:05:27 +00:00
hipify [ROCm] Supports ROCm6.0 reorganization and cleanup (#111486) 2023-11-16 18:37:12 +00:00
jit
model_dump
tensorboard upgrade lintrunner to the lowest supported versions on python 3.12 (#113562) 2023-11-15 18:12:01 +00:00
viz Fix docstring errors in _composable_state.py, remote_device.py, value_ranges.py, utils.py, run.py, rendezvous.py, launch.py, argparse_util.py, __init__.py, _cycles.py (#112953) 2023-11-08 01:13:09 +00:00
__init__.py Add torch.utils.deterministic.fill_uninitialized_memory flag (#111377) 2023-11-01 16:10:09 +00:00
_config_module.py [utils] move config_typing.pyi to torch.utils (#113929) 2023-11-17 18:51:57 +00:00
_config_typing.pyi [utils] move config_typing.pyi to torch.utils (#113929) 2023-11-17 18:51:57 +00:00
_content_store.py Fix recompilation issue with content store (#113533) 2023-11-13 23:58:13 +00:00
_contextlib.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
_cpp_extension_versioner.py
_cuda_trace.py
_cxx_pytree.py Revert "[pytree] register pytree node type in both C++ pytree and Python pytree (#112111)" 2023-11-10 17:24:40 +00:00
_device.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
_foreach_utils.py fix docstring issues in torch.utils (#113335) 2023-11-13 19:37:25 +00:00
_freeze.py [Codemod][python/main_function] caffe2: (#113357) 2023-11-15 22:17:31 +00:00
_mode_utils.py
_python_dispatch.py fix docstring issues in torch.utils (#113335) 2023-11-13 19:37:25 +00:00
_pytree.py Revert "[pytree] register pytree node type in both C++ pytree and Python pytree (#112111)" 2023-11-10 17:24:40 +00:00
_stats.py
_traceback.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
_triton.py [Inductor] Generalize inductor triton backend device agnostic (#109486) 2023-09-24 07:49:20 +00:00
_zip.py [Codemod][python/main_function] caffe2: (#113357) 2023-11-15 22:17:31 +00:00
backend_registration.py fix docstring issues in torch.utils (#113335) 2023-11-13 19:37:25 +00:00
bundled_inputs.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
checkpoint.py fix docstring issues in torch.utils (#113335) 2023-11-13 19:37:25 +00:00
collect_env.py fix docstring issues in torch.utils (#113335) 2023-11-13 19:37:25 +00:00
cpp_backtrace.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
cpp_extension.py Fixed error with cuda_ver in cpp_extension.py (#113555) 2023-11-14 00:12:22 +00:00
deterministic.py Add torch.utils.deterministic.fill_uninitialized_memory flag (#111377) 2023-11-01 16:10:09 +00:00
dlpack.py
file_baton.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
flop_counter.py fix docstring issues in torch.utils (#113335) 2023-11-13 19:37:25 +00:00
hooks.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
mkldnn.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
mobile_optimizer.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
model_zoo.py
show_pickle.py
throughput_benchmark.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00
weak.py Resolve docstring errors in throughput_benchmark.py, weak.py, _traceback.py, file_baton.py, _contextlib.py, _device.py, cpp_backtrace.py, bundled_inputs.py, run_cpu.py, hooks.py, mobile_optimizer.py, _freeze.py, __init__.py, mkldnn.py, dlpack.py (#113311) 2023-11-15 17:40:04 +00:00