Commit b5cf7063 authored by Tom Aarsen's avatar Tom Aarsen
Browse files

Removing unnecessary else's

parent 4a05df34
...@@ -133,7 +133,6 @@ class Cusparse_Context: ...@@ -133,7 +133,6 @@ class Cusparse_Context:
def create_linear_map(signed=True): def create_linear_map(signed=True):
if signed: if signed:
return torch.linspace(-1.0, 1.0, 256) return torch.linspace(-1.0, 1.0, 256)
else:
return torch.linspace(0.0, 1.0, 256) return torch.linspace(0.0, 1.0, 256)
...@@ -186,12 +185,11 @@ def create_dynamic_map(signed=True, n=7): ...@@ -186,12 +185,11 @@ def create_dynamic_map(signed=True, n=7):
def get_special_format_str(): def get_special_format_str():
if not torch.cuda.is_available(): return 'col_turing' if not torch.cuda.is_available(): return 'col_turing'
major, minor = torch.cuda.get_device_capability() major, _minor = torch.cuda.get_device_capability()
if major <= 7: if major <= 7:
return "col_turing" return "col_turing"
elif major == 8: if major == 8:
return "col_ampere" return "col_ampere"
else:
return "col_turing" return "col_turing"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment