requirements.txt

# Dependencies for local analysis and processing scripts
# (Training dependencies are managed in Docker - see docker/Dockerfile.*)
#
# Install with uv (two steps to avoid pulling nvidia packages):
#   uv pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm7.1/
#   uv pip install -r requirements.txt
#
# For full installation including hw_queue_eval:
#   uv pip install -e ".[hw-queue]"
#
# For development:
#   uv pip install -e ".[all]"

# Base dependencies (PyTorch installed separately - see above)
pyyaml>=6.0

# For analysis scripts
matplotlib>=3.7.0
numpy>=1.20.0
pandas>=1.3.0
openpyxl>=3.0.0
seaborn>=0.12.0
beautifulsoup4>=4.12.0

# For GEMM analysis (scripts/gemm_analysis/)
git+https://github.com/AMD-AGI/TraceLens.git

# For Magpie integration (GPU hardware control, benchmark report adapter)
git+https://github.com/AMD-AGI/Magpie.git

# For aorta-report CLI (src/aorta/report/)
click>=8.0.0

# For hw_queue_eval (optional - install with: pip install -e ".[hw-queue]")
# numpy>=1.20.0  # Already included above
# tabulate>=0.9.0

# For trace processing (merge_gpu_trace_ranks.py, etc.)
# No additional deps - uses only Python standard library (json, argparse, pathlib)