mirror of
https://github.com/harvard-edge/cs249r_book.git
synced 2026-05-10 15:49:25 -05:00
1eb30f5f86
Align the MLSys·im code, docs, paper, website, workflows, and lab wheel for the 0.1.1 release. This also fixes runtime/API issues found during release review and prepares the paper PDF plus archive package.
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
"""
|
|
Example: Custom System Design
|
|
=============================
|
|
This script demonstrates how to build a hypothetical system from scratch
|
|
without using the vetted registries. This is how researchers can use
|
|
mlsysim to model unreleased or generic hardware.
|
|
"""
|
|
|
|
import mlsysim
|
|
from mlsysim.hardware.types import HardwareNode, ComputeCore, MemoryHierarchy
|
|
from mlsysim.models.types import CNNWorkload
|
|
from mlsysim.core.scenarios import Scenario
|
|
|
|
def main():
|
|
print("--- Designing a Hypothetical 'Generic Drone' ---")
|
|
|
|
# 1. Manually define hardware (Supply)
|
|
drone_chip = HardwareNode(
|
|
name="Hypothetical Drone NPU",
|
|
release_year=2026,
|
|
compute=ComputeCore(peak_flops="10 TFLOPs/s"),
|
|
memory=MemoryHierarchy(capacity="2 GB", bandwidth="50 GB/s"),
|
|
tdp="10 W",
|
|
dispatch_tax="0.5 ms"
|
|
)
|
|
|
|
# 2. Manually define workload (Demand)
|
|
my_model = CNNWorkload(
|
|
name="Custom Vision Model",
|
|
architecture="CNN",
|
|
parameters="50 Mparam",
|
|
inference_flops="10 Gflop"
|
|
)
|
|
|
|
# 3. Bundle into a Scenario
|
|
my_scenario = Scenario(
|
|
name="Generic Drone Vision",
|
|
description="A custom vision task on unreleased drone hardware.",
|
|
workload=my_model,
|
|
system=drone_chip,
|
|
sla_latency="30 ms"
|
|
)
|
|
|
|
# 4. Evaluate the custom design
|
|
print(f"Evaluating {my_scenario.name}...")
|
|
report = my_scenario.evaluate()
|
|
print(report.scorecard())
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|
|
# Expected output (mlsysim v0.1.1):
|
|
# --- Designing a Hypothetical 'Generic Drone' ---
|
|
# Evaluating Generic Drone Vision...
|
|
# +============================================================+
|
|
# | MLSys-im SYSTEM EVALUATION
|
|
# | Scenario: Generic Drone Vision
|
|
# +============================================================+
|
|
# | Level 1: Feasibility [PASS]
|
|
# | Model fits in memory (100.0 MB / 2000.0 MB)
|
|
# +------------------------------------------------------------+
|
|
# | Level 2: Performance [PASS]
|
|
# | Latency: 2.71 millisecond (Target: 30 ms)
|
|
# +------------------------------------------------------------+
|
|
# | Level 3: Macro/Economics [PASS]
|
|
# | Annual Carbon: 42.1 kg | TCO: $11,512
|
|
# +============================================================+
|