search

cosdt / torch_backend

https://cosdt.github.io/torch_backend/
Other
3 stars 0 forks source link

refactor cache host allocator #17

Closed Zhenbin-8 closed 3 months ago

Zhenbin-8 commented 3 months ago
  1. extract event pool to be device-agnostic;
  2. inherit at::CachingHostAllocatorImpl to reuse host block caching mechanism.
FFFrog commented 3 months ago

@Zhenbin-8 文件名、还有内部类名修改一下，类似“NPUCachingHostAllocator”就可以