mirror of
https://github.com/unslothai/unsloth.git
synced 2026-08-15 03:43:55 +00:00
This commit is contained in:
parent
e93eeceea3
commit
c160215048
1 changed files with 13 additions and 2 deletions
|
|
@ -320,7 +320,13 @@ def test_rocm_apu_inventory_keeps_the_gtt_total(monkeypatch):
|
|||
class _FakeRocmProps(_FakeProps):
|
||||
"""Discrete card as a current ROCm runtime describes it: the flag is filled in."""
|
||||
|
||||
def __init__(self, name = "MI300X", total = 191505498112, arch = "gfx942", integrated = 0):
|
||||
def __init__(
|
||||
self,
|
||||
name = "MI300X",
|
||||
total = 191505498112,
|
||||
arch = "gfx942",
|
||||
integrated = 0,
|
||||
):
|
||||
super().__init__(name, total)
|
||||
self.gcnArchName = arch
|
||||
self.is_integrated = integrated
|
||||
|
|
@ -396,7 +402,12 @@ def test_rocm_apu_sysfs_overlay_still_declines_against_the_gtt_total(monkeypatch
|
|||
class _FakeUnclassifiedApuProps(_FakeProps):
|
||||
"""gfx1103 Phoenix: a real APU that sits outside the classifier's arch set."""
|
||||
|
||||
def __init__(self, total = _APU_CARVE_OUT, *, integrated = 0):
|
||||
def __init__(
|
||||
self,
|
||||
total = _APU_CARVE_OUT,
|
||||
*,
|
||||
integrated = 0,
|
||||
):
|
||||
super().__init__("AMD Radeon 780M Graphics", total)
|
||||
self.gcnArchName = "gfx1103"
|
||||
if integrated is not None:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue