Searched refs:memStack_ (Results 1 – 1 of 1) sorted by relevance
544 MemoryStack* memStack_; member in __anon81e7630e0511::StackAllocator548 StackAllocator::StackAllocator(cudaStream_t stream) : stream_(stream), memStack_(0) in StackAllocator()553 memStack_ = initializer.getMemoryPool(deviceId)->getFreeMemStack(); in StackAllocator()563 if (memStack_ != 0) in ~StackAllocator()564 memStack_->pool->returnMemStack(memStack_); in ~StackAllocator()577 if (memStack_ == 0) in allocate()594 uchar* ptr = memStack_->requestMemory(memSize); in allocate()608 if (memStack_ == 0) in free()611 memStack_->returnMemory(mat->datastart); in free()