1 #pragma once 2 3 #include <c10/hip/HIPCachingAllocator.h> 4 #include <ATen/hip/impl/HIPAllocatorMasqueradingAsCUDA.h> 5 #include <ATen/hip/impl/HIPStreamMasqueradingAsCUDA.h> 6 7 namespace c10 { 8 // forward declaration 9 class DataPtr; 10 namespace hip { 11 namespace HIPCachingAllocatorMasqueradingAsCUDA { 12 13 C10_HIP_API Allocator* get(); 14 C10_HIP_API void recordStreamMasqueradingAsCUDA(const DataPtr& ptr, HIPStreamMasqueradingAsCUDA stream); 15 16 } // namespace HIPCachingAllocatorMasqueradingAsCUDA 17 } // namespace hip 18 } // namespace c10 19