viable/strict/1787807055: Move expandable_segments() out of the header (#194872)
PyTorch PR #194872 将 CUDAAllocatorConfig::expandable_segments() 从内联头文件移出，因为该函数依赖 PYTORCH_C10_DRIVER_API_SUPPORTED 宏，该宏在 c10/cuda/CMakeLists.txt 中仅对 c10_cuda 私有定义。内联导致外部翻译单元编译时宏未定义，函数返回 false，导致 CUDACachingAllocatorReserveDeviceTest 失败。移出后宏检查仅在定义它的编译单元中执行，无需将宏设为 PUBLIC。
Development
- First Reportviable/strict/1787807055: Move expandable_segments() out of the header (#194872)PyTorch Core
- Industry Responsetrunk/88bb71a5c62743868032d53b6ac9291520642396: Move expandable_segments() out of the header (#194872)PyTorch Core
- Current AssessmentPyTorch 作为主流深度学习框架，其内存分配器的行为直接影响训练和推理性能。expandable_segments 是 CUDA 缓存分配器的重要特性，该修复确保外部用户能正确启用和检测该特性，有助于提升框架的稳定性和可预测性。Agent Pulse · analysis
PyTorch 的 PR #194872 将 CUDAAllocatorConfig::expandable_segments() 从头文件内联实现移出。该函数依赖 PYTORCH_C10_DRIVER_API_SUPPORTED 宏，该宏在 c10/cuda/CMakeLists.txt 中仅对 c10_cuda 私有定义。由于函数是内联的，每个包含该头文件的翻译单元都会重新编译，但宏未定义，导致外部消费者看到无条件 false，而库内部看到真实值。CUDACachingAllocatorReserveDeviceTest 通过 setAllocatorSettings 启用 expandable_segments 并断言该函数，但外部调用永远无法通过。将函数体移出内联后，宏检查仅在定义它的编译单元中执行，无需将宏设为 PUBLIC。测试计划使用 gcc-15 和 g++-15 构建并运行测试。PR 由 Claude Code 辅助起草，已由 Skylion007 批准。
该变更揭示了内联函数与编译时宏在跨编译单元传播时的可见性问题。将依赖私有宏的函数移出头文件，可以避免宏泄漏到公共 API，同时确保行为一致。这提示开发者：在头文件中内联函数时，若依赖编译时条件，需考虑宏的可见性范围，否则可能导致外部调用与库内部行为不一致。
PyTorch 作为主流深度学习框架，其内存分配器的行为直接影响训练和推理性能。expandable_segments 是 CUDA 缓存分配器的重要特性，该修复确保外部用户能正确启用和检测该特性，有助于提升框架的稳定性和可预测性。
该修复提升了 PyTorch 的稳定性和可用性，减少因宏可见性导致的功能失效问题，降低用户排查成本，增强开发者对框架的信任，间接支持 PyTorch 生态的商业应用。
该修复可能促使 PyTorch 进一步审查其他内联函数对私有宏的依赖，并可能推动相关测试覆盖。未来可关注 expandable_segments 在更多场景下的默认启用或性能优化。