Release v0.3.0

Author: sebcrozet

.github/workflows/ci.yaml

@@ -96,7 +96,7 @@ jobs:
           sweep-cache: true
 
       - name: Run clippy lints
-        run: SLANG_DIR=$SLANG_DIR cargo clippy --locked --workspace --all-targets -- --deny warnings
+        run: SLANG_DIR=$SLANG_DIR cargo clippy --locked --workspace --all-targets --features runtime -- --deny warnings
 
   # Check documentation.
   doc:
@@ -128,7 +128,7 @@ jobs:
           sweep-cache: true
 
       - name: Check documentation
-        run: SLANG_DIR=$SLANG_DIR cargo doc --locked --workspace --document-private-items --no-deps
+        run: SLANG_DIR=$SLANG_DIR cargo doc --locked --workspace --document-private-items --features runtime --no-deps
     # Testing.
   test:
     needs: setup-slang # Depends on setup-slang
@@ -163,4 +163,4 @@ jobs:
           sweep-cache: true
       - name: Run Cargo Tests
         run: |
-          SLANG_DIR=$SLANG_DIR LIBGL_ALWAYS_SOFTWARE=1 cargo test --verbose
+          SLANG_DIR=$SLANG_DIR LIBGL_ALWAYS_SOFTWARE=1 cargo test --verbose --features runtime

Cargo.toml

@@ -3,7 +3,7 @@ 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.2.0"
+version = "0.3.0"
 edition = "2024"
 license = "Apache-2.0"
 
@@ -25,10 +25,11 @@ nalgebra = { version = "0.34", features = ["encase"] }
 
 cudarc = {  version = "0.16", optional = true }
 
-slang-hal = { version = "0.2", features = ["derive"] }
+slang-hal = { version = "0.3", features = ["derive"] }
 include_dir = "0.7"
 
 [dev-dependencies]
+minislang = "0.3"
 nalgebra = { version = "0.34", features = ["rand"] }
 futures-test = "0.3"
 serial_test = "3"
@@ -40,13 +41,13 @@ anyhow = "1"
 wgpu = "27"
 
 [build-dependencies]
-minislang = "0.2"
-slang-hal-build = "0.2"
+minislang = "0.3"
+slang-hal-build = "0.3"
 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" }
+#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" }

build.rs

@@ -1,12 +1,8 @@
-use {
-    minislang::{shader_slang::CompileTarget, SlangCompiler},
-    std::env,
-    std::path::{Path, PathBuf},
-    std::str::FromStr,
-};
+#[cfg(feature = "comptime")]
+use std::env;
 
 #[cfg(not(feature = "comptime"))]
-pub fn main() { }
+pub fn main() {}
 
 #[cfg(feature = "comptime")]
 pub fn main() {

examples/gemm_bench.rs

@@ -1,9 +1,9 @@
 use indexmap::IndexMap;
 use minislang::SlangCompiler;
 use nalgebra::DMatrix;
-use slang_hal::{BufferUsages, Shader};
 use slang_hal::backend::WebGpu;
 use slang_hal::backend::{Backend, Encoder};
+use slang_hal::{BufferUsages, Shader};
 use stensor::linalg::{Gemm, GemmVariant};
 use stensor::shapes::ViewShapeBuffers;
 use stensor::tensor::GpuTensor;

src/lib.rs

@@ -3,9 +3,9 @@
 #![allow(clippy::result_large_err)]
 #![warn(missing_docs)]
 
-use slang_hal::SlangCompiler;
 pub use geometry::*;
 pub use linalg::*;
+use slang_hal::SlangCompiler;
 
 pub mod geometry;
 pub mod linalg;
@@ -14,6 +14,7 @@ pub mod tensor;
 
 // pub mod utils;
 
+/// Directory of slang shaders from `stensor`.
 pub const SLANG_SRC_DIR: include_dir::Dir<'_> =
     include_dir::include_dir!("$CARGO_MANIFEST_DIR/shaders");
 

src/linalg/contiguous.rs

@@ -62,9 +62,9 @@ mod test {
     use crate::tensor::GpuTensor;
     use minislang::SlangCompiler;
     use nalgebra::DMatrix;
-    use slang_hal::{BufferUsages, Shader};
     use slang_hal::backend::WebGpu;
     use slang_hal::backend::{Backend, Encoder};
+    use slang_hal::{BufferUsages, Shader};
     use wgpu::{Features, Limits};
 
     #[futures_test::test]

src/linalg/gemm.rs

@@ -202,8 +202,8 @@ mod test {
     use approx::relative_eq;
     use minislang::SlangCompiler;
     use nalgebra::DMatrix;
-    use slang_hal::{BufferUsages, Shader};
     use slang_hal::backend::{Backend, Encoder, WebGpu};
+    use slang_hal::{BufferUsages, Shader};
     use wgpu::{Features, Limits};
 
     #[futures_test::test]

src/linalg/gemv.rs

@@ -313,9 +313,9 @@ mod test {
     use approx::assert_relative_eq;
     use minislang::SlangCompiler;
     use nalgebra::{DMatrix, DVector};
-    use slang_hal::{BufferUsages, Shader};
     use slang_hal::backend::WebGpu;
     use slang_hal::backend::{Backend, Encoder};
+    use slang_hal::{BufferUsages, Shader};
     use wgpu::{Features, Limits};
 
     #[futures_test::test]

src/linalg/op_assign.rs

@@ -159,9 +159,9 @@ mod test {
     use crate::tensor::GpuTensor;
     use minislang::SlangCompiler;
     use nalgebra::DVector;
+    use slang_hal::BufferUsages;
     use slang_hal::backend::WebGpu;
     use slang_hal::backend::{Backend, Buffer, Encoder};
-    use slang_hal::BufferUsages;
     use slang_hal::shader::Shader;
 
     #[futures_test::test]

src/linalg/reduce.rs

@@ -57,10 +57,10 @@ mod test {
     use crate::tensor::GpuTensor;
     use minislang::SlangCompiler;
     use nalgebra::DVector;
-    use slang_hal::{BufferUsages, ShaderArgs};
     use slang_hal::backend::WebGpu;
     use slang_hal::backend::{Backend, Encoder};
     use slang_hal::shader::Shader;
+    use slang_hal::{BufferUsages, ShaderArgs};
 
     #[derive(ShaderArgs)]
     pub struct ReduceArgs<'a, B: Backend> {