From 7ec50ff3cf31c720cec0e9379ad9d6f0a4c8b1fd Mon Sep 17 00:00:00 2001 From: project6 Date: Mon, 10 Aug 2026 07:55:39 +0000 Subject: [PATCH] =?UTF-8?q?fix:=20total=5Fmem=20=E2=86=92=20total=5Fmemory?= =?UTF-8?q?=20(corex=20torch=20API)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- verify_single_card.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/verify_single_card.py b/verify_single_card.py index 3e0dc8f1..a5caedf2 100644 --- a/verify_single_card.py +++ b/verify_single_card.py @@ -26,7 +26,7 @@ print(f" torch: {torch.__version__}") print(f" CUDA available: {torch.cuda.is_available()}") if torch.cuda.is_available(): print(f" GPU: {torch.cuda.get_device_name(0)}") - print(f" Memory: {torch.cuda.get_device_properties(0).total_mem / 1024**3:.1f} GB") + print(f" Memory: {torch.cuda.get_device_properties(0).total_memory / 1024**3:.1f} GB") else: print(" ERROR: No CUDA device!") sys.exit(1) @@ -322,4 +322,4 @@ print("=" * 70) print(f" CUDA kernel topk_softmax: {'✓ READY' if moe_ext else '✗ NOT AVAILABLE'}") print(f" GDN chunked (no NaN): {'✓' if not any_nan else '✗ HAS NaN'}") print(f" GPU memory used: {torch.cuda.memory_allocated()/1024**2:.0f} MB") -print(f" GPU memory free: {(torch.cuda.get_device_properties(0).total_mem - torch.cuda.memory_allocated())/1024**3:.1f} GB") +print(f" GPU memory free: {(torch.cuda.get_device_properties(0).total_memory - torch.cuda.memory_allocated())/1024**3:.1f} GB")