How to optimize ReductionSum with Metal?

Graphics & Games General
PYNing OP
Created Jun ’22
Replies 1
Boosts 0
Views 737
Participants 2

I would like to write a ReductionSum Metal Shader like this:

https://github.com/alibaba/MNN/blob/master/source/backend/metal/MetalReduction.metal#L32

Sometimes the reduced dimension is large while the other dimensions is small, which cause few threads can be launched and inefficient.

Is there any way to optimize it?

Replies  1
Boosts  0
Views  737
Participants  2
iPerKard OP
Jun ’22

I suggest you to read the comprehensive article "Optimizing Parallel Reduction in Metal for Apple M1" from Matthew Kieber-Emmons : https://kieber-emmons.medium.com/optimizing-parallel-reduction-in-metal-for-apple-m1-8e8677b49b01

0
How to optimize ReductionSum with Metal?
First post date Last post date
 
 
Q