Krzysztof Drewniak 84718d37db [MLIR][GPU] Add gpu.set_default_device op ...

This op is added to allow MLIR code running on multi-GPU systems to
select the GPU they want to execute operations on when no GPU is
otherwise specified.

Reviewed By: mehdi_amini

Differential Revision: https://reviews.llvm.org/D119883

2022-02-17 21:30:09 +00:00

7.5 KiB

Raw Blame History

View Raw