BinaryExpr: account for ignore_tile_position when checking preconditions

ValeevGroup · Sep 18, 2024 · a0c0450 · a0c0450
1 parent d622383
commit a0c0450
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/TiledArray/dist_eval/binary_eval.h b/src/TiledArray/dist_eval/binary_eval.h
@@ -107,7 +107,10 @@ class BinaryEvalImpl : public DistEvalImpl<typename Op::result_type, Policy>,
         right_ntiles_discarded_(0)
 #endif
   {
-    TA_ASSERT(left.trange() == right.trange());
+    TA_ASSERT(ignore_tile_position()
+                  ? left.trange().elements_range().extent() ==
+                        right.trange().elements_range().extent()
+                  : left.trange() == right.trange());
   }
 
   virtual ~BinaryEvalImpl() {}