OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
ceb07656c2
pytorch/torch/_dynamo/variables
History
Chen, Zejun ceb07656c2 [dynamo] use APIs to use device interface instead of raw object in dynamo capture (#113000) 
This PR makes up for the https://github.com/pytorch/pytorch/pull/108312.
This PR uses the _get_registered_device_interfaces_ to get the device interface, instead of using raw objects.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/113000
Approved by: https://github.com/jgong5, https://github.com/jansel
2023-11-08 01:45:00 +00:00
..
__init__.py [dynamo] Add LazyVariableTracker (#111306) 2023-11-07 19:55:19 +00:00
base.py [dynamo] Remove VariableTracker.propagate (#111726) 2023-11-07 19:55:19 +00:00
builder.py [dynamo] use APIs to use device interface instead of raw object in dynamo capture (#113000) 2023-11-08 01:45:00 +00:00
builtin.py [dynamo] Remove VariableTracker.propagate (#111726) 2023-11-07 19:55:19 +00:00
constant.py [dynamo] Remove VariableTracker.propagate (#111726) 2023-11-07 19:55:19 +00:00
ctx_manager.py [dynamo] Remove VariableTracker.propagate (#111726) 2023-11-07 19:55:19 +00:00
dicts.py [dynamo] Avoid eager imports of classes with custom VariableTrackers (#112319) 2023-11-07 22:45:54 +00:00
distributed.py [dynamo] Remove VariableTracker.propagate (#111726) 2023-11-07 19:55:19 +00:00
functions.py [dynamo] Remove VariableTracker.propagate (#111726) 2023-11-07 19:55:19 +00:00
higher_order_ops.py [dynamo] Eagerly install guards (#111415) 2023-11-07 19:55:19 +00:00
iter.py [dynamo] Remove VariableTracker.add_options (#111725) 2023-11-07 19:55:19 +00:00
lazy.py [dynamo] Remove VariableTracker.propagate (#111726) 2023-11-07 19:55:19 +00:00
lists.py [dynamo] Remove VariableTracker.propagate (#111726) 2023-11-07 19:55:19 +00:00
misc.py [dynamo] Remove incorrect sources (#112961) 2023-11-07 22:01:40 +00:00
nn_module.py [dynamo] Remove VariableTracker.propagate (#111726) 2023-11-07 19:55:19 +00:00
optimizer.py [dynamo] Eagerly install guards (#111415) 2023-11-07 19:55:19 +00:00
tensor.py [dynamo] Remove VariableTracker.propagate (#111726) 2023-11-07 19:55:19 +00:00
torch_function.py [dynamo] Eagerly install guards (#111415) 2023-11-07 19:55:19 +00:00
torch.py [dynamo] use APIs to use device interface instead of raw object in dynamo capture (#113000) 2023-11-08 01:45:00 +00:00
user_defined.py [dynamo] Avoid eager imports of classes with custom VariableTrackers (#112319) 2023-11-07 22:45:54 +00:00