xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-14 08:43:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

worst PR ever

Browse Source
This commit is contained in:
Christopher Anderson 2025-08-25 09:38:22 +10:00
parent 2a06dc8e87
commit 33c43b68c3
1 changed files with 100 additions and 76 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
comfy/customzluda/zluda.py
View File

@ -473,7 +473,9 @@ except Exception as e:
MEM_BUS_WIDTH = { MEM_BUS_WIDTH = {
"AMD Radeon RX 9070 XT": 256, "AMD Radeon RX 9070 XT": 256,
"AMD Radeon RX 9070": 256, "AMD Radeon RX 9070": 256,
"AMD Radeon RX 9060 XT": 192, "AMD Radeon RX 9070 GRE": 192,
"AMD Radeon RX 9060 XT": 128,
"AMD Radeon RX 9060": 128,
"AMD Radeon RX 7900 XTX": 384, "AMD Radeon RX 7900 XTX": 384,
"AMD Radeon RX 7900 XT": 320, "AMD Radeon RX 7900 XT": 320,
"AMD Radeon RX 7900 GRE": 256, "AMD Radeon RX 7900 GRE": 256,
@ -483,12 +485,14 @@ MEM_BUS_WIDTH = {
"AMD Radeon RX 7650 GRE": 128, "AMD Radeon RX 7650 GRE": 128,
"AMD Radeon RX 7600 XT": 128, "AMD Radeon RX 7600 XT": 128,
"AMD Radeon RX 7600": 128, "AMD Radeon RX 7600": 128,
"AMD Radeon RX 7500 XT": 96, "AMD Radeon RX 7400": 128,
"AMD Radeon RX 6950 XT": 256, "AMD Radeon RX 6950 XT": 256,
"AMD Radeon RX 6900 XT": 256, "AMD Radeon RX 6900 XT": 256,
"AMD Radeon RX 6800 XT": 256, "AMD Radeon RX 6800 XT": 256,
"AMD Radeon RX 6800": 256, "AMD Radeon RX 6800": 256,
"AMD Radeon RX 6750 XT": 192, "AMD Radeon RX 6750 XT": 192,
"AMD Radeon RX 6750 GRE 12GB": 192,
"AMD Radeon RX 6750 GRE 10GB": 160,
"AMD Radeon RX 6700 XT": 192, "AMD Radeon RX 6700 XT": 192,
"AMD Radeon RX 6700": 160, "AMD Radeon RX 6700": 160,
"AMD Radeon RX 6650 XT": 128, "AMD Radeon RX 6650 XT": 128,
@ -496,6 +500,26 @@ MEM_BUS_WIDTH = {
"AMD Radeon RX 6600": 128, "AMD Radeon RX 6600": 128,
"AMD Radeon RX 6500 XT": 64, "AMD Radeon RX 6500 XT": 64,
"AMD Radeon RX 6400": 64, "AMD Radeon RX 6400": 64,
"AMD Radeon RX 5700 XT": 256,
"AMD Radeon RX 5700": 256,
"AMD Radeon RX 5600 XT": 192,
"AMD Radeon RX 5500 XT": 128,
"AMD Radeon RX 5500": 128,
"AMD Radeon RX 5300": 96,
# AMD Radeon Pro R9000/W7000/W6000/W5000 series, Apple exclusive WX series not listed
"AMD Radeon AI PRO R9700": 256,
"AMD Radeon PRO W7900": 384,
"AMD Radeon PRO W7800 48GB": 384,
"AMD Radeon PRO W7800": 256,
"AMD Radeon PRO W7700": 256,
"AMD Radeon PRO W7600": 128,
"AMD Radeon PRO W7500": 128,
"AMD Radeon PRO W7400": 128,
"AMD Radeon PRO W6800": 256,
"AMD Radeon PRO W6600": 128,
"AMD Radeon PRO W6400": 64,
"AMD Radeon PRO W5700": 256,
"AMD Radeon PRO W5500": 128,
} }
# ------------------- Device Properties Implementation ------------------- # ------------------- Device Properties Implementation -------------------
@ -513,23 +537,23 @@ class DeviceProperties:
# # ------------------- Audio Ops Patch ------------------- # # ------------------- Audio Ops Patch -------------------
# if is_zluda: # if is_zluda:
# _torch_stft = torch.stft # _torch_stft = torch.stft
# _torch_istft = torch.istft # _torch_istft = torch.istft
# def z_stft(input: torch.Tensor, window: torch.Tensor, *args, **kwargs): # def z_stft(input: torch.Tensor, window: torch.Tensor, *args, **kwargs):
# return _torch_stft(input=input.cpu(), window=window.cpu(), *args, **kwargs).to(input.device) # return _torch_stft(input=input.cpu(), window=window.cpu(), *args, **kwargs).to(input.device)
# def z_istft(input: torch.Tensor, window: torch.Tensor, *args, **kwargs): # def z_istft(input: torch.Tensor, window: torch.Tensor, *args, **kwargs):
# return _torch_istft(input=input.cpu(), window=window.cpu(), *args, **kwargs).to(input.device) # return _torch_istft(input=input.cpu(), window=window.cpu(), *args, **kwargs).to(input.device)
# def z_jit(f, *_, **__): # def z_jit(f, *_, **__):
# f.graph = torch._C.Graph() # f.graph = torch._C.Graph()
# return f # return f
# torch._dynamo.config.suppress_errors = True # torch._dynamo.config.suppress_errors = True
# torch.stft = z_stft # torch.stft = z_stft
# torch.istft = z_istft # torch.istft = z_istft
# torch.jit.script = z_jit # torch.jit.script = z_jit
# # ------------------- End Audio Patch ------------------- # # ------------------- End Audio Patch -------------------
# ------------------- Top-K Fallback Patch ------------------- # ------------------- Top-K Fallback Patch -------------------