CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-05-13 02:34:43 +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:07:21 +02:00
parent 9b878431d3
commit 343b7f6aad
2 changed files with 4 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/libraries/amd_smi.py
-6
View File
@@ -106,12 +106,6 @@ def init_ctypes(amd_smi : ctypes.CDLL) -> ctypes.CDLL:
amd_smi.amdsmi_get_gpu_vram_usage.argtypes = [ ctypes.c_void_p, ctypes.c_void_p ]
amd_smi.amdsmi_get_gpu_vram_usage.restype = ctypes.c_uint32
amd_smi.amdsmi_get_gpu_memory_total.argtypes = [ ctypes.c_void_p, ctypes.c_uint32, ctypes.POINTER(ctypes.c_uint64) ]
amd_smi.amdsmi_get_gpu_memory_total.restype = ctypes.c_uint32
amd_smi.amdsmi_get_gpu_memory_usage.argtypes = [ ctypes.c_void_p, ctypes.c_uint32, ctypes.POINTER(ctypes.c_uint64) ]
amd_smi.amdsmi_get_gpu_memory_usage.restype = ctypes.c_uint32
amd_smi.amdsmi_get_gpu_activity.argtypes = [ ctypes.c_void_p, ctypes.c_void_p ]
amd_smi.amdsmi_get_gpu_activity.restype = ctypes.c_uint32
facefusion/system.py
+4 -7
View File
@@ -130,11 +130,8 @@ def detect_amd_graphic_devices() -> List[GraphicDevice]:
asic_info = amd_smi_module.create_asic_info_configuration()
amd_smi_library.amdsmi_get_gpu_asic_info(device_handle, ctypes.byref(asic_info))
memory_total = ctypes.c_uint64()
amd_smi_library.amdsmi_get_gpu_memory_total(device_handle, 0, ctypes.byref(memory_total))
memory_usage = ctypes.c_uint64()
amd_smi_library.amdsmi_get_gpu_memory_usage(device_handle, 0, ctypes.byref(memory_usage))
memory = amd_smi_module.create_memory_configuration()
amd_smi_library.amdsmi_get_gpu_vram_usage(device_handle, ctypes.byref(memory))
temperature = ctypes.c_int64()
amd_smi_library.amdsmi_get_temp_metric(device_handle, 0, 0, ctypes.byref(temperature))
@@ -159,12 +156,12 @@ def detect_amd_graphic_devices() -> List[GraphicDevice]:
{
'total':
{
'value': memory_total.value // (1024 * 1024 * 1024),
'value': memory.vram_total // (1024 * 1024 * 1024),
'unit': 'GB'
},
'used':
{
'value': memory_usage.value // (1024 * 1024 * 1024),
'value': memory.vram_used // (1024 * 1024 * 1024),
'unit': 'GB'
}
},