scripts: dump-symbols

Author: Nugine

crates/vsimd/src/macros.rs

@@ -181,7 +181,6 @@ macro_rules! dispatch {
 
     (@resolve_static, "sse2", $($arg_name: ident),*) => {
         #[cfg(all(
-            // not(miri),
             any(target_arch = "x86", target_arch = "x86_64"),
             target_feature = "sse2"
         ))]
@@ -247,7 +246,6 @@ macro_rules! dispatch {
     };
 
     (@resolve_dynamic, "sse2") => {
-        // #[cfg(all(not(miri), any(target_arch = "x86", target_arch = "x86_64")))]
         #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
         if $crate::isa::SSE2::is_enabled() {
             return sse2;
@@ -354,7 +352,6 @@ macro_rules! dispatch {
         simd        = {$simd_fn:path},
         target      = {"sse2"},
     ) => {
-        // #[cfg(all(not(miri), any(target_arch = "x86", target_arch = "x86_64")))]
         #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
         #[inline]
         #[target_feature(enable = "sse2")]
@@ -386,7 +383,7 @@ macro_rules! dispatch {
         target      = {"simd128"},
     ) => {
         #[cfg(target_arch = "wasm32")]
-        #[inline]
+        #[cfg_attr(not(vsimd_dump_symbols), inline)]
         #[target_feature(enable = "simd128")]
         $vis unsafe fn simd128($($arg_name:$arg_type),*) -> $ret {
             use $crate::isa::{WASM128, InstructionSet as _};

justfile

@@ -154,17 +154,21 @@ dump-asm:
     ./scripts/dump-symbols.py --mode asm | bash -ex
     COMMIT_HASH=`git rev-parse --short HEAD`
     cd target/symbols
-    tokei -f -s files -t assembly -c 150 > $COMMIT_HASH-asm.txt
+    F=$COMMIT_HASH-asm.txt
+    tokei -f -s files -t assembly -c 150 > $F
     tokei -f -s lines -t assembly -c 150
+    echo target/symbols/$F
 
 dump-llvm-ir:
     #!/bin/bash -ex
     cd {{justfile_directory()}}
     ./scripts/dump-symbols.py --mode llvm-ir | bash -ex
     COMMIT_HASH=`git rev-parse --short HEAD`
     cd target/symbols
-    tokei -f -s files -t LLVM -c 150 > $COMMIT_HASH-llvm-ir.txt
+    F=$COMMIT_HASH-llvm-ir.txt
+    tokei -f -s files -t LLVM -c 150 > $F
     tokei -f -s lines -t LLVM -c 150
+    echo target/symbols/$F
 
 bench-quick:
     RUSTFLAGS='-Ctarget-cpu=native' cargo run -p simd-benches --bin simd-benches --profile bench --features unstable

scripts/dump-symbols.py

@@ -19,7 +19,17 @@
         "utf32_swap_endianness": ["avx2", "ssse3", "neon", "simd128"],
         "utf16_swap_endianness": ["avx2", "ssse3", "neon", "simd128"],
     },
-    # ...
+    "uuid-simd": {
+        "parse_simple": ["avx2", "ssse3", "sse2", "neon", "simd128"],
+        "parse_hyphenated": ["avx2", "sse4.1", "neon", "simd128"],
+        "format_simple": ["avx2", "ssse3", "sse2", "neon", "simd128"],
+        "format_hyphenated": ["avx2", "sse4.1", "neon", "simd128"],
+    },
+    "base32-simd": {
+        "check": ["avx2", "ssse3", "neon", "simd128"],
+        "encode": ["avx2", "sse4.1", "neon", "simd128"],
+        "decode": ["avx2", "sse4.1", "neon", "simd128"],
+    },
 }
 
 TARGETS = {
@@ -28,10 +38,14 @@
     "ssse3": ["x86_64-unknown-linux-gnu", "i686-unknown-linux-gnu"],
     "sse2": ["x86_64-unknown-linux-gnu", "i686-unknown-linux-gnu"],
     "neon": ["aarch64-unknown-linux-gnu", "armv7-unknown-linux-gnueabihf"],
-    # "simd128": ["wasm32-unknown-unknown"],
-    "simd128": [],  # TODO: https://github.com/pacak/cargo-show-asm/issues/91
+    "simd128": ["wasm32-unknown-unknown"],
 }
 
+
+def space_join(l):
+    return " ".join(l)
+
+
 if __name__ == "__main__":
     opt = argparse.ArgumentParser()
     opt.add_argument("--mode", type=str, choices=["asm", "llvm-ir"], required=True)
@@ -44,16 +58,26 @@
             for feature in features:
                 for target in TARGETS[feature]:
                     print(f"mkdir -p target/symbols/{target}")
+
                     symbol = f"{pkg.replace('-', '_')}::multiversion::{name}::{feature.replace('.', '')}"
-                    if args.mode == "asm":
-                        print(
-                            f"cargo asm -p {pkg} --simplify --features unstable --target {target} -- {symbol} "
-                            f"| awk NF"
-                            f"> target/symbols/{target}/{symbol}.asm"
-                        )
-                    elif args.mode == "llvm-ir":
-                        print(
-                            f"RUSTFLAGS=-Cdebuginfo=0 "
-                            f"cargo asm -p {pkg} --llvm --features unstable --target {target} -- {symbol} "
-                            f"> target/symbols/{target}/{symbol}.ll"
-                        )
+
+                    rustflags = ["--cfg vsimd_dump_symbols"]
+                    if target == "wasm32-unknown-unknown":
+                        rustflags.append("-C target-feature=+simd128")
+
+                    match args.mode:
+                        case "asm":
+                            extra_flags = "--wasm" if target == "wasm32-unknown-unknown" else ""
+                            print(
+                                f'RUSTFLAGS="{space_join(rustflags)}" '
+                                f"cargo asm -p {pkg} --simplify --features unstable --target {target} {extra_flags} -- {symbol} "
+                                f"| awk NF"
+                                f"> target/symbols/{target}/{symbol}.asm"
+                            )
+                        case "llvm-ir":
+                            rustflags.append("-Cdebuginfo=0")
+                            print(
+                                f'RUSTFLAGS="{space_join(rustflags)}" '
+                                f"cargo asm -p {pkg} --llvm --features unstable --target {target} -- {symbol} "
+                                f"> target/symbols/{target}/{symbol}.ll"
+                            )