mawei/StarPilot
1
0
Fork 0
mirror of https://github.com/firestar5683/StarPilot.git synced 2026-09-30 11:23:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make external GPU CPU pinning conditional

Browse Source
This commit is contained in:
firestar5683
2026-09-08 18:48:40 -05:00
parent 2360ff9b0f
commit ca3d8a3816
2 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
starpilot/assets/tests/test_model_pipeline.py
+10
View File
@@ -405,10 +405,20 @@ def test_external_gpu_compile_uses_agnos_isolated_cpu(monkeypatch):
command = ["python3", "compile_modeld.py"]
monkeypatch.setattr(model_compiler.sys, "platform", "linux")
monkeypatch.setattr(model_compiler.platform, "machine", lambda: "aarch64")
monkeypatch.setattr(model_compiler.os, "sched_getaffinity", lambda _: {7}, raising=False)
assert model_compiler.external_gpu_compile_command(command) == ["taskset", "-c", "7", *command]
def test_external_gpu_compile_skips_unavailable_agnos_cpu(monkeypatch):
command = ["python3", "compile_modeld.py"]
monkeypatch.setattr(model_compiler.sys, "platform", "linux")
monkeypatch.setattr(model_compiler.platform, "machine", lambda: "aarch64")
monkeypatch.setattr(model_compiler.os, "sched_getaffinity", lambda _: {0, 1, 2, 3}, raising=False)
assert model_compiler.external_gpu_compile_command(command) is command
def test_external_gpu_compile_does_not_pin_other_platforms(monkeypatch):
command = ["python3", "compile_modeld.py"]
monkeypatch.setattr(model_compiler.sys, "platform", "darwin")