Răsfoiți Sursa

Add nvidia and amd backends (#6157)

AidanBeltonS 1 an în urmă
părinte
comite
c5b8595e3f
1 a modificat fișierele cu 4 adăugiri și 2 ștergeri
  1. 4 2
      ggml-sycl.cpp

+ 4 - 2
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);
         }