AdamNiederer / AdamNiederer/faster

how to accomplish `zip_mut_with` in faster?

未关闭
#37 2 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看
主要语言
Rust
星标
1.6k
派生
52
PR 合并指标
30 天内没有已合并 PR

描述

hey,

I have spent a few minutes trying unsuccessfully to write a simd version of code that uses ndarray's [`zip_mut_with`](https://docs.rs/ndarray/0.10.14/ndarray/struct.ArrayBase.html#method.zip_mut_with) - is there currently a way of combining a `simd_iter_mut` and `simd_iter` with `zip`?

Essentially, I'd like to modify the elements of an array in place with an operation that uses the values in another array of the same shape.

I tried this:

```rust
let mut xs = ndarray::Array::from_elem((64,), 0.0f32);
let ys = ndarray::Array::from_elem((64,), 1.0f32);
(xs.as_slice_mut().unwrap().simd_iter_mut(f32s(0.0)),
ys.as_slice().unwrap().simd_iter(f32s(0.0)))
.zip()
.simd_for_each(|(x, y)| {
*x += *y; // or whatever, not sure if this is correct for the closure
});
```

and got this error:

```console
error[E0599]: no method named `simd_for_each` found for type `faster::Zip<(faster::SIMDIter<&mut [f32]>, faster::SIMDIter<&[f32]>)>` in the current scope
--> src/mlp/m3r.rs:298:26
|
298 | .simd_for_each(|(z, b)| {
| ^^^^^^^^^^^^^
...
```

Any recommendations for how to proceed? First time using the library, apologies if this is an obvious question.

贡献指南

这个仓库没有索引到贡献指南

评估

这个 Issue 还没有评估数据。

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。