Files
nvfp4-megamoe-kernel/tests/unit/test_fp4_isolate_runner.py

11 lines
494 B
Python

"""Run each function isolation test in a separate process."""
import subprocess, sys
for t in ["round_rne", "abs_scaled", "fp8_encode", "fp8_decode", "e2m1_quant"]:
print(f"\n{'='*50}\n{t}\n{'='*50}")
r = subprocess.run([sys.executable, "tests/unit/test_fp4_isolate.py", t],
capture_output=True, text=True, timeout=120)
print(r.stdout[-300:] if r.stdout else "")
if r.stderr:
print(f"ERR: ...{r.stderr[-200:]}")
print(f"Exit: {r.returncode}")