Set deepgemm to the default value in the hopper architecture. (#4613)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user