Eddie Yan e64a814ae7 [CUDA] Add experimental green context support for SM carveout (#159104) ... Low-level PyTorch APIs should be usable/stable enough at this point but we might move the underlying driver API usage a bit from here... Built on top of @drisspg 's branch Pull Request resolved: https://github.com/pytorch/pytorch/pull/159104 Approved by: https://github.com/ngimel, https://github.com/malfet, https://github.com/kwen2501 Co-authored-by: drisspg <drisspguessous@gmail.com> Co-authored-by: Nikita Shulga <2453524+malfet@users.noreply.github.com>		2025-10-22 21:38:52 +00:00
..
_internal	[CUDA] Add experimental green context support for SM carveout (#159104)	2025-10-22 21:38:52 +00:00
__init__.py	Add pyrefly suppressions 2/n (#164513)	2025-10-03 02:46:13 +00:00
_comparison.py	[1/N] Add strict parameter to Python zip calls (#165531)	2025-10-18 05:26:33 +00:00
_creation.py	[BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552)	2025-08-07 00:09:56 +00:00
_utils.py