code_public/lidar_rendu
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Journaliser les replis CPU du GPU et retirer une copie VRAM inutile

Browse Source
This commit is contained in:
Antoine Jacquin
2026-09-18 21:23:39 +02:00
parent 93d1656d9f
commit f3708f8449
2 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
lidar_pipeline/gpu.py
View File

@ -355,7 +355,10 @@ def to_gpu(arr):
if _gpu_available():
try:
return _cp.asarray(arr.astype(np.float32))
except Exception:
except Exception as e:
# L'erreur d'origine (souvent OOM) ne doit pas être avalée :
# sans elle, un repli CPU est indissociable d'un simple bug.
logger.warning(f"Transfert GPU échoué ({e}) — repli CPU pour ce calcul")
disable_gpu()
return arr.astype(np.float32)
@ -386,7 +389,8 @@ def xp_gaussian_filter(arr, sigma):
if _cp is not None and isinstance(arr, _cp.ndarray):
try:
return _cp_ndimage.gaussian_filter(arr, sigma)
except Exception:
except Exception as e:
logger.warning(f"Filtre gaussien GPU échoué ({e}) — repli CPU")
arr = to_cpu(arr)
return ndimage.gaussian_filter(arr, sigma)
@ -395,7 +399,8 @@ def xp_uniform_filter(arr, size):
if _cp is not None and isinstance(arr, _cp.ndarray):
try:
return _cp_ndimage.uniform_filter(arr, size)
except Exception:
except Exception as e:
logger.warning(f"Filtre uniform GPU échoué ({e}) — repli CPU")
arr = to_cpu(arr)
return ndimage.uniform_filter(arr, size)
@ -404,7 +409,8 @@ def xp_minimum_filter(arr, footprint=None, size=None):
if _cp is not None and isinstance(arr, _cp.ndarray):
try:
return _cp_ndimage.minimum_filter(arr, footprint=footprint, size=size)
except Exception:
except Exception as e:
logger.warning(f"Filtre minimum GPU échoué ({e}) — repli CPU")
arr = to_cpu(arr)
return ndimage.minimum_filter(arr, footprint=footprint, size=size)
@ -413,7 +419,8 @@ def xp_maximum_filter(arr, footprint=None, size=None):
if _cp is not None and isinstance(arr, _cp.ndarray):
try:
return _cp_ndimage.maximum_filter(arr, footprint=footprint, size=size)
except Exception:
except Exception as e:
logger.warning(f"Filtre maximum GPU échoué ({e}) — repli CPU")
arr = to_cpu(arr)
return ndimage.maximum_filter(arr, footprint=footprint, size=size)