CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-08-29 12:00:38 +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 149324553d
commit d9180bd4cf
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.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_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.argtypes = [ ctypes.c_void_p, ctypes.c_void_p ]
amd_smi.amdsmi_get_gpu_activity.restype = ctypes.c_uint32 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() asic_info = amd_smi_module.create_asic_info_configuration()
amd_smi_library.amdsmi_get_gpu_asic_info(device_handle, ctypes.byref(asic_info)) amd_smi_library.amdsmi_get_gpu_asic_info(device_handle, ctypes.byref(asic_info))
memory_total = ctypes.c_uint64() memory = amd_smi_module.create_memory_configuration()
amd_smi_library.amdsmi_get_gpu_memory_total(device_handle, 0, ctypes.byref(memory_total)) amd_smi_library.amdsmi_get_gpu_vram_usage(device_handle, ctypes.byref(memory))
memory_usage = ctypes.c_uint64()
amd_smi_library.amdsmi_get_gpu_memory_usage(device_handle, 0, ctypes.byref(memory_usage))
temperature = ctypes.c_int64() temperature = ctypes.c_int64()
amd_smi_library.amdsmi_get_temp_metric(device_handle, 0, 0, ctypes.byref(temperature)) 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': 'total':
{ {
'value': memory_total.value // (1024 * 1024 * 1024), 'value': memory.vram_total // (1024 * 1024 * 1024),
'unit': 'GB' 'unit': 'GB'
}, },
'used': 'used':
{ {
'value': memory_usage.value // (1024 * 1024 * 1024), 'value': memory.vram_used // (1024 * 1024 * 1024),
'unit': 'GB' 'unit': 'GB'
} }
}, },