diff --git a/loopy/check.py b/loopy/check.py index da4a5618f..29f7b3e9f 100644 --- a/loopy/check.py +++ b/loopy/check.py @@ -1359,7 +1359,7 @@ def check_for_nested_base_storage(kernel: LoopKernel) -> None: for ary in kernel.temporary_variables.values(): if ary.base_storage: - storage_array = name_to_array.get(ary.base_storage, None) + storage_array = name_to_array.get(ary.base_storage) if storage_array is None: raise LoopyError("Nothing known about storage array " diff --git a/loopy/kernel/tools.py b/loopy/kernel/tools.py index bd5f0249e..ce48b27eb 100644 --- a/loopy/kernel/tools.py +++ b/loopy/kernel/tools.py @@ -857,7 +857,7 @@ def get_auto_axis_iname_ranking_by_stride(kernel, insn): if var.name in auto_axis_inames: # excludes '1', i.e. the constant new_stride = coeff*stride - old_stride = iname_to_stride_expr.get(var.name, None) + old_stride = iname_to_stride_expr.get(var.name) if old_stride is None or new_stride < old_stride: iname_to_stride_expr[var.name] = new_stride