HomePhabricator

[mlir] Set CUDA/ROCm context before creating resources.

Authored by csigg on Jan 26 2021, 5:24 AM.

Description

[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

Details

Committed
csiggJan 26 2021, 10:07 AM
Reviewer
herhut
Differential Revision
D94495: [mlir] Set CUDA/ROCm context before creating resources.
Parents
rG5f9707b7960e: AMDGPU: Fix redundant FP spilling/assert in some functions
Branches
Unknown
Tags
Unknown