Home
last modified time | relevance | path

Searched defs:CUDACachingHostAllocatorImpl (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/cuda/
DCachingHostAllocator.cpp71 struct CUDACachingHostAllocatorImpl struct
74 void allocate_host_memory(size_t size, void** ptr) override { in allocate_host_memory()
96 void free_block(Block* block) override { in free_block()
107 void record_stream( in record_stream()
115 bool query_event(EventPool::Event& event) override { in query_event()
126 EventPool::Event create_event_internal(DeviceIndex idx) { in create_event_internal()
132 TaskThreadPool* getThreadPool() { in getThreadPool()
139 void mapPagesForRegister( in mapPagesForRegister()
159 void registerPages(const void* ptr, size_t size) { in registerPages()
174 void allocWithCudaHostRegister(void** ptr, size_t roundSize) { in allocWithCudaHostRegister()