Skip to content

Commit 6c8d01a

Browse files
committed
add regex support
1 parent bb6b97e commit 6c8d01a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/llama-model.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
#include <cstring>
1212
#include <functional>
1313
#include <map>
14+
#include <regex>
1415
#include <sstream>
1516
#include <stdexcept>
1617

@@ -1464,7 +1465,8 @@ bool llama_model::load_tensors(llama_model_loader & ml) {
14641465
if (ml.tensor_buft_overrides) {
14651466
std::string tensor_name = tn.str();
14661467
for (const auto * overrides = ml.tensor_buft_overrides; overrides->pattern != nullptr; ++overrides) {
1467-
if (tensor_name.find(overrides->pattern) != std::string::npos) {
1468+
std::regex pattern(overrides->pattern);
1469+
if (std::regex_search(tensor_name, pattern)) {
14681470
LLAMA_LOG_DEBUG("tensor %s buffer type overriden to %s\n", tensor_name.c_str(), ggml_backend_buft_name(overrides->buft));
14691471
buft = overrides->buft;
14701472
break;

0 commit comments

Comments
 (0)