test: verify_build.sh for real hardware validation
This commit is contained in:
78
verify_build.sh
Executable file
78
verify_build.sh
Executable file
@@ -0,0 +1,78 @@
|
||||
#!/usr/bin/env bash
|
||||
# verify_build.sh — 在BI-V100真机上验证完整build链
|
||||
set -uo pipefail
|
||||
cd "$(dirname "$0")"
|
||||
|
||||
echo "=== 1. patch_ops.sh syntax ==="
|
||||
bash -n qwen3_6_scripts/patch_ops.sh && echo "✓ OK" || echo "✗ FAIL"
|
||||
|
||||
echo ""
|
||||
echo "=== 2. build ix_unified_bridge.so ==="
|
||||
bash ex_engine/build_unified_bridge.sh 2>&1 | tail -10
|
||||
|
||||
echo ""
|
||||
echo "=== 3. verify bridge load ==="
|
||||
python3 << 'PY'
|
||||
import importlib.util, glob
|
||||
so = glob.glob("ex_engine/build/ix_unified_bridge*.so")
|
||||
if not so:
|
||||
print("✗ .so not built"); exit(1)
|
||||
spec = importlib.util.spec_from_file_location("ix_unified_bridge", so[0])
|
||||
mod = importlib.util.module_from_spec(spec)
|
||||
spec.loader.exec_module(mod)
|
||||
funcs = [x for x in dir(mod) if not x.startswith('_')]
|
||||
print(f"✓ {len(funcs)} functions: {funcs}")
|
||||
PY
|
||||
|
||||
echo ""
|
||||
echo "=== 4. ix_unified dispatch smoke test ==="
|
||||
python3 << 'PY'
|
||||
import sys, os
|
||||
sys.path.insert(0, "ex_engine/build")
|
||||
sys.path.insert(0, "ex_engine/python")
|
||||
os.environ["IX_BRIDGE_PATH"] = "ex_engine/build"
|
||||
from ix_unified import ix
|
||||
print(f"bridge={ix._bridge is not None}, ixformer={ix._ixf is not None}")
|
||||
|
||||
import torch
|
||||
# silu_and_mul
|
||||
x = torch.randn(2, 512, device="cuda", dtype=torch.float16)
|
||||
out = ix.silu_and_mul(x)
|
||||
print(f"✓ silu_and_mul: {x.shape} → {out.shape}")
|
||||
|
||||
# rms_norm
|
||||
inp = torch.randn(4, 2048, device="cuda", dtype=torch.float16)
|
||||
outp = torch.empty_like(inp)
|
||||
w = torch.ones(2048, device="cuda", dtype=torch.float16)
|
||||
ix.rms_norm(outp, inp, w, 1e-6)
|
||||
print(f"✓ rms_norm: {inp.shape}")
|
||||
|
||||
# moe_topk_softmax
|
||||
gate = torch.randn(8, 256, device="cuda", dtype=torch.float16)
|
||||
weights, indices = ix.moe_topk_softmax(gate, 8, True)
|
||||
print(f"✓ moe_topk_softmax: {gate.shape} → weights={weights.shape} indices={indices.shape}")
|
||||
|
||||
print("ALL SMOKE TESTS PASSED")
|
||||
PY
|
||||
|
||||
echo ""
|
||||
echo "=== 5. prebuilt .so still load ==="
|
||||
python3 << 'PY'
|
||||
import importlib.util, os
|
||||
so_dir = "qwen3_6_scripts/prebuilt/corex-3.2.3-ivcore10"
|
||||
for name in ["corex_moe_direct_routed", "corex_gdn_packed_decode", "corex_gdn_causal_conv"]:
|
||||
so = os.path.join(so_dir, f"{name}.so")
|
||||
spec = importlib.util.spec_from_file_location(name, so)
|
||||
mod = importlib.util.module_from_spec(spec)
|
||||
spec.loader.exec_module(mod)
|
||||
funcs = [x for x in dir(mod) if not x.startswith('_')]
|
||||
print(f"✓ {name}: {funcs}")
|
||||
PY
|
||||
|
||||
echo ""
|
||||
echo "=== 6. py_compile all ==="
|
||||
cd qwen3_6_scripts
|
||||
find . -path './wheels' -prune -o -name '*.py' -print0 | xargs -0 python3 -m py_compile 2>&1 && echo "✓ all OK" || echo "✗ errors"
|
||||
|
||||
echo ""
|
||||
echo "=== DONE ==="
|
||||
Reference in New Issue
Block a user