Skip to content
Commit 116dac00 authored by Christian Sigg's avatar Christian Sigg Committed by A. Unique TensorFlower
Browse files

Add AllReduceOp to GPU dialect with lowering to NVVM. 

The reduction operation is currently fixed to "add", and the scope is fixed to "workgroup".

The implementation is currently limited to sizes that are multiple 32 (warp size) and no larger than 1024.

PiperOrigin-RevId: 271290265
parent 94298cea
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment