llvm-project/mlir/test/mlir-cuda-runner
Christian Sigg d2f0f847af Support custom accumulator provided as region to gpu.all_reduce.
In addition to specifying the type of accumulation through the 'op' attribute, the accumulation can now also be specified as arbitrary code region.

Adds a gpu.yield op to specify the result of the accumulation.

Also support more types (integers) and accumulations (mul).

PiperOrigin-RevId: 275065447
2019-10-16 10:43:44 -07:00
..
all-reduce-op.mlir Support custom accumulator provided as region to gpu.all_reduce. 2019-10-16 10:43:44 -07:00
all-reduce-region.mlir Support custom accumulator provided as region to gpu.all_reduce. 2019-10-16 10:43:44 -07:00
gpu-to-cubin.mlir JitRunner: support entry functions returning void 2019-08-20 07:46:17 -07:00
lit.local.cfg Add an mlir-cuda-runner tool. 2019-07-04 07:53:54 -07:00