huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 10:26:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Integrate the kernels bits.

Browse Source
This commit is contained in:
laurent
2023-06-22 09:59:00 +01:00
parent 1309932933
commit 083ced4428
7 changed files with 294 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
Cargo.toml
View File

@ -4,16 +4,22 @@ version = "0.1.0"
edition = "2021"
description = "Minimalist ML framework."
repository = "https://github.com/LaurentMazare/wtensor"
repository = "https://github.com/LaurentMazare/candle"
keywords = ["blas", "tensor", "machine-learning"]
categories = ["science"]
license = "MIT/Apache-2.0"
readme = "README.md"
[workspace]
members = [
"kernels",
]
[dependencies]
safetensors = "0.3.1"
thiserror = "1"
cudarc = { version = "0.9.9", optional = true }
candle-kernels = { path = "kernels", optional = true }
[dev-dependencies]
anyhow = "1"
@ -23,4 +29,4 @@ tokenizers = "0.13.3"
[features]
default = []
cuda = ["dep:cudarc"]
cuda = ["dep:cudarc", "dep:candle-kernels"]