diff options
author | AidanBeltonS <87009434+AidanBeltonS@users.noreply.github.com> | 2024-03-21 06:10:52 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-03-21 11:40:52 +0530 |
commit | c5b8595e3f4f4ed319ef71c9c9d868d1b7a27626 (patch) | |
tree | 3a608dda682b43016b0239c5111ea7df9752f982 | |
parent | 42e21c68826f2e56b9592dccd9f3c43895b6890d (diff) |
Add nvidia and amd backends (#6157)
-rw-r--r-- | ggml-sycl.cpp | 6 |
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); } |