github-starred/cs249r_book
2
0
Fork 0
mirror of https://github.com/harvard-edge/cs249r_book.git synced 2026-04-30 17:48:27 -05:00
Code Issues 77 Packages Projects Releases 24 Wiki Activity

docs(mlsysim): add CLI instructions to getting started guide

Browse Source
This commit is contained in:
Vijay Janapa Reddi
2026-03-13 08:47:54 -04:00
parent 2bc58dba3d
commit 4206f3171b
5 changed files with 98 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
mlsysim/tests/test_solvers.py
View File

@@ -1,41 +0,0 @@
import pytest
from mlsysim.core.solver import DistributedModel, ReliabilityModel, EconomicsModel
from mlsysim.models import Models
from mlsysim.systems import Systems
from mlsysim.infra import Infra
def test_distributed_solver():
solver = DistributedModel()
gpt3 = Models.GPT3
cluster = Systems.Clusters.Research_256
result = solver.solve(gpt3, cluster, batch_size=32)
assert hasattr(result, "node_profile")
assert hasattr(result, "communication_latency")
assert hasattr(result, "scaling_efficiency")
assert result.scaling_efficiency > 0.0
assert result.scaling_efficiency <= 1.0
def test_reliability_solver():
solver = ReliabilityModel()
cluster = Systems.Clusters.Frontier_8K
result = solver.solve(cluster, job_duration_hours=100.0)
assert hasattr(result, "fleet_mtbf")
assert hasattr(result, "failure_probability")
assert hasattr(result, "optimal_checkpoint_interval")
assert result.failure_probability > 0.0
def test_economics_solver():
solver = EconomicsModel()
cluster = Systems.Clusters.Research_256
grid = Infra.Quebec
result = solver.solve(cluster, duration_days=30, grid=grid)
assert hasattr(result, "tco_usd")
assert hasattr(result, "carbon_footprint_kg")
assert result.tco_usd > 0
assert result.carbon_footprint_kg > 0