Guilherme Leobas 1be6a070bc Add support for torch.cond in vmap (#114523) ... Fixes: https://github.com/pytorch/pytorch/issues/114136 Patch enables conversion of a BatchedTensor into FakeTensor and write torch.cond vmap support using torch.where Pull Request resolved: https://github.com/pytorch/pytorch/pull/114523 Approved by: https://github.com/zou3519		2023-12-20 19:54:38 +00:00
..
__init__.py	[Dynamo][8/N] Wrap itertools.* as ItertoolsVariable (#115802)	2023-12-16 01:42:02 +00:00
base.py	Remove replace_all and make VTs mutable (#113725)	2023-12-10 09:31:21 +00:00
builder.py	Revert "[Dynamo][9/N] Make SkipFilesVariable wrap functions only (#115963)"	2023-12-20 12:06:55 +00:00
builtin.py	add support for dynamic shapes in round (#115259)	2023-12-19 15:45:50 +00:00
constant.py	[dynamo] Improve support for dynamic shapes str.format and _assert (#115203)	2023-12-06 04:54:45 +00:00
ctx_manager.py	[contextlib] Wrapping a function with set_grad_enabled will consume its global mutation (#113359)	2023-11-09 19:16:20 +00:00
dicts.py	Remove replace_all and make VTs mutable (#113725)	2023-12-10 09:31:21 +00:00
distributed.py	[DeviceMesh] Rename _device_mesh.py to device_mesh.py to prepare for beta (#115099) (#115193)	2023-12-08 08:44:32 +00:00
functions.py	Adds allreduce to inductor remap (#115950)	2023-12-18 22:00:22 +00:00
higher_order_ops.py	Add support for torch.cond in vmap (#114523)	2023-12-20 19:54:38 +00:00
iter.py	[Dynamo][8/N] Wrap itertools.* as ItertoolsVariable (#115802)	2023-12-16 01:42:02 +00:00
lazy.py	[dynamo] Remove VariableTracker.propagate (#111726)	2023-11-07 19:55:19 +00:00
lists.py	Remove replace_all and make VTs mutable (#113725)	2023-12-10 09:31:21 +00:00
misc.py	Revert "[Dynamo][9/N] Make SkipFilesVariable wrap functions only (#115963)"	2023-12-20 12:06:55 +00:00
nn_module.py	Remove replace_all and make VTs mutable (#113725)	2023-12-10 09:31:21 +00:00
optimizer.py	Remove replace_all and make VTs mutable (#113725)	2023-12-10 09:31:21 +00:00
tensor.py	Add support for torch.cond in vmap (#114523)	2023-12-20 19:54:38 +00:00
torch_function.py	Ensure wrapping subclasses with as_subclass is supported (#116091)	2023-12-20 14:37:08 +00:00
torch.py	[Dynamo][7/N] Wrap python modules under torch as regular PythonModuleVariable (#115724)	2023-12-13 21:23:14 +00:00
user_defined.py	Revert "[Dynamo][9/N] Make SkipFilesVariable wrap functions only (#115963)"	2023-12-20 12:06:55 +00:00