Searched refs:MallocFuncType (Results 1 – 3 of 3) sorted by relevance
14 using MallocFuncType = void*(size_t, int, cudaStream_t); variable52 std::function<MallocFuncType> alloc_fn,72 std::function<MallocFuncType> alloc_fn,165 std::function<MallocFuncType> alloc_fn_;
46 std::function<MallocFuncType> alloc_fn, in CUDAPluggableAllocator()378 std::function<MallocFuncType> alloc_fn, in createCustomAllocator()
1201 std::function<MallocFuncType> malloc_fn = in registerCudaPluggableAllocator()1203 reinterpret_cast<MallocFuncType*>(malloc_ptr); in registerCudaPluggableAllocator()