Searched defs:is_cuda_malloc (Results 1 – 2 of 2) sorted by relevance
195 CircularPool::CircularPool(int max_size_in_gb, int arena_size, bool is_cuda_malloc) in CircularPool()203 bool createOneArena, bool is_cuda_malloc) { in CreateCircularPool()
278 Arena::Arena(size_t val_in_MB, bool is_cuda_malloc) in Arena()281 Status Arena::CreateArena(std::shared_ptr<Arena> *p_ba, size_t val_in_MB, bool is_cuda_malloc) { in CreateArena()