Banit Agrawal c4bbc6433e [PyTorch CCA] Add an API to get expandable segment sizes (#163771) ...

Summary: This diffs add an API to query expandable segment size for each stream so that we can use this info to warmup the segment in advance, so we dont incur any performance penalty during steady state inference for new CUDA memory allocations.

Differential Revision: D76447308

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163771
Approved by: https://github.com/bbus

2025-10-01 02:16:58 +00:00

..

benchmark

Set RUNPATH so installed tests can find the required shared libraries (#136627)

2024-10-25 09:38:08 +00:00

core

Migrate DeviceType to torch/headeronly (#163999)

2025-09-30 23:13:27 +00:00

cuda

[PyTorch CCA] Add an API to get expandable segment sizes (#163771)

2025-10-01 02:16:58 +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] Add igamma/igammac ops (#161927)

2025-09-02 20:52:02 +00:00

mobile

[2/N] Fix extra warnings brought by clang-tidy-17 (#137459)

2024-10-08 19:05:02 +00:00

test

Add -Wno-ctad-maybe-unsupported compiler flag (#162223)

2025-09-06 06:11:37 +00:00

util

Combine strong and weak refcounts in intrusive_ptr in a single refcount (#163394)

2025-09-22 17:53:28 +00:00

xpu

Add uuid to XPU device properties (#161392)

2025-09-02 06:41:32 +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

remove //c10:headers (#98420)

2023-04-05 19:33:10 +00:00

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