cuda module: fix type signature for oneshot function
It is only ever invoked once. It has a statically known signature based on how it's used, so there's no good reason to allow it to accept any defaults -- and defaulting to None messes with type safety. Saturate is always given, so don't claim it is only sometimes.
This commit is contained in:
parent
cf35d9b4ce
commit
8ff25c0bca
|
@ -157,11 +157,11 @@ class CudaModule(NewExtensionModule):
|
|||
|
||||
return cuda_version, arch_list, detected
|
||||
|
||||
def _filter_cuda_arch_list(self, cuda_arch_list, lo=None, hi=None, saturate=None):
|
||||
def _filter_cuda_arch_list(self, cuda_arch_list, lo: str, hi: T.Optional[str], saturate: str) -> T.List[str]:
|
||||
"""
|
||||
Filter CUDA arch list (no codenames) for >= low and < hi architecture
|
||||
bounds, and deduplicate.
|
||||
If saturate is provided, architectures >= hi are replaced with saturate.
|
||||
Architectures >= hi are replaced with saturate.
|
||||
"""
|
||||
|
||||
filtered_cuda_arch_list = []
|
||||
|
@ -170,8 +170,6 @@ class CudaModule(NewExtensionModule):
|
|||
if lo and version_compare(arch, '<' + lo):
|
||||
continue
|
||||
if hi and version_compare(arch, '>=' + hi):
|
||||
if not saturate:
|
||||
continue
|
||||
arch = saturate
|
||||
if arch not in filtered_cuda_arch_list:
|
||||
filtered_cuda_arch_list.append(arch)
|
||||
|
|
Loading…
Reference in New Issue