[all-commits] [llvm/llvm-project] 8262cd: [mlir] Set CUDA/ROCm context before creating resou...

[Christian Sigg via All-commits]

  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: 8262cd8a0e24fd5d3d528d675e106031d4232d8d
      https://github.com/llvm/llvm-project/commit/8262cd8a0e24fd5d3d528d675e106031d4232d8d
  Author: Christian Sigg <csigg at google.com>
  Date:   2021-01-26 (Tue, 26 Jan 2021)

  Changed paths:
    M mlir/tools/mlir-cuda-runner/cuda-runtime-wrappers.cpp
    M mlir/tools/mlir-rocm-runner/rocm-runtime-wrappers.cpp

  Log Message:
  -----------
  [mlir] Set CUDA/ROCm context before creating resources.

The current context is thread-local state, and in preparation of GPU async execution (on multiple threads) we need to set the context before calling API that create resources.

Reviewed By: herhut

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