pytorch/torch/_dynamo/variables
Yidi Wu 2ec95ffe57 [cond] support unbacked symbool inputs (#133589)
Fixes https://github.com/pytorch/pytorch/issues/133577.

In dynamo, when received an unbacked symbool input, we create an unbacked symint to replace it.

The alternative approach of `not realizing the pred LazyVariable in cond` doesn't work because we need to get the proxy of the symbool input.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/133589
Approved by: https://github.com/ezyang
2024-08-19 23:36:48 +00:00
..
__init__.py [dynamo] support random.Random (#133725) 2024-08-19 22:34:44 +00:00
base.py [dynamo] revert map/zip iterator related changes (#132528) 2024-08-04 18:46:55 +00:00
builder.py [cond] support unbacked symbool inputs (#133589) 2024-08-19 23:36:48 +00:00
builtin.py [dynamo] simplify implementation for builtins.sum (#133779) 2024-08-19 22:14:34 +00:00
constant.py [easy][dynamo] Add tx as an arg in getitem_const (#132899) 2024-08-07 21:35:41 +00:00
ctx_manager.py [Dynamo] Support torch.cuda.device ctx manager (#133385) 2024-08-16 17:05:55 +00:00
dicts.py [BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200) 2024-08-15 15:50:19 +00:00
distributed.py Revert "[dtensor] move DTensor to public namespace (#133113)" 2024-08-19 05:00:19 +00:00
functions.py [RFC][dynamo] add decorator to register polyfill for unsupported C++ function to avoid graph break (#133712) 2024-08-19 22:14:33 +00:00
higher_order_ops.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
iter.py [dynamo] revert map/zip iterator related changes (#132528) 2024-08-04 18:46:55 +00:00
lazy.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
lists.py [dynamo] Support object creation of classes with custom __new__ (#132977) 2024-08-16 03:09:23 +00:00
misc.py [dynamo] support random.Random (#133725) 2024-08-19 22:34:44 +00:00
nn_module.py Revert #132806, #132736, #132539, #132487 (#133570) 2024-08-15 20:54:21 +00:00
optimizer.py [Inductor UT] Reuse inductor UT for intel GPU test/inductor/test_compiled_optimizers.py (#133083) 2024-08-17 01:15:26 +00:00
script_object.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
sdpa.py Grouped Query Attention (#132689) 2024-08-07 05:35:36 +00:00
tensor.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
torch_function.py Fix _dynamo.variables.torch_function.global_mangled_class_name (#132744) 2024-08-09 22:19:01 +00:00
torch.py Revert "[dtensor] move DTensor to public namespace (#133113)" 2024-08-19 05:00:19 +00:00
user_defined.py [dynamo] support random.Random (#133725) 2024-08-19 22:34:44 +00:00