[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
10d7ac5eb2
commit
ffd7932860
2 changed files with 9 additions and 2 deletions
|
|
@ -1673,7 +1673,9 @@ def resolve_lemonade_rocm_choice(
|
|||
|
||||
|
||||
def _lemonade_release_plans(
|
||||
llama_tag: str, host: HostInfo, published_release_tag: str = ""
|
||||
llama_tag: str,
|
||||
host: HostInfo,
|
||||
published_release_tag: str = "",
|
||||
) -> tuple[str, list[InstallReleasePlan]]:
|
||||
"""Single-attempt plan for a data-center AMD GPU, sourced from lemonade.
|
||||
release_tag is lemonade's own counter so updates compare against lemonade;
|
||||
|
|
|
|||
|
|
@ -333,7 +333,11 @@ class TestResolveUpstreamAssetChoice:
|
|||
# TEST: install_llama_prebuilt.py -- data-center ROCm (lemonade) routing
|
||||
|
||||
|
||||
def _lemonade_release(tag = "b1300", published_at = "2026-01-01T00:00:00Z", gfxs = None):
|
||||
def _lemonade_release(
|
||||
tag = "b1300",
|
||||
published_at = "2026-01-01T00:00:00Z",
|
||||
gfxs = None,
|
||||
):
|
||||
base = f"https://github.com/lemonade-sdk/llamacpp-rocm/releases/download/{tag}"
|
||||
arches = gfxs if gfxs is not None else ["gfx908", "gfx90a"]
|
||||
names = [f"llama-{tag}-ubuntu-rocm-{g}-x64.zip" for g in arches]
|
||||
|
|
@ -447,6 +451,7 @@ class TestDataCenterLemonadeRouting:
|
|||
if "lemonade-sdk" in url:
|
||||
return [_lemonade_release(tag = "b1301")] # would win if pin ignored
|
||||
return {"tag_name": "b9637"}
|
||||
|
||||
mock_fetch.side_effect = _router
|
||||
host = rocm_host(rocm_gfx_target = "gfx908")
|
||||
_tag, plans = prebuilt_mod.resolve_simple_install_release_plans(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue