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

# Pin source trees before running. These commands demonstrate reproducible
# entry points while leaving tactic choice and architecture selection to the
# detected target and pinned release.

: "${CUTLASS_SRC:=}"
: "${CUDNN_FRONTEND_SRC:=}"

if [[ -n "$CUTLASS_SRC" && -d "$CUTLASS_SRC/.git" ]]; then
  git -C "$CUTLASS_SRC" rev-parse HEAD
  test -f "$CUTLASS_SRC/examples/00_basic_gemm/basic_gemm.cu"
  test -d "$CUTLASS_SRC/python/CuTeDSL"
else
  echo "CUTLASS_SRC not set; pin NVIDIA/cutlass v4.5.3 or an explicit commit"
fi

if [[ -n "$CUDNN_FRONTEND_SRC" && -d "$CUDNN_FRONTEND_SRC/.git" ]]; then
  git -C "$CUDNN_FRONTEND_SRC" rev-parse HEAD
else
  echo "CUDNN_FRONTEND_SRC not set; pin NVIDIA/cudnn-frontend v1.26.0"
fi

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

for package in ("nvidia-cutlass-dsl", "nvidia-cudnn-frontend"):
    try:
        print(f"{package}={metadata.version(package)}")
    except metadata.PackageNotFoundError:
        print(f"{package}=missing")
PY