Set deepgemm to the default value in the hopper architecture. (#4613)

This commit is contained in:
lukec
2025-03-21 13:03:00 +08:00
committed by GitHub
parent 7b5fc71972
commit 4c56e5dbee
2 changed files with 16 additions and 3 deletions

View File

@@ -1006,6 +1006,13 @@ def get_amdgpu_memory_capacity():
)
def get_device_sm():
if torch.cuda.is_available():
major, minor = torch.cuda.get_device_capability()
return major * 10 + minor
return 0
def get_nvgpu_memory_capacity():
try:
# Run nvidia-smi and capture the output