Skip to content

Commit 6907868

Browse files
jianjunzabrown
andauthored
Add FP16 and I64 support for wasi-nn WinML backend. (bytecodealliance#8964)
* Add FP16 and I64 support for wasi-nn WinML backend. Some devices may not support FP32. prtest:full * Remove unnecessary features. * Address comments. * Check alignment before from_raw_parts. * Implement PartialEq for Tensor. * Remove duplicated shape info from set_input. * Update alignment checker. * Add comments about creating TensorFloat16Bit from f32 array. * Use PartialEq attribute. * Audit new WinML dependencies --------- Co-authored-by: Andrew Brown <[email protected]>
1 parent 895180d commit 6907868

File tree

6 files changed

+338
-59
lines changed

6 files changed

+338
-59
lines changed

Cargo.lock

Lines changed: 60 additions & 16 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

crates/wasi-nn/Cargo.toml

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,13 @@ ort = { version = "2.0.0-rc.2", default-features = false, features = [
4040

4141
[target.'cfg(windows)'.dependencies.windows]
4242
version = "0.52"
43-
features = ["AI_MachineLearning", "Storage_Streams", "Foundation_Collections"]
43+
features = [
44+
"AI_MachineLearning",
45+
"Storage_Streams",
46+
"Foundation_Collections",
47+
# For getting IVectorView<i64> from tensor.dimensions.
48+
"implement",
49+
]
4450
optional = true
4551

4652
[build-dependencies]

0 commit comments

Comments
 (0)