mrq/bitsandbytes-rocm
1
0
Fork 1
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

Removing unnecessary else's

Browse Source
This commit is contained in:
Tom Aarsen 2022-10-27 13:25:07 +02:00
parent 4a05df34c2
commit b5cf706341
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
bitsandbytes/functional.py
View File

@ -133,8 +133,7 @@ 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)
def create_dynamic_map(signed=True, n=7): def create_dynamic_map(signed=True, n=7):
@ -186,13 +185,12 @@ 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"