mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-30 21:19:53 +00:00
Add some warnings for pin and unpin errors. (#11561)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user