make flag opt-in
This commit is contained in:
parent
25ae798615
commit
93cfa8d156
|
|
@ -3826,16 +3826,12 @@ static void ggml_backend_cuda_graph_optimize(ggml_backend_t backend, ggml_cgraph
|
||||||
|
|
||||||
const bool use_cuda_graph = ggml_cuda_set_cuda_graph_enabled(cuda_ctx);
|
const bool use_cuda_graph = ggml_cuda_set_cuda_graph_enabled(cuda_ctx);
|
||||||
|
|
||||||
static bool disable_graph_optimization = [] {
|
static bool enable_graph_optimization = [] {
|
||||||
const char * env = getenv("GGML_CUDA_DISABLE_GRAPH_OPT");
|
const char * env = getenv("GGML_CUDA_GRAPH_OPT");
|
||||||
bool disable = env != nullptr && atoi(env) == 1;
|
return env != nullptr && atoi(env) == 1;
|
||||||
|
|
||||||
env = getenv("GGML_CUDA_GRAPH_OPT");
|
|
||||||
GGML_ASSERT(env == nullptr && "GGML_CUDA_GRAPH_OPT is deprecated, use GGML_CUDA_DISABLE_GRAPH_OPT instead");
|
|
||||||
return disable;
|
|
||||||
}();
|
}();
|
||||||
|
|
||||||
if (disable_graph_optimization) {
|
if (!enable_graph_optimization) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue