From 8603c25e3ce074d4717584a2c2ba60622e67b287 Mon Sep 17 00:00:00 2001 From: 0cc4m Date: Mon, 24 Apr 2023 15:53:48 +0200 Subject: [PATCH] Fix device selection env variable names --- ggml-opencl.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ggml-opencl.cpp b/ggml-opencl.cpp index 904e32ac9..e7f40bbef 100644 --- a/ggml-opencl.cpp +++ b/ggml-opencl.cpp @@ -104,10 +104,10 @@ cl_program build_program_from_source(cl_context ctx, cl_device_id dev, const cha void ggml_cl_init() { cl_int err = 0; - char * KCPP_CLBLAST_PLATFORM = getenv("KCPP_CLBLAST_PLATFORM"); - char * KCPP_CLBLAST_DEVICES = getenv("KCPP_CLBLAST_DEVICES"); - int plat_num = (KCPP_CLBLAST_PLATFORM == NULL ? 0 : atoi(KCPP_CLBLAST_PLATFORM)); - int dev_num = (KCPP_CLBLAST_DEVICES == NULL ? 0 : atoi(KCPP_CLBLAST_DEVICES)); + char * GGML_CLBLAST_PLATFORM = getenv("GGML_CLBLAST_PLATFORM"); + char * GGML_CLBLAST_DEVICE = getenv("GGML_CLBLAST_DEVICE"); + int plat_num = (GGML_CLBLAST_PLATFORM == NULL ? 0 : atoi(GGML_CLBLAST_PLATFORM)); + int dev_num = (GGML_CLBLAST_DEVICE == NULL ? 0 : atoi(GGML_CLBLAST_DEVICE)); printf("\nInitializing CLBlast (First Run)..."); printf("\nAttempting to use: Platform=%d, Device=%d (If invalid, program will crash)\n",plat_num,dev_num); cl_uint num_platforms;