burn_dragon_kernel 0.5.0

Fused GPU kernel crate for burn_dragon execution paths
Documentation 
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

[package]
name = "burn_dragon_kernel"
description = "Fused GPU kernel crate for burn_dragon execution paths"
version = "0.5.0"
edition = "2024"
rust-version = "1.92"
authors = ["mosure <mitchell@mosure.me>"]
license = "MIT OR Apache-2.0"
keywords = ["burn", "cuda", "kernel", "recurrent"]
categories = ["science", "science::neuroscience"]
homepage = "https://github.com/mosure/burn_bdh"
repository = "https://github.com/mosure/burn_bdh"
readme = "../../README.md"

[features]
default = []
cuda = ["dep:burn-cuda", "dep:cudarc"]

[dependencies]
burn = { version = "0.21.0-pre.3", default-features = false, features = ["fusion", "ndarray", "std", "wgpu"] }
burn-autodiff = { version = "0.21.0-pre.3", default-features = false }
burn-cuda = { version = "0.21.0-pre.3", default-features = false, features = ["std"], optional = true }
burn-wgpu = { version = "0.21.0-pre.3", default-features = false, features = ["fusion", "std", "template"] }
burn-cubecl = { version = "0.21.0-pre.3", default-features = false, features = ["std", "fusion"] }
burn-fusion = { version = "0.21.0-pre.3", default-features = false, features = ["std"] }
burn-std = "0.21.0-pre.3"
burn-ir = "0.21.0-pre.3"
pollster = "0.3"
serde = { version = "1", features = ["derive"] }
serde_json = "1"
wgpu = "26"
cudarc = { version = "0.17.8", features = ["cuda-version-from-build-system"], optional = true }

[dev-dependencies]
burn-ndarray = "0.21.0-pre.3"

[lib]
path = "src/lib.rs"
crate-type = ["rlib"]