OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2025-02-25 18:55:30 -06:00
Code Issues Packages Projects Releases Wiki Activity

Added safe call macros for CUDA, CuBlas and CuSparse.

Browse Source
This commit is contained in:
Kjetil Olsen Lye
2023-03-24 16:01:48 +01:00
parent 1a576f9007
commit bf9dd4e1dd
7 changed files with 381 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
tests/cuistl/test_cublas_safe_call.cpp Normal file
View File

@@ -0,0 +1,47 @@
/*
Copyright 2022-2023 SINTEF AS
This file is part of the Open Porous Media project (OPM).
OPM is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
OPM is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with OPM. If not, see <http://www.gnu.org/licenses/>.
*/
#include <config.h>
#define BOOST_TEST_MODULE TestCublasSafeCall
#include <boost/test/unit_test.hpp>
#include <cublas_v2.h>
#include <opm/simulators/linalg/cuistl/detail/cublas_safe_call.hpp>
BOOST_AUTO_TEST_CASE(TestCreateHandle)
{
cublasHandle_t cublasHandle;
BOOST_CHECK_NO_THROW(OPM_CUBLAS_SAFE_CALL(cublasCreate(&cublasHandle)););
}
BOOST_AUTO_TEST_CASE(TestThrows)
{
std::vector<cublasStatus_t> errorCodes {{CUBLAS_STATUS_NOT_INITIALIZED,
CUBLAS_STATUS_ALLOC_FAILED,
CUBLAS_STATUS_INVALID_VALUE,
CUBLAS_STATUS_ARCH_MISMATCH,
CUBLAS_STATUS_MAPPING_ERROR,
CUBLAS_STATUS_EXECUTION_FAILED,
CUBLAS_STATUS_INTERNAL_ERROR,
CUBLAS_STATUS_NOT_SUPPORTED,
CUBLAS_STATUS_LICENSE_ERROR}};
for (auto code : errorCodes) {
BOOST_CHECK_THROW(OPM_CUBLAS_SAFE_CALL(code), std::exception);
}
}

40
tests/cuistl/test_cuda_safe_call.cpp Normal file
View File

@@ -0,0 +1,40 @@
/*
Copyright 2022-2023 SINTEF AS
This file is part of the Open Porous Media project (OPM).
OPM is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
OPM is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with OPM. If not, see <http://www.gnu.org/licenses/>.
*/
#include <config.h>
#define BOOST_TEST_MODULE TestCudaSafeCall
#include <boost/test/unit_test.hpp>
#include <cuda_runtime.h>
#include <opm/simulators/linalg/cuistl/detail/cuda_safe_call.hpp>
BOOST_AUTO_TEST_CASE(TestCudaMalloc)
{
void* pointer;
BOOST_CHECK_NO_THROW(OPM_CUDA_SAFE_CALL(cudaMalloc(&pointer, 1)););
}
BOOST_AUTO_TEST_CASE(TestThrows)
{
// Just testing a subset here.
std::vector<cudaError_t> errorCodes {{cudaErrorAddressOfConstant, cudaErrorAlreadyAcquired}};
for (auto code : errorCodes) {
BOOST_CHECK_THROW(OPM_CUDA_SAFE_CALL(code), std::exception);
}
}

49
tests/cuistl/test_cusparse_safe_call.cpp Normal file
View File

@@ -0,0 +1,49 @@
/*
Copyright 2022-2023 SINTEF AS
This file is part of the Open Porous Media project (OPM).
OPM is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
OPM is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with OPM. If not, see <http://www.gnu.org/licenses/>.
*/
#include <config.h>
#define BOOST_TEST_MODULE TestCusparseSafeCall
#include <boost/test/unit_test.hpp>
#include <cusparse.h>
#include <opm/simulators/linalg/cuistl/detail/cusparse_safe_call.hpp>
BOOST_AUTO_TEST_CASE(TestCreateHandle)
{
cusparseHandle_t cusparseHandle;
BOOST_CHECK_NO_THROW(OPM_CUSPARSE_SAFE_CALL(cusparseCreate(&cusparseHandle)););
}
BOOST_AUTO_TEST_CASE(TestThrows)
{
std::vector<cusparseStatus_t> errorCodes {{CUSPARSE_STATUS_NOT_INITIALIZED,
CUSPARSE_STATUS_ALLOC_FAILED,
CUSPARSE_STATUS_INVALID_VALUE,
CUSPARSE_STATUS_ARCH_MISMATCH,
CUSPARSE_STATUS_MAPPING_ERROR,
CUSPARSE_STATUS_EXECUTION_FAILED,
CUSPARSE_STATUS_INTERNAL_ERROR,
CUSPARSE_STATUS_MATRIX_TYPE_NOT_SUPPORTED,
CUSPARSE_STATUS_ZERO_PIVOT,
CUSPARSE_STATUS_NOT_SUPPORTED,
CUSPARSE_STATUS_INSUFFICIENT_RESOURCES}};
for (auto code : errorCodes) {
BOOST_CHECK_THROW(OPM_CUSPARSE_SAFE_CALL(code), std::exception);
}
}