Add avx512 feature

This enables the AVX-512 kernels in RTen.
robertknight · Apr 30, 2024 · e990429 · e990429
1 parent db7889f
commit e990429
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/ocrs-cli/Cargo.toml b/ocrs-cli/Cargo.toml
@@ -22,6 +22,10 @@ url = "2.4.0"
 home = "0.5.9"
 anyhow = "1.0.79"
 
+[features]
+# Use AVX-512 instructions if available. Requires nightly Rust.
+avx512 = ["rten/avx512"]
+
 [[bin]]
 name = "ocrs"
 path = "src/main.rs"

diff --git a/ocrs/Cargo.toml b/ocrs/Cargo.toml
@@ -27,3 +27,7 @@ rten-imageio = { version = "0.8.0" }
 
 [lib]
 crate-type = ["lib", "cdylib"]
+
+[features]
+# Use AVX-512 instructions if available. Requires nightly Rust.
+avx512 = ["rten/avx512"]