From d859cd8aecba2899606380f21b9e22bef37dd540 Mon Sep 17 00:00:00 2001
From: Alexandru Fikl <alexfikl@gmail.com>
Date: Mon, 27 Sep 2021 10:14:51 -0500
Subject: [PATCH] complain if unflattened strides do not match
---
arraycontext/container/traversal.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/arraycontext/container/traversal.py b/arraycontext/container/traversal.py
index 08d9ba9..e7bf731 100644
--- a/arraycontext/container/traversal.py
+++ b/arraycontext/container/traversal.py
@@ -582,6 +582,12 @@ def unflatten(
"functionality that is not implemented in the array "
f"context '{type(actx).__name__}'") from exc
+ if hasattr(template_subary, "strides"):
+ if template_subary.strides != subary.strides:
+ raise ValueError(
+ f"strides do not match template: got {subary.strides}, "
+ f"expected {template_subary.strides}")
+
return subary
else:
return deserialize_container(template_subary, [
--
GitLab