CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-07-09 03:57:58 +02:00
Code Issues Packages Projects Releases Wiki Activity

simplify memory call for amd

Browse Source
This commit is contained in:
henryruhs
2026-05-08 13:30:15 +02:00
parent 343b7f6aad
commit f8c90b4b25
2 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/libraries/amd_smi.py
+2 -4
View File
@@ -121,13 +121,13 @@ def init_ctypes(amd_smi : ctypes.CDLL) -> ctypes.CDLL:
def find_device_handles(amd_smi_library : ctypes.CDLL) -> List[ctypes.c_void_p]: def find_device_handles(amd_smi_library : ctypes.CDLL) -> List[ctypes.c_void_p]:
device_handles : List[ctypes.c_void_p] = [] device_handles : List[ctypes.c_void_p] = []
socket_count = ctypes.c_uint32(0) socket_count = ctypes.c_uint32()
amd_smi_library.amdsmi_get_socket_handles(ctypes.byref(socket_count), ctypes.POINTER(ctypes.c_void_p)()) amd_smi_library.amdsmi_get_socket_handles(ctypes.byref(socket_count), ctypes.POINTER(ctypes.c_void_p)())
socket_handles = (ctypes.c_void_p * socket_count.value)() socket_handles = (ctypes.c_void_p * socket_count.value)()
amd_smi_library.amdsmi_get_socket_handles(ctypes.byref(socket_count), socket_handles) amd_smi_library.amdsmi_get_socket_handles(ctypes.byref(socket_count), socket_handles)
for socket_index in range(socket_count.value): for socket_index in range(socket_count.value):
device_count = ctypes.c_uint32(0) device_count = ctypes.c_uint32()
amd_smi_library.amdsmi_get_processor_handles(socket_handles[socket_index], ctypes.byref(device_count), ctypes.POINTER(ctypes.c_void_p)()) amd_smi_library.amdsmi_get_processor_handles(socket_handles[socket_index], ctypes.byref(device_count), ctypes.POINTER(ctypes.c_void_p)())
processor_handles = (ctypes.c_void_p * device_count.value)() processor_handles = (ctypes.c_void_p * device_count.value)()
amd_smi_library.amdsmi_get_processor_handles(socket_handles[socket_index], ctypes.byref(device_count), processor_handles) amd_smi_library.amdsmi_get_processor_handles(socket_handles[socket_index], ctypes.byref(device_count), processor_handles)
@@ -136,5 +136,3 @@ def find_device_handles(amd_smi_library : ctypes.CDLL) -> List[ctypes.c_void_p]:
device_handles.append(ctypes.c_void_p(processor_handles[device_index])) device_handles.append(ctypes.c_void_p(processor_handles[device_index]))
return device_handles return device_handles
facefusion/libraries/nvidia_ml.py
-2
View File
@@ -93,5 +93,3 @@ def find_device_handles(nvidia_ml_library : ctypes.CDLL) -> List[ctypes.c_void_p
device_handles.append(device_handle) device_handles.append(device_handle)
return device_handles return device_handles