Packages

nx_vulkan

0.1.0
0.1.0

GPU tensor backend for Nx via Vulkan compute. Pure-Rust path (vulkano) plus C++ spirit path. Validated on Axon training, Scholar linear regression, and the eXMC NUTS sampler. Works on Linux + FreeBSD NVIDIA where CUDA does not exist.

HexDocs

Current section

Files

Jump to
Readme
1 Version
2 Dependencies
Dependants
Files
Activity
nx_vulkan native nx_vulkan_vulkano Cargo.toml
Raw

native/nx_vulkan_vulkano/Cargo.toml 

[package]

name = "nx_vulkan_vulkano"

version = "0.0.1"

authors = ["Igor Ostaptchenko <igor@octanix.com>"]

edition = "2021"

description = "Pure-Rust Rustler NIF for synthesised Vulkan chain shader dispatch via vulkano. Sibling of nx_vulkan_native — no C++ shim, no spirit backend; Arc<Buffer> resource lifetimes."



[lib]

name = "nx_vulkan_vulkano"

path = "src/lib.rs"

crate-type = ["cdylib"]



[dependencies]

rustler = "0.36"

vulkano = "0.34"

# Vulkano 0.34 uses ahash::HashMap throughout its public API

# (e.g. shader.specialize takes HashMap<u32, _, ahash::RandomState>).

# We match the version vulkano pulls in transitively to share the

# same RandomState type.

ahash = "0.8"



[profile.release]

opt-level = 3

lto = "fat"

codegen-units = 1