Merge pull request #4917 from kjetilly/remove_cuda_thread_exit

Removed call to deprecated cudaThreadExit
This commit is contained in:
Bård Skaflestad 2023-10-09 11:41:23 +02:00 committed by GitHub
commit ab6241f1f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -42,7 +42,6 @@ setDevice(int mpiRank, [[maybe_unused]] int numberOfMpiRanks)
// TODO: We need to be more sophistacted here. We have no guarantee this will pick the correct device.
const auto deviceId = mpiRank % deviceCount;
OPM_CUDA_SAFE_CALL(cudaDeviceReset());
OPM_CUDA_SAFE_CALL(cudaThreadExit());
OPM_CUDA_SAFE_CALL(cudaSetDevice(deviceId));
OpmLog::info("Set CUDA device to " + std::to_string(deviceId) + " (out of " + std::to_string(deviceCount)
+ " devices).");