Merge pull request #73 from stephenswat/fix/strided_copy_scalar

stephenswat · web-flow · commit 368bc7b68944 · 2025-04-10T15:33:17.000+02:00
Allow strided copy to use scalar coordinates
diff --git a/lib/core/covfie/core/backend/transformer/strided.hpp b/lib/core/covfie/core/backend/transformer/strided.hpp
@@ -82,7 +82,12 @@ struct strided {
 
                 for (std::size_t i = 0; i < covariant_output_t::dimensions; ++i)
                 {
-                    res[idx][i] = nother.at(t)[i];
+                    if constexpr (covariant_output_t::dimensions == 1 && !std::unsigned_integral<std::decay_t<decltype(t)>>)
+                    {
+                        res[idx][i] = nother.at(t[0])[i];
+                    } else {
+                        res[idx][i] = nother.at(t)[i];
+                    }
                 }
             },
             sizes

Original file line number	Diff line number	Diff line change
`@@ -82,7 +82,12 @@ struct strided {`
`82`	`82`
`83`	`83`	`for (std::size_t i = 0; i < covariant_output_t::dimensions; ++i)`
`84`	`84`	`{`
`85`		`- res[idx][i] = nother.at(t)[i];`
	`85`	`+ if constexpr (covariant_output_t::dimensions == 1 && !std::unsigned_integral<std::decay_t<decltype(t)>>)`
	`86`	`+ {`
	`87`	`+ res[idx][i] = nother.at(t[0])[i];`
	`88`	`+ } else {`
	`89`	`+ res[idx][i] = nother.at(t)[i];`
	`90`	`+ }`
`86`	`91`	`}`
`87`	`92`	`},`
`88`	`93`	`sizes`