dylanyunlong/project_6
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

fix: total_mem → total_memory

Browse Source
This commit is contained in:
Claude
2026-08-11 13:18:53 +00:00
parent 2f19498ae6
commit c6b9ee93e9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
verify_forward.py
View File

@@ -11,7 +11,7 @@ os.environ.setdefault("BI100_GDN_COREX_PACKED_DECODE", "1")
import torch
print(f"torch {torch.__version__}, CUDA {torch.cuda.is_available()}")
if torch.cuda.is_available():
print(f"GPU: {torch.cuda.get_device_name(0)}, {torch.cuda.get_device_properties(0).total_mem // 1024**2} MB")
print(f"GPU: {torch.cuda.get_device_name(0)}, {torch.cuda.get_device_properties(0).total_memory // 1024**2} MB")
# Step 1: 验证所有.so加载
print("\n=== Step 1: .so加载 ===")