Skip to content

Commit 0f3dec3

Browse files
committed
build: update deps
Signed-off-by: YdrMaster <ydrml@hotmail.com>
1 parent a4df34e commit 0f3dec3

File tree

3 files changed

+7
-7
lines changed

3 files changed

+7
-7
lines changed

Cargo.toml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ regex = "1.11"
3030
env_logger = "0.11"
3131
build-script-cfg = "0.0"
3232

33-
operators = { git = "https://github.com/YdrMaster/operators-rs", rev = "7aa1051", default-features = false }
34-
search-cl-tools = { git = "https://github.com/InfiniTensor/clrt", rev = "6846d52" }
35-
search-infini-tools = { git = "https://github.com/InfiniTensor/infini-toolkit", rev = "9943c78" }
36-
search-cuda-tools = { git = "https://github.com/YdrMaster/cuda-driver", rev = "e2ec203" }
33+
operators = { git = "https://github.com/YdrMaster/operators-rs", rev = "1c419ad", default-features = false }
34+
search-cl-tools = { git = "https://github.com/InfiniTensor/clrt", rev = "9b6289d" }
35+
search-infini-tools = { git = "https://github.com/InfiniTensor/infini-rt", rev = "0e57976" }
36+
search-cuda-tools = { git = "https://github.com/YdrMaster/cuda-driver", rev = "5b9dbd9" }

models/llama/infini/Cargo.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,6 @@ build-script-cfg.workspace = true
1515
search-infini-tools.workspace = true
1616

1717
[dev-dependencies]
18-
test-utils.workspace = true
18+
test-utils = { workspace = true, features = ["llama"] }
1919
gguf.workspace = true
2020
regex.workspace = true

models/llama/infini/src/infer.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
use crate::{Operators, RandomSample, Weights};
2-
use gguf::GGufModel;
2+
use gguf::{ggml_quants::digit_layout::types, GGufModel};
33
use llama::{ext::ggml_quants::f16, LlamaRequest, LlamaStorage, LlamaWorker, Tensor};
44
use operators::{
55
infini::Device,
@@ -82,7 +82,7 @@ fn test_infer() {
8282
.kv_cache(meta.nctx)
8383
.map(|size| stream.malloc::<u8>(size));
8484
let sin_cos = <Operators as llama::Operators>::build_sin_cos(
85-
meta.dt_embd,
85+
types::F32,
8686
meta.nctx,
8787
meta.dh,
8888
&stream,

0 commit comments

Comments
 (0)