From baf3f7ae634f1d7cd77e367eaf34527c6cf94df0 Mon Sep 17 00:00:00 2001 From: ausk Date: Wed, 2 Sep 2026 16:22:13 +0800 Subject: [PATCH] Update tensor retrieval method for layer normalization Change tensor retrieval to use host_tensor for weight and bias to fix SIGSEGV when run in gpu mode. --- src/dino_backbone.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/dino_backbone.cpp b/src/dino_backbone.cpp index 71c1b68..e991194 100644 --- a/src/dino_backbone.cpp +++ b/src/dino_backbone.cpp @@ -239,8 +239,8 @@ bool DinoBackbone::build_feats_graph(ggml_context* ctx, const std::vector const std::vector& outL = c.out_layers; const size_t NL = outL.size(); - ggml_tensor* nw = ml_.tensor("vit.norm.weight"); - ggml_tensor* nb = ml_.tensor("vit.norm.bias"); + ggml_tensor* nw = ml_.host_tensor("vit.norm.weight"); + ggml_tensor* nb = ml_.host_tensor("vit.norm.bias"); if (be_.is_offloading()) { // vit.norm.* are kept as host gguf tensors (no backend buffer) for the // host-side layernorm used by other paths. Here they feed an IN-GRAPH