stensor 0.4.2

Cross-platform GPU tensor library with Slang and Rust.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

[package]
name = "stensor"
authors = ["Sébastien Crozet <sebcrozet@dimforge.com>"]
description = "Cross-platform GPU tensor library with Slang and Rust."
repository = "https://github.com/dimforge/stensor"
version = "0.4.2"
edition = "2024"
license = "Apache-2.0"

[features]
comptime = [ "slang-hal/comptime" ]
runtime = [ "slang-hal/runtime" ]

webgpu = [ "slang-hal/webgpu" ]
vulkan = [ "slang-hal/vulkan" ]
metal = [ "slang-hal/metal" ]
cpu = [ "slang-hal/cpu" ]
cuda = [ "cudarc", "slang-hal/cuda" ]
cublas = [ "slang-hal/cublas" ]

[dependencies]
encase = "0.12"
bytemuck = "1"
nalgebra = { version = "0.34", features = ["encase"] }

cudarc = {  version = "0.16", optional = true }

slang-hal = { version = "0.4", features = ["derive"] }
include_dir = "0.7"

[dev-dependencies]
minislang = "0.4"
nalgebra = { version = "0.34", features = ["rand"] }
futures-test = "0.3"
serial_test = "3"
approx = "0.5"
async-std = { version = "1", features = ["attributes"] }
plotly = "0.12.1"
indexmap = "2"
anyhow = "1"
wgpu = "27"

[build-dependencies]
minislang = "0.4"
slang-hal-build = "0.4"
include_dir = "0.7"

[patch.crates-io]
#shader-slang = { path = "../slang-rs" }
#minislang = { path = "../slang-hal/crates/minislang" }
#slang-hal-build = { path = "../slang-hal/crates/slang-hal-build" }
#slang-hal-derive = { path = "../slang-hal/crates/slang-hal-derive" }
#slang-hal = { path = "../slang-hal/crates/slang-hal" }