Oguz Ulgen defa0d3a2d Add a side table for triton kernels to avoid using itertools.partial (#110633) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/110633 Approved by: https://github.com/jansel		2023-10-08 02:01:59 +00:00
..
__init__.py	[HigherOrderOp] expose torch.cond (#110293)	2023-10-07 20:39:52 +00:00
cond.py	[HigherOrderOp] expose torch.cond (#110293)	2023-10-07 20:39:52 +00:00
out_dtype.py	[HigherOrderOp] fallthrough some keys by default. (#110478)	2023-10-05 16:25:42 +00:00
triton_kernel_wrap.py	Add a side table for triton kernels to avoid using itertools.partial (#110633)	2023-10-08 02:01:59 +00:00
utils.py	Higher order operator util for raising if inputs require grads (#106078)	2023-08-01 00:13:13 +00:00
wrap.py	torch.compile + selective activation checkpointing (#105489)	2023-09-21 16:24:11 +00:00