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
19db4f6014
pytorch/torch/_dynamo/variables
History
rzou 19db4f6014 [capture_triton] fix special kwargs path (#132143) 
I didn't test this path when creating the orchestrator. This PR fixes
that path to work in the capture_triton path. The problem is that we are
handling a value that is an int (in the capture_triton path) and a
ConstantVariable (in the Dynamo triton path) so we abstract that out in
the orchestrator.

Test Plan:
- new tests

Pull Request resolved: https://github.com/pytorch/pytorch/pull/132143
Approved by: https://github.com/oulgen
2024-07-30 20:30:40 +00:00
..
__init__.py Let dynamo inline functional_call (#128646) 2024-07-30 14:22:23 +00:00
base.py Remove mypy ignore from torch/_dynamo/variables/lazy.py (#131785) 2024-07-28 17:13:53 +00:00
builder.py [dynamo] lazily guard and specialize on the symint when used in f-string. (#131529) 2024-07-25 16:16:34 +00:00
builtin.py [dynamo] Support dict unpack of MutableMapping objects (#131961) 2024-07-30 05:49:58 +00:00
constant.py [dynamo] add lazy IteratorVariable implementations for map and zip (#131413) 2024-07-26 10:47:38 +00:00
ctx_manager.py Relax constraints for creating a GenericContextWrappingVariable (#129091) 2024-07-29 15:40:59 +00:00
dicts.py [dynamo] Support dict unpack of MutableMapping objects (#131961) 2024-07-30 05:49:58 +00:00
distributed.py Annotate all InstructionTranslator (#131509) 2024-07-24 23:45:53 +00:00
functions.py [capture_triton] fix special kwargs path (#132143) 2024-07-30 20:30:40 +00:00
higher_order_ops.py Let dynamo inline functional_call (#128646) 2024-07-30 14:22:23 +00:00
iter.py [dynamo] make more unpack_var_sequence calls forced (#132069) 2024-07-30 02:30:08 +00:00
lazy.py Remove mypy ignore from torch/_dynamo/variables/lazy.py (#131785) 2024-07-28 17:13:53 +00:00
lists.py [dynamo] add lazy IteratorVariable implementations for map and zip (#131413) 2024-07-26 10:47:38 +00:00
misc.py Annotate all InstructionTranslator (#131509) 2024-07-24 23:45:53 +00:00
nn_module.py [dynamo] Do not guard on keys for _forward_hooks and _forward_pre_hooks (#131682) 2024-07-26 04:39:54 +00:00
optimizer.py Annotate all InstructionTranslator (#131509) 2024-07-24 23:45:53 +00:00
script_object.py [BE] mypy: disallow untyped decorators (#131428) 2024-07-23 21:50:55 +00:00
sdpa.py Revert "Grouped Query Attention (#128898)" 2024-07-30 13:11:24 +00:00
tensor.py dynamo: mutations on .data should be invisible to autograd (#131403) 2024-07-26 14:22:20 +00:00
torch_function.py Annotate all InstructionTranslator (#131509) 2024-07-24 23:45:53 +00:00
torch.py Annotate all InstructionTranslator (#131509) 2024-07-24 23:45:53 +00:00
user_defined.py [dynamo] make more unpack_var_sequence calls forced (#132069) 2024-07-30 02:30:08 +00:00