pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00

Files

Yu, Guangye 5f0030ba63 [xpu][fix] Support xpu custom raw_alloc/delete in caching allocator (#168957 )

# Motivation
Memory Pool needs to support the custom `raw_alloc` and `raw_delete` from a custom allocator.

# Solution
When the custom allocator is provided in the memory pool, use its `raw_alloc` and `raw_delete`. Otherwise, use the `sycl::aligned_alloc_device` and `sycl::free` from SYCL runtime.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/168957
Approved by: https://github.com/EikanWang, https://github.com/gujinghui
ghstack dependencies: #168956

2025-12-01 16:47:05 +00:00

impl

…

test

[BE] fix typos in c10/ (#156078 )

2025-06-18 10:24:44 +00:00

CMakeLists.txt

Move XPUEvent to c10 (#158336 )

2025-11-12 11:29:42 +00:00

XPUCachingAllocator.cpp

[xpu][fix] Support xpu custom raw_alloc/delete in caching allocator (#168957 )

2025-12-01 16:47:05 +00:00