Implement SIMD

README.md

@@ -39,6 +39,7 @@ A brief description for `example`
 | example/blake2b.py        | Blake2b hashing algorithm                                            |
 | example/blake2b_direct.py | Make the hash result returned as a value, not as an output parameter |
 | example/blake2b_iter.py   | A benchmarking example using the blake2b hash function               |
+| example/blake2b_simd.py   | Use SIMD instructions to accelerate the blake2b hash function        |
 | example/fibonacci.py      | Fibonacci, which contains loop and recursion                         |
 | example/fibonacci_env.py  | Call python/native function in wasm                                  |
 | example/pi.py             | Calculate π using the agm algorithm                                  |

example/blake2b_simd.py

@@ -0,0 +1,23 @@
+import pywasm
+
+runtime = pywasm.core.Runtime()
+module = runtime.instance_from_file('example/blake2b_simd/bin/blake2b_simd.wasm')
+memory = runtime.exported_memory(module, 'memory')
+
+data = bytearray(b'abc')
+data_size = len(data)
+data_ptr = runtime.invocate(module, 'alloc', [data_size])[0]
+hash_size = 64
+hash_ptr = runtime.invocate(module, 'alloc', [hash_size])[0]
+memory.put(data_ptr, data)
+runtime.invocate(module, 'blake2b', [data_ptr, data_size, hash_ptr, hash_size])
+hash = memory.get(hash_ptr, hash_size)
+print(hash.hex())
+assert hash == bytearray([
+    0xba, 0x80, 0xa5, 0x3f, 0x98, 0x1c, 0x4d, 0x0d, 0x6a, 0x27, 0x97, 0xb6, 0x9f, 0x12, 0xf6, 0xe9,
+    0x4c, 0x21, 0x2f, 0x14, 0x68, 0x5a, 0xc4, 0xb7, 0x4b, 0x12, 0xbb, 0x6f, 0xdb, 0xff, 0xa2, 0xd1,
+    0x7d, 0x87, 0xc5, 0x39, 0x2a, 0xab, 0x79, 0x2d, 0xc2, 0x52, 0xd5, 0xde, 0x45, 0x33, 0xcc, 0x95,
+    0x18, 0xd3, 0x8a, 0xa8, 0xdb, 0xf1, 0x92, 0x5a, 0xb9, 0x23, 0x86, 0xed, 0xd4, 0x00, 0x99, 0x23,
+])
+runtime.invocate(module, 'dealloc', [data_ptr, data_size])
+runtime.invocate(module, 'dealloc', [hash_ptr, hash_size])

example/blake2b_simd/.cargo/config.toml

@@ -0,0 +1,5 @@
+[build]
+target = "wasm32-unknown-unknown"
+
+[target.wasm32-unknown-unknown]
+rustflags = ["-C", "target-feature=+simd128"]

example/blake2b_simd/.gitignore

@@ -0,0 +1 @@
+/target

example/blake2b_simd/Cargo.toml

@@ -0,0 +1,10 @@
+[package]
+name = "blake2b_simd"
+version = "1.0.0"
+edition = "2024"
+
+[lib]
+crate-type = ["cdylib"]
+
+[dependencies]
+blake2ya = "1.0"