[Mlir-commits] [mlir] c5803ee - [MLIR][GPU] Remove call to cudaSetDevice(), which no longer exists

[Krzysztof Drewniak]

Author: Krzysztof Drewniak
Date: 2022-02-17T21:38:05Z
New Revision: c5803ee4faa7305f7dcb58170acdb3247f405e04

URL: https://github.com/llvm/llvm-project/commit/c5803ee4faa7305f7dcb58170acdb3247f405e04
DIFF: https://github.com/llvm/llvm-project/commit/c5803ee4faa7305f7dcb58170acdb3247f405e04.diff

LOG: [MLIR][GPU] Remove call to cudaSetDevice(), which no longer exists

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

Added: 
    

Modified: 
    mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp b/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
index dd66056289cec..44ed5b0cd2057 100644
--- a/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
+++ b/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
@@ -194,5 +194,4 @@ mgpuMemHostRegisterMemRef(int64_t rank, StridedMemRefType<char, 1> *descriptor,
 
 extern "C" MLIR_CUDA_WRAPPERS_EXPORT void mgpuSetDefaultDevice(int32_t device) {
   defaultDevice = device;
-  CUDA_REPORT_IF_ERROR(cudaSetDevice(device));
 }