From 5e5f551b04ca8443e9e9dfd248f4fbc353b22cd9 Mon Sep 17 00:00:00 2001
From: Adam Kern <akern40@gmail.com>
Date: Thu, 8 Aug 2024 16:28:37 -0400
Subject: [PATCH] Uses broadcast indices to remove D::Smaller: Copy trait bound

---
 src/tri.rs | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/src/tri.rs b/src/tri.rs
index 53dc43fda..b7d297fcc 100644
--- a/src/tri.rs
+++ b/src/tri.rs
@@ -17,7 +17,6 @@ use crate::{
     Axis,
     Data,
     Dimension,
-    IntoDimension,
     Zip,
 };
 
@@ -26,7 +25,6 @@ where
     S: Data<Elem = A>,
     D: Dimension,
     A: Clone + Zero,
-    D::Smaller: Copy,
 {
     /// Upper triangular of an array.
     ///
@@ -74,10 +72,12 @@ where
 
         let mut res = Array::zeros(self.raw_dim());
         let ncols = self.len_of(Axis(n - 1));
-        Zip::indexed(self.rows())
+        let nrows = self.len_of(Axis(n - 2));
+        let indices = Array::from_iter(0..nrows);
+        Zip::from(self.rows())
             .and(res.rows_mut())
-            .for_each(|i, src, mut dst| {
-                let row_num = i.into_dimension().last_elem();
+            .and_broadcast(&indices)
+            .for_each(|src, mut dst, row_num| {
                 let mut lower = match k >= 0 {
                     true => row_num.saturating_add(k as usize),        // Avoid overflow
                     false => row_num.saturating_sub(k.unsigned_abs()), // Avoid underflow, go to 0
@@ -135,10 +135,13 @@ where
 
         let mut res = Array::zeros(self.raw_dim());
         let ncols = self.len_of(Axis(n - 1));
-        Zip::indexed(self.rows())
+        let nrows = self.len_of(Axis(n - 2));
+        let indices = Array::from_iter(0..nrows);
+        Zip::from(self.rows())
             .and(res.rows_mut())
-            .for_each(|i, src, mut dst| {
-                let row_num = i.into_dimension().last_elem();
+            .and_broadcast(&indices)
+            .for_each(|src, mut dst, row_num| {
+                // let row_num = i.into_dimension().last_elem();
                 let mut upper = match k >= 0 {
                     true => row_num.saturating_add(k as usize).saturating_add(1), // Avoid overflow
                     false => row_num.saturating_sub((k + 1).unsigned_abs()),      // Avoid underflow