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

19 lines
538 B
Python

"""Runner: test each f32→i32 approach in a separate process."""
import subprocess
import sys
for approach in ["arith", "nvvm_ptx", "nvvm_multiline"]:
print(f"\n{'='*50}")
print(f"Testing: {approach}")
print(f"{'='*50}")
result = subprocess.run(
[sys.executable, "test_ptx_subproc.py", approach],
capture_output=True,
text=True,
timeout=120,
)
print(result.stdout)
if result.stderr:
print(f"STDERR: {result.stderr[:500]}")
print(f"Exit code: {result.returncode}")