From: AidanBeltonS <redacted>
Date: Thu, 21 Mar 2024 06:10:52 +0000 (+0000)
Subject: Add nvidia and amd backends (llama/6157)
X-Git-Tag: upstream/0.0.1642~825
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=5860eb04a2c995c267e11110f42b3c99073d8187;p=pkg%2Fggml%2Fsources%2Fggml

Add nvidia and amd backends (llama/6157)
---

diff --git a/src/ggml-sycl.cpp b/src/ggml-sycl.cpp
index d51f23b4..cc9ee076 100644
--- a/src/ggml-sycl.cpp
+++ b/src/ggml-sycl.cpp
@@ -977,8 +977,10 @@ namespace dpct
         static int convert_backend_index(std::string & backend) {
             if (backend == "ext_oneapi_level_zero:gpu") return 0;
             if (backend == "opencl:gpu") return 1;
-            if (backend == "opencl:cpu") return 2;
-            if (backend == "opencl:acc") return 3;
+            if (backend == "ext_oneapi_cuda:gpu") return 2;
+            if (backend == "ext_oneapi_hip:gpu") return 3;
+            if (backend == "opencl:cpu") return 4;
+            if (backend == "opencl:acc") return 5;
             printf("convert_backend_index: can't handle backend=%s\n", backend.c_str());
             GGML_ASSERT(false);
         }