From 110de04bde476c892e28d362a8a5a9853cb7c8b8 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 6 Feb 2017 15:19:03 -0600
Subject: [PATCH] align_spaces: detect identical inames in other dim_types
---
islpy/__init__.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/islpy/__init__.py b/islpy/__init__.py
index b96eebc..0f81310 100644
--- a/islpy/__init__.py
+++ b/islpy/__init__.py
@@ -1166,10 +1166,12 @@ def align_spaces(obj, tgt, obj_bigger_ok=False, across_dim_types=False):
for dt in dim_types:
obj = _align_dim_type(dt, obj, tgt, obj_bigger_ok, obj_names, tgt_names)
else:
- for dt in dim_types:
- obj_names = [obj.get_dim_name(dt, i) for i in range(obj.dim(dt))]
- tgt_names = [tgt.get_dim_name(dt, i) for i in range(tgt.dim(dt))]
+ obj_names = [obj.get_dim_name(dt, i)
+ for dt in dim_types for i in range(obj.dim(dt))]
+ tgt_names = [tgt.get_dim_name(dt, i)
+ for dt in dim_types for i in range(tgt.dim(dt))]
+ for dt in dim_types:
obj = _align_dim_type(dt, obj, tgt, obj_bigger_ok, obj_names, tgt_names)
return obj
--
GitLab