406 struct CudaMallocAsyncAllocator : public CUDAAllocator {  struct
407   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()
587   void recordStream(const DataPtr& ptr, cuda::CUDAStream stream) override {  in recordStream()
610   ShareableHandle shareIpcHandle(void* handle) override {  in shareIpcHandle()
617   std::shared_ptr<void> getIpcDevPtr(std::string handle) override {  in getIpcDevPtr()
624   void recordHistory(  in recordHistory()
635   void attachOutOfMemoryObserver(OutOfMemoryObserver observer) override {  in attachOutOfMemoryObserver()
642   void attachAllocatorTraceTracker(AllocatorTraceTracker tracker) override {  in attachAllocatorTraceTracker()
649   std::shared_ptr<AllocatorState> getCheckpointState(  in getCheckpointState()
658   CheckpointDelta setCheckpointPoolState(  in setCheckpointPoolState()
669   DeviceStats getDeviceStats(c10::DeviceIndex device) override {  in getDeviceStats()
733   void resetAccumulatedStats(c10::DeviceIndex device) override {  in resetAccumulatedStats()
739   void resetPeakStats(c10::DeviceIndex device) override {  in resetPeakStats()
757   SnapshotInfo snapshot() override {  in snapshot()
769   void beginAllocateToPool(  in beginAllocateToPool()
782   void endAllocateToPool(c10::DeviceIndex device, MempoolId_t mempool_id)  in endAllocateToPool()
817   void releasePool(c10::DeviceIndex device, MempoolId_t mempool_id) override {  in releasePool()
832   void* raw_alloc(size_t nbytes) override {  in raw_alloc()
843   void* raw_alloc_with_stream(size_t nbytes, cudaStream_t stream) override {  in raw_alloc_with_stream()
853   void raw_delete(void* ptr) override {  in raw_delete()
856   void enablePeerAccess(c10::DeviceIndex dev, c10::DeviceIndex dev_to_access)  in enablePeerAccess()
872   cudaError_t memcpyAsync(  in memcpyAsync()
886   std::string name() override {  in name()
889   void copy_data(void* dest, const void* src, std::size_t count) const final {  in copy_data()