Constrain Output type for indexing ops (#131)

valadaptive · web-flow · commit 13423ce7f0e7 · 2025-11-17T06:35:28.000Z
An oversight from #112: we can't currently use indexing ops in the native-width associated types (`f32s`, etc.) because we never specified their output type. This PR fixes that.
diff --git a/fearless_simd/src/generated/simd_trait.rs b/fearless_simd/src/generated/simd_trait.rs
@@ -927,8 +927,8 @@ pub trait SimdBase<Element: SimdElement, S: Simd>:
     + 'static
     + crate::Bytes
     + SimdFrom<Element, S>
-    + core::ops::Index<usize>
-    + core::ops::IndexMut<usize>
+    + core::ops::Index<usize, Output = Element>
+    + core::ops::IndexMut<usize, Output = Element>
 {
     const N: usize;
     #[doc = r" A SIMD vector mask with the same number of elements."]
diff --git a/fearless_simd_gen/src/mk_simd_trait.rs b/fearless_simd_gen/src/mk_simd_trait.rs
@@ -64,7 +64,7 @@ fn mk_simd_base() -> TokenStream {
         pub trait SimdBase<Element: SimdElement, S: Simd>:
             Copy + Sync + Send + 'static
             + crate::Bytes + SimdFrom<Element, S>
-            + core::ops::Index<usize> + core::ops::IndexMut<usize>
+            + core::ops::Index<usize, Output = Element> + core::ops::IndexMut<usize, Output = Element>
         {
             const N: usize;
             /// A SIMD vector mask with the same number of elements.

Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,7 @@ fn mk_simd_base() -> TokenStream {`
`64`	`64`	`pub trait SimdBase<Element: SimdElement, S: Simd>:`
`65`	`65`	`Copy + Sync + Send + 'static`
`66`	`66`	`+ crate::Bytes + SimdFrom<Element, S>`
`67`		`- + core::ops::Index<usize> + core::ops::IndexMut<usize>`
	`67`	`+ + core::ops::Index<usize, Output = Element> + core::ops::IndexMut<usize, Output = Element>`
`68`	`68`	`{`
`69`	`69`	`const N: usize;`
`70`	`70`	`/// A SIMD vector mask with the same number of elements.`