vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix comments in pca

Browse source
This commit is contained in:
Calvin Laurenson 2024-06-15 18:05:08 -07:00
parent 3c4df6ccf3
commit 145f09fc92
No known key found for this signature in database
GPG key ID: 2265749925202DAD
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/cvector-generator/pca.hpp
View file

@ -64,7 +64,6 @@ struct pca_model {
struct ggml_tensor * dev_eigenvector; struct ggml_tensor * dev_eigenvector;
pca_model(struct ggml_tensor * t_input) { pca_model(struct ggml_tensor * t_input) {
// TODO: enable GPU support when support for GGML_OP_SQRT is added
#ifdef GGML_USE_CUDA #ifdef GGML_USE_CUDA
fprintf(stderr, "%s: using CUDA backend\n", __func__); fprintf(stderr, "%s: using CUDA backend\n", __func__);
backend = ggml_backend_cuda_init(0); // init device 0 backend = ggml_backend_cuda_init(0); // init device 0
@ -73,6 +72,7 @@ struct pca_model {
} }
#endif #endif
// TODO: enable Metal support when support for GGML_OP_SQRT is added
// #ifdef GGML_USE_METAL // #ifdef GGML_USE_METAL
// fprintf(stderr, "%s: using Metal backend\n", __func__); // fprintf(stderr, "%s: using Metal backend\n", __func__);
// backend = ggml_backend_metal_init(); // backend = ggml_backend_metal_init();