Searched defs:CudaMallocAsyncAllocator (Results 1 – 1 of 1) sorted by relevance
406 struct CudaMallocAsyncAllocator : public CUDAAllocator { struct407 DataPtr allocate(size_t size) override { in allocate()421 DeleterFnPtr raw_deleter() const override { in raw_deleter()428 void init(int dev_count) override { in init()444 bool initialized() override { in initialized()448 static inline void assertValidDevice(c10::DeviceIndex device) { in assertValidDevice()453 void setMemoryFraction(double fraction, c10::DeviceIndex device) override { in setMemoryFraction()484 void emptyCache() override { in emptyCache()499 void cacheInfo(c10::DeviceIndex device, size_t* maxWorkspaceGuess) override { in cacheInfo()574 void* getBaseAllocation(void* ptr, size_t* size) override { in getBaseAllocation()[all …]