Add some warnings for pin and unpin errors. (#11561)

This commit is contained in:
comfyanonymous
2025-12-29 15:26:42 -08:00
committed by GitHub
parent 9ca7e143af
commit 0e6221cc79

View File

@@ -1169,6 +1169,7 @@ def pin_memory(tensor):
TOTAL_PINNED_MEMORY += size
return True
else:
logging.warning("Pin error.")
discard_cuda_async_error()
return False
@@ -1199,6 +1200,7 @@ def unpin_memory(tensor):
TOTAL_PINNED_MEMORY = 0
return True
else:
logging.warning("Unpin error.")
discard_cuda_async_error()
return False