probe_diag.py

#!/usr/bin/env python
import torch, inspect
print("torch", torch.__version__, flush=True)
import flash_attn
print("flash_attn", flash_attn.__version__, flush=True)
import vortex
print("vortex", getattr(vortex, "__version__", "?"), flush=True)
import vortex.ops.attn_interface as ai
src = inspect.getsource(ai)
i = src.find("flash_attn_gpu.fwd")
print("=== attn_interface around fwd call ===")
print(src[max(0, i - 1000):i + 500], flush=True)
import evo2
print("evo2", getattr(evo2, "__version__", "?"), flush=True)