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_native Cargo.toml
Raw

native/nx_vulkan_native/Cargo.toml 

[package]

name = "nx_vulkan_native"

version = "0.0.1"

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

edition = "2021"



[lib]

name = "nx_vulkan_native"

path = "src/lib.rs"

crate-type = ["cdylib"]



[dependencies]

# Rust-side rustler crate (separate from the Hex orchestrator package

# in mix.exs). 0.36 has the term-type fix that 0.30 lacked.

rustler = "0.36"



[build-dependencies]

# build.rs compiles the extern "C" shim against spirit's

# Backend_par_vulkan.cpp. cc handles the C++ compile.

cc = "1.0"