mirror of
https://github.com/harvard-edge/cs249r_book.git
synced 2026-07-21 17:57:24 -05:00
Vijay Janapa Reddi
ff614a3682
Update tutorials and landing pages to canonical nested registry paths, fix solver import examples, add Platforms/Datasets/Literature/Ops zoo pages plus provenance docs, and gate doc drift with test_doc_registry_paths.
53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
# generate_appendix.py
|
|
"""
|
|
mlsysim Appendix Generator
|
|
==========================
|
|
Generates Quarto-compatible Markdown tables for the textbook's backmatter.
|
|
Extracts live data from the mlsysim Hardware and Model registries.
|
|
"""
|
|
|
|
from mlsysim.core.constants import Q_
|
|
from mlsysim.hardware.registry import Hardware
|
|
from mlsysim.models.registry import Models
|
|
|
|
def fmt_q(q: Q_, precision: int = 1) -> str:
|
|
"""Format a quantity for the table."""
|
|
if q is None: return "---"
|
|
return f"{q.magnitude:,.{precision}f} {q.units:~P}"
|
|
|
|
def generate_hardware_appendix():
|
|
"""Generates the Hardware Specifications table for the appendix."""
|
|
header = "| Accelerator | Year | Peak FP16 | Memory BW | Memory Capacity | TDP |\n"
|
|
divider = "|:---|:---:|:---:|:---:|:---:|:---:|\n"
|
|
|
|
rows = []
|
|
# Cloud Tiers
|
|
for h in [Hardware.Cloud.A100, Hardware.Cloud.H100, Hardware.Cloud.H200, Hardware.Cloud.MI300X]:
|
|
row = f"| {h.name} | {h.release_year} | {fmt_q(h.compute.peak_flops)} | {fmt_q(h.memory.bandwidth)} | {fmt_q(h.memory.capacity)} | {fmt_q(h.tdp, 0)} |"
|
|
rows.append(row)
|
|
|
|
return header + divider + "\n".join(rows)
|
|
|
|
def generate_model_appendix():
|
|
"""Generates the Model Workload table for the appendix."""
|
|
header = "| Model | Architecture | Parameters | Inference FLOPS | Layers |\n"
|
|
divider = "|:---|:---:|:---:|:---:|:---:|\n"
|
|
|
|
rows = []
|
|
for m in [
|
|
Models.Language.GPT2,
|
|
Models.Language.GPT3,
|
|
Models.Vision.ResNet50,
|
|
Models.Vision.MobileNetV2,
|
|
]:
|
|
row = f"| {m.name} | {m.architecture} | {fmt_q(m.parameters)} | {fmt_q(m.inference_flops)} | {m.layers or '---'} |"
|
|
rows.append(row)
|
|
|
|
return header + divider + "\n".join(rows)
|
|
|
|
if __name__ == "__main__":
|
|
print("## Hardware Specifications Appendix\n")
|
|
print(generate_hardware_appendix())
|
|
print("\n\n## Model Workload Appendix\n")
|
|
print(generate_model_appendix())
|