]> git.djapps.eu Git - pkg/ggml/sources/ggml/commitdiff
Add nvidia and amd backends (llama/6157)
authorAidanBeltonS <redacted>
Thu, 21 Mar 2024 06:10:52 +0000 (06:10 +0000)
committerGeorgi Gerganov <redacted>
Wed, 27 Mar 2024 11:20:00 +0000 (13:20 +0200)
src/ggml-sycl.cpp

index d51f23b410595ed7653cce928bc724aba26cec7a..cc9ee0762b903e7d9496a9e221f8673adce1b401 100644 (file)
@@ -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);
         }