Enable CUDA graphs on volta+turing (#25749)
This commit is contained in:
@@ -3982,7 +3982,7 @@ static bool ggml_cuda_graph_set_enabled(ggml_backend_cuda_context * cuda_ctx, co
|
|||||||
ggml_cuda_graph * graph = cuda_ctx->cuda_graph(graph_key);
|
ggml_cuda_graph * graph = cuda_ctx->cuda_graph(graph_key);
|
||||||
|
|
||||||
if (graph->graph == nullptr) {
|
if (graph->graph == nullptr) {
|
||||||
if (ggml_cuda_info().devices[cuda_ctx->device].cc < GGML_CUDA_CC_AMPERE) {
|
if (ggml_cuda_info().devices[cuda_ctx->device].cc < GGML_CUDA_CC_VOLTA) {
|
||||||
if (!graph->disable_due_to_gpu_arch) {
|
if (!graph->disable_due_to_gpu_arch) {
|
||||||
GGML_LOG_DEBUG("%s: disabling CUDA graphs due to GPU architecture\n", __func__);
|
GGML_LOG_DEBUG("%s: disabling CUDA graphs due to GPU architecture\n", __func__);
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user