pytorch/torch/utils
Pian Pawakapan 0c7856973b [export] enumerate unsupported sympy.Functions (#134271) (#134598)
Summary:
There's 2 concepts of unsupported sympy.Functions in symbolic_shapes:
1) unsupported by the export solver, meaning the solver doesn't know how to provide useful fixes for those functions
2) unsupported by the sympy interpreter - meaning we can't reify them into FX nodes because the functions aren't present in PythonReferenceAnalysis

This splits the current call into a call for each version, with the Export solver the only user of 1). For 1), we enumerate the functions in _sympy/functions.py, and subtract the functions we know we can support. For 2) there's only 3 functions we've seen pop up in test cases.

cc jgong5 mingfeima XiaobingSuper sanchitintel ashokei jingxu10

Differential Revision: D61863394

Pulled By: pianpwk

Pull Request resolved: https://github.com/pytorch/pytorch/pull/134598
Approved by: https://github.com/angelayi
2024-08-28 00:34:38 +00:00
..
_strobelight [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_sympy [export] enumerate unsupported sympy.Functions (#134271) (#134598) 2024-08-28 00:34:38 +00:00
backcompat
benchmark
bottleneck
data [BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200) 2024-08-15 15:50:19 +00:00
hipify [Reland2] Update NVTX to NVTX3 (#109843) 2024-08-20 16:33:26 +00:00
jit
model_dump
tensorboard
viz [Memory Snapshot][Viz] Add Allocator Settings Tab (#132518) 2024-08-13 17:35:12 +00:00
__init__.py
_backport_slots.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_config_module.py allow SubConfigProxy of arbitrary depth (#133418) 2024-08-14 18:43:00 +00:00
_config_typing.pyi
_content_store.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_contextlib.py
_cpp_extension_versioner.py
_cxx_pytree.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_device.py Fix DeviceContext bug (#133729) 2024-08-20 07:14:37 +00:00
_exposed_in.py Revert "[BE] typing for decorators - _library/custom_ops (#131578)" 2024-07-28 03:29:32 +00:00
_foreach_utils.py
_freeze.py [BE] mypy: disallow untyped decorators (#131428) 2024-07-23 21:50:55 +00:00
_get_clean_triton.py Fix edge case in inductor triton clean script (#130837) 2024-08-19 23:46:11 +00:00
_import_utils.py
_mode_utils.py
_ordered_set.py [BE] Fix MYPY issues (#133872) 2024-08-20 16:12:04 +00:00
_python_dispatch.py [BE] typing for decorators - fx/_compatibility (part 1) (#134202) 2024-08-22 17:07:33 +00:00
_pytree.py [pytree] Only import optree if it's used (#131478) 2024-07-24 00:10:49 +00:00
_stats.py
_thunk.py Refactor thunkify to return proper thunk abstraction (#132407) 2024-08-06 02:35:45 +00:00
_traceback.py
_triton.py Remove AMD restrictions on triton hashing (#133616) 2024-08-16 08:02:48 +00:00
_typing_utils.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py Update AC pass use_reentrant message (#134472) 2024-08-26 21:57:38 +00:00
collect_env.py
cpp_backtrace.py
cpp_extension.py Fix Extension attribute name in CppExtension example (#134046) 2024-08-21 13:58:16 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py [BE][Ez]: FURB142,FURB92 misc preview fixes (#133880) 2024-08-21 13:54:51 +00:00
hooks.py
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py Move module_tracker to logging for confused hierarchy (#134467) 2024-08-26 19:39:08 +00:00
show_pickle.py
throughput_benchmark.py
weak.py