OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
4bdecd94ea
pytorch/torch/_higher_order_ops
History
Yuanhao Ji d7eb3a492c [Typing] Enable torch.types.IntLikeType / FloatLikeType / BoolLikeType (#152157) 
### Changes

Replace `Union[SymInt, int]` and `Union[int, SymInt]` with `IntLikeType`.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/152157
Approved by: https://github.com/Skylion007
2025-04-25 19:00:10 +00:00
..
__init__.py Revert "[map] always turn on dynamo for map (#150962)" 2025-04-14 21:09:22 +00:00
_invoke_quant.py [BaseHOP] change hop(subgraph, operands) to hop(subgraph, *operands) (#146730) 2025-02-20 02:30:36 +00:00
aoti_call_delegate.py [export] Make aoti_call_delegate hop traceable (#148804) 2025-04-03 20:44:31 +00:00
associative_scan.py [HOP] Reworked DispatchKey.Autograd (#151107) 2025-04-15 19:55:46 +00:00
auto_functionalize.py Mark auto_functionalized HOPs as cacheable (#151194) 2025-04-14 20:05:32 +00:00
base_hop.py [HOP] Reworked DispatchKey.Autograd (#151107) 2025-04-15 19:55:46 +00:00
cond.py [HOP] Reworked DispatchKey.Autograd (#151107) 2025-04-15 19:55:46 +00:00
effects.py Fix with effect lowering for list return type (#149510) 2025-03-19 19:35:08 +00:00
executorch_call_delegate.py [HOP] Reworked DispatchKey.Autograd (#151107) 2025-04-15 19:55:46 +00:00
flat_apply.py [dynamo] Make nonstrict_trace work with some pytree.register_constant-ed instances (#148007) 2025-03-05 21:28:26 +00:00
flex_attention.py [HOP] Reworked DispatchKey.Autograd (#151107) 2025-04-15 19:55:46 +00:00
foreach_map.py [BaseHOP] change hop(subgraph, operands) to hop(subgraph, *operands) (#146730) 2025-02-20 02:30:36 +00:00
hints_wrap.py [HOP] Reworked DispatchKey.Autograd (#151107) 2025-04-15 19:55:46 +00:00
invoke_subgraph.py [invoke_subgraph] Compile time traces (#151409) 2025-04-24 13:20:50 +00:00
map.py [map] defer importing AOTConfig and create_joint dependency (#151479) 2025-04-23 19:16:40 +00:00
out_dtype.py [HOP] Reworked DispatchKey.Autograd (#151107) 2025-04-15 19:55:46 +00:00
run_const_graph.py [HOP] Reworked DispatchKey.Autograd (#151107) 2025-04-15 19:55:46 +00:00
scan.py [HOP] Reworked DispatchKey.Autograd (#151107) 2025-04-15 19:55:46 +00:00
schema.py move find_hop_schema into _higher_order_ops/schema.py (#151147) 2025-04-23 18:26:37 +00:00
strict_mode.py [HOP] Reworked DispatchKey.Autograd (#151107) 2025-04-15 19:55:46 +00:00
torchbind.py [HOP] Reworked DispatchKey.Autograd (#151107) 2025-04-15 19:55:46 +00:00
triton_kernel_wrap.py [Typing] Enable torch.types.IntLikeType / FloatLikeType / BoolLikeType (#152157) 2025-04-25 19:00:10 +00:00
utils.py [hop] support base_hop._gen_schema (#149688) 2025-04-09 16:42:55 +00:00
while_loop.py [HOP] Reworked DispatchKey.Autograd (#151107) 2025-04-15 19:55:46 +00:00
wrap.py Require that all HOPs be imported at import torch time (#145939) 2025-01-29 22:27:52 +00:00