chilli 9bccafc31c Made FlexAttention rewrite getitem calls to use aten.index in score_mod (#124799) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/124799 Approved by: https://github.com/drisspg ghstack dependencies: #124444		2024-04-26 01:02:28 +00:00
..
__init__.py	[dynamo] Improve constant-prop for regex/torch.__version__ (#123705)	2024-04-12 19:03:13 +00:00
base.py	[BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261)	2024-04-17 19:29:34 +00:00
builder.py	[dynamo][cpp-guards] Differentiate dict guards wrt to guarding on key order (#124779)	2024-04-25 08:20:35 +00:00
builtin.py	[dynamo][cpp-guards] Differentiate dict guards wrt to guarding on key order (#124779)	2024-04-25 08:20:35 +00:00
constant.py	[BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261)	2024-04-17 19:29:34 +00:00
ctx_manager.py	[dynamo] Support warnings.catch_warnings (#123511)	2024-04-08 22:27:46 +00:00
dicts.py	[dynamo] Add support for DELETE_SUBSCR (#123526)	2024-04-25 22:07:24 +00:00
distributed.py	Don't create world pg variable out of thin air when rewriting c10d collectives (#122561)	2024-03-26 20:12:08 +00:00
functions.py	[BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261)	2024-04-17 19:29:34 +00:00
higher_order_ops.py	Made FlexAttention rewrite getitem calls to use aten.index in score_mod (#124799)	2024-04-26 01:02:28 +00:00
iter.py	[dynamo] Improve constant-prop for regex/torch.__version__ (#123705)	2024-04-12 19:03:13 +00:00
lazy.py	[dynamo] Replace VariableTracker.apply with visit/realize_all (#122218)	2024-03-20 07:53:18 +00:00
lists.py	[Dynamo] Fix NamedTuple hasattr bug (#124531)	2024-04-21 04:36:22 +00:00
misc.py	Introduce set_example_value and use it throughout Dynamo (#124176)	2024-04-17 22:57:11 +00:00
nn_module.py	[dynamo][easy] forbid_in_graph check to use getattr_static (#124445)	2024-04-20 14:11:05 +00:00
optimizer.py	Defer marking_static_address (#124309)	2024-04-19 17:20:57 +00:00
sdpa.py	[dynamo] Refactor reconstruct() not to return anything (#120150)	2024-02-17 17:13:41 +00:00
tensor.py	Introduce set_example_value and use it throughout Dynamo (#124176)	2024-04-17 22:57:11 +00:00
torch_function.py	[dynamo] Optimize SourcelessBuilder (#122063)	2024-03-19 04:23:30 +00:00
torch.py	[dynamo] support object.__setattr__(obj, name, value) (#124068)	2024-04-17 15:57:14 +00:00
user_defined.py	[dynamo][cpp-guards] Differentiate dict guards wrt to guarding on key order (#124779)	2024-04-25 08:20:35 +00:00