Skip to content

Commit c5b8595

Browse files
authored
Add nvidia and amd backends (#6157)
1 parent 42e21c6 commit c5b8595

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

ggml-sycl.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -977,8 +977,10 @@ namespace dpct
977977
static int convert_backend_index(std::string & backend) {
978978
if (backend == "ext_oneapi_level_zero:gpu") return 0;
979979
if (backend == "opencl:gpu") return 1;
980-
if (backend == "opencl:cpu") return 2;
981-
if (backend == "opencl:acc") return 3;
980+
if (backend == "ext_oneapi_cuda:gpu") return 2;
981+
if (backend == "ext_oneapi_hip:gpu") return 3;
982+
if (backend == "opencl:cpu") return 4;
983+
if (backend == "opencl:acc") return 5;
982984
printf("convert_backend_index: can't handle backend=%s\n", backend.c_str());
983985
GGML_ASSERT(false);
984986
}

0 commit comments

Comments
 (0)