Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

add cdna check

Browse Source
This commit is contained in:
Seunghoon Lee
2024-10-26 14:09:50 +09:00
parent 81bd236cc3
commit a76893bd72
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/devices.py
+4 -4
View File
@@ -285,13 +285,13 @@ def test_bf16():
bf16_ok = False
return bf16_ok
elif backend == 'rocm' or backend == 'zluda':
gcn_arch = None
agent = None
if backend == 'rocm':
gcn_arch = rocm.Agent.parse_gfx_version(getattr(torch.cuda.get_device_properties(device), "gcnArchName", "gfx0000"))
agent = rocm.Agent(getattr(torch.cuda.get_device_properties(device), "gcnArchName", "gfx0000"))
else:
from modules.zluda_installer import default_agent
gcn_arch = 0x0 if default_agent is None else default_agent.gfx_version
if gcn_arch < 0x1100: # all cards before RDNA 3
agent = default_agent
if agent is not None and agent.gfx_version < 0x1100 and agent.arch != rocm.MicroArchitecture.CDNA: # all cards before RDNA 3 except for CDNA cards
bf16_ok = False
return bf16_ok
try: