pytorch/c10
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
..
benchmark Set RUNPATH so installed tests can find the required shared libraries (#136627) 2024-10-25 09:38:08 +00:00
core Move toUnderlying to headeronly (#165694) 2025-10-22 05:31:16 +00:00
cuda [CUDA] Add experimental green context support for SM carveout (#159104) 2025-10-22 21:38:52 +00:00
hip Use torch_compile_options for c10 libraries (#147821) 2025-03-18 01:54:23 +00:00
macros Move c10/macros/Macros.h to headeronly (#158365) 2025-07-16 18:46:52 +00:00
metal [MPS] sparse matmuls (#165232) 2025-10-18 09:04:42 +00:00
mobile [2/N] Fix extra warnings brought by clang-tidy-17 (#137459) 2024-10-08 19:05:02 +00:00
test Fix AllocatorConfig parse roundup division bug (#165304) 2025-10-19 15:34:44 +00:00
util Simplify c10::guts::apply (#164566) 2025-10-22 00:47:43 +00:00
xpu Move allocation size config to AllocatorConfig for cross-allocator sharing (#159553) 2025-10-22 01:48:56 +00:00
BUCK.oss Migrate c10/macros/cmake_macros.h.in to torch/headeronly (#158035) 2025-07-15 19:52:59 +00:00
BUILD.bazel
build.bzl Remove non-header-only dep from c10_headers target (#155858) 2025-06-16 21:41:25 +00:00
CMakeLists.txt Migrate c10/macros/cmake_macros.h.in to torch/headeronly (#158035) 2025-07-15 19:52:59 +00:00
ovrsource_defs.bzl Revert "Consistently use c10_ovrsource in arvr mode everywhere (#164128)" 2025-09-30 14:43:52 +00:00