pytorch/caffe2
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
..
core Record the XPU and XCCL build settings in the compiled binary (#147161) 2025-05-20 09:21:39 +00:00
perfkernels Enable all PIE rules on ruff (#165814) 2025-10-18 07:36:18 +00:00
serialize [AOTI] raise PyTorchStreamWriter open failed error code on windows (#162799) 2025-09-13 01:41:14 +00:00
utils [2/N] Fix clang-tidy readability checks (#164652) 2025-10-06 01:06:01 +00:00
.clang-format [BE][clang-format] make macro PyObject_HEAD_INIT(type) and PyVarObject_HEAD_INIT(type, size) have its own line (#136949) 2024-10-02 18:39:22 +00:00
CMakeLists.txt [CUDA] Add experimental green context support for SM carveout (#159104) 2025-10-22 21:38:52 +00:00