#!/usr/bin/env bash
set -euo pipefail

# Read-only environment receipt. This script never selects an architecture by
# product nickname; it asks the installed driver and toolkit.

command -v nvidia-smi >/dev/null && nvidia-smi --query-gpu=index,name,uuid,driver_version,compute_cap,pci.bus_id,memory.total --format=csv,noheader || true
command -v nvcc >/dev/null && nvcc --version || true
command -v nvidia-smi >/dev/null && nvidia-smi topo -m || true
command -v nvidia-smi >/dev/null && nvidia-smi nvlink --status || true
command -v dcgmi >/dev/null && dcgmi discovery -l || true

python3 - <<'PY'
import importlib.metadata as metadata
import json

packages = [
    "torch",
    "triton",
    "flashinfer-python",
    "transformer-engine",
    "nvidia-modelopt",
    "vllm",
    "sglang",
    "lmcache",
    "tensorrt-llm",
    "ai-dynamo",
    "nixl",
]
versions = {}
for package in packages:
    try:
        versions[package] = metadata.version(package)
    except metadata.PackageNotFoundError:
        versions[package] = None
print(json.dumps(versions, indent=2, sort_keys=True))
PY