Bug fix.

ValeevGroup · May 29, 2024 · 64dbcb1 · 64dbcb1
1 parent fa242b4
commit 64dbcb1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/TiledArray/einsum/tiledarray.h b/src/TiledArray/einsum/tiledarray.h
@@ -256,6 +256,7 @@ auto replicate_array(Array from, TiledRange const &prepend_trng) {
         auto res_coord_ix = res_tr.element_to_tile(res_rng.lobound());
         auto from_coord_ix = decltype(res_coord_ix)(
             next(begin(res_coord_ix), delta_rank), end(res_coord_ix));
+        if (from.is_zero(from_coord_ix)) return typename Array::scalar_type{0};
         replicate_tensor(repped, from.find_local(from_coord_ix).get(false));
         tile = repped;
         return tile.norm();