[all-commits] [llvm/llvm-project] 84718d: [MLIR][GPU] Add gpu.set_default_device op

[Krzysztof Drewniak via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 84718d37db577f57514df6ac544e3db88aa75684
      https://github.com/llvm/llvm-project/commit/84718d37db577f57514df6ac544e3db88aa75684
  Author: Krzysztof Drewniak <Krzysztof.Drewniak at amd.com>
  Date:   2022-02-17 (Thu, 17 Feb 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/GPU/GPUOps.td
    M mlir/lib/Conversion/GPUCommon/GPUToLLVMConversion.cpp
    M mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
    M mlir/lib/ExecutionEngine/RocmRuntimeWrappers.cpp
    M mlir/test/Dialect/GPU/ops.mlir

  Log Message:
  -----------
  [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