summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAidanBeltonS <87009434+AidanBeltonS@users.noreply.github.com>2024-03-21 06:10:52 +0000
committerGitHub <noreply@github.com>2024-03-21 11:40:52 +0530
commitc5b8595e3f4f4ed319ef71c9c9d868d1b7a27626 (patch)
tree3a608dda682b43016b0239c5111ea7df9752f982
parent42e21c68826f2e56b9592dccd9f3c43895b6890d (diff)
Add nvidia and amd backends (#6157)
-rw-r--r--ggml-sycl.cpp6
1 files changed, 4 insertions, 2 deletions
diff --git a/ggml-sycl.cpp b/ggml-sycl.cpp
index d51f23b4..cc9ee076 100644
--- a/ggml-sycl.cpp
+++ b/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);
}