Add no-dim version of simd_reduce_sum/mean

microsoft / microxcaling

PyTorch emulation library for Microscaling (MX)-compatible data formats

MIT License

164 stars 21 forks source link

Add no-dim version of simd_reduce_sum/mean #20

Closed rizhao-msft closed 9 months ago

rizhao-msft commented 9 months ago

torch.sum (and similar functions) have two versions, one with a dim argument, and one without. This update should allow simd_reduce_sum/mean to support both. Previously only the with-dim version was supported.

Should fix #18