set nullptr to memory pool element if it failed during initialization.
This commit is contained in:
parent
ce4df17e42
commit
bd56886fd6
1 changed files with 2 additions and 0 deletions
|
@ -5853,6 +5853,8 @@ void ggml_init_cublas() {
|
||||||
size_t treshold = UINT64_MAX;
|
size_t treshold = UINT64_MAX;
|
||||||
CUDA_CHECK(cudaMemPoolSetAttribute(g_cudaMemPools[id], cudaMemPoolAttrReleaseThreshold, &treshold));
|
CUDA_CHECK(cudaMemPoolSetAttribute(g_cudaMemPools[id], cudaMemPoolAttrReleaseThreshold, &treshold));
|
||||||
fprintf(stderr, ", CUDA memory pool is supported\n");
|
fprintf(stderr, ", CUDA memory pool is supported\n");
|
||||||
|
} else {
|
||||||
|
g_cudaMemPools[id] = nullptr;
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
fprintf(stderr, ", CUDA memory pool is not supported\n");
|
fprintf(stderr, ", CUDA memory pool is not supported\n");
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue