API Reference nx_vulkan v#0.3.0

Copy Markdown View Source

Modules

Nx tensor backend on Vulkan compute (vulkano/Rust), in native f32 and f64.

Catalog of family specs for templated chain-shader synthesis.

JIT GLSL codegen from Nx.Defn.Expr trees — the heart of the thrust-3 fusion compiler.

Nx.Defn.Compiler for the Vulkan backend — thrust 3, the fusion compiler.

GPU device classification for hardware-dependent perf heuristics.

Counts host fallbacks so a silent one becomes a test failure.

Named fused kernels for MCMC hot paths.

Rustler NIF for the vulkano compute backend.

Long-lived per-machine GPU node. A named GenServer that serializes GPU work and owns the watchdog/timeout layer. Clients submit work via with_node/2 (or the lower-level exec/2); the node runs it one call at a time and reports timeouts/dead-server/etc. as error tuples.

Pipeline cache stub. The spirit C++ backend (which owned the VkPipelineCache) has been removed. Vulkano manages its own pipeline caching internally. These functions are no-ops retained for API compatibility with callers that haven't been updated.

Phase 1 — templated GLSL shader synthesis.

Per-family hooks for the templated chain shader.

Phase 1 — end-to-end shader synthesis driver.

Pure-Rust (vulkano) Nx.Backend implementation.