Thomas Bohnstingl e889252493 Implementation of scan (#134102) ... This operation is supposed to be the pendant to the `associative_scan`, but can operate with non-associative functions. @ydwu4 Pull Request resolved: https://github.com/pytorch/pytorch/pull/134102 Approved by: https://github.com/ydwu4		2024-09-10 04:51:16 +00:00
..
__init__.py	[dynamo] reland map/zip iterator related changes (#135074)	2024-09-06 20:38:02 +00:00
base.py	[dynamo] reland map/zip iterator related changes (#135074)	2024-09-06 20:38:02 +00:00
builder.py	Also handle compiler collective when input variable doesn't exist on all ranks (#135147)	2024-09-06 13:18:36 +00:00
builtin.py	[dynamo] reland map/zip iterator related changes (#135074)	2024-09-06 20:38:02 +00:00
constant.py	[dynamo] reland map/zip iterator related changes (#135074)	2024-09-06 20:38:02 +00:00
ctx_manager.py	[Dynamo] Support torch.cuda.device ctx manager (#133385)	2024-08-16 17:05:55 +00:00
dicts.py	[Dynamo] Fix Huggingface PretrainedConfig get non const attr (#135413)	2024-09-07 19:16:29 +00:00
distributed.py	[reland][dtensor] move DTensor to public namespace (#134203)	2024-09-08 17:08:40 +00:00
functions.py	Implement VariableTracker.python_type() (#134215)	2024-09-05 16:35:47 +00:00
higher_order_ops.py	Implementation of scan (#134102)	2024-09-10 04:51:16 +00:00
iter.py	[dynamo] reland map/zip iterator related changes (#135074)	2024-09-06 20:38:02 +00:00
lazy.py	Add basic mypy annotations to dynamo (#132415)	2024-08-04 18:43:36 +00:00
lists.py	[dynamo] reland map/zip iterator related changes (#135074)	2024-09-06 20:38:02 +00:00
misc.py	Implement VariableTracker.python_type() (#134215)	2024-09-05 16:35:47 +00:00
nn_module.py	[fbode-testing][dynamo][reland][inline-inbuilt-nn-modules] Mark attri… (#134136)	2024-08-22 17:54:58 +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	Implement VariableTracker.python_type() (#134215)	2024-09-05 16:35:47 +00:00
torch_function.py	[Dynamo] Support pop torch function mode stack (#133131)	2024-08-20 07:14:42 +00:00
torch.py	[Dynamo] Adding CallFunctionNoArgsSource and (#135425)	2024-09-09 22:46:00 +00:00
user_defined.py	[dynamo] reland map/zip iterator related changes (#135074)	2024-09-06 20:38:02 +00:00