mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-07 02:04:26 +00:00
Set up basic tests with gtest
This commit is contained in:
13
test/test_utils/test_cublas_fixture.cpp
Normal file
13
test/test_utils/test_cublas_fixture.cpp
Normal file
@@ -0,0 +1,13 @@
|
||||
#include "gtest/gtest.h"
|
||||
#include "cublas_v2.h"
|
||||
#include "test_cublas_fixture.h"
|
||||
|
||||
cublasHandle_t CublasTestFixture::cublasHandle;
|
||||
|
||||
void CublasTestFixture::SetUpTestSuite() {
|
||||
cublasCreate(&cublasHandle);
|
||||
}
|
||||
|
||||
void CublasTestFixture::TearDownTestSuite() {
|
||||
cublasDestroy(cublasHandle);
|
||||
}
|
||||
10
test/test_utils/test_cublas_fixture.h
Normal file
10
test/test_utils/test_cublas_fixture.h
Normal file
@@ -0,0 +1,10 @@
|
||||
#include "gtest/gtest.h"
|
||||
#include "cublas_v2.h"
|
||||
|
||||
class CublasTestFixture : public ::testing::Test {
|
||||
protected:
|
||||
static cublasHandle_t cublasHandle;
|
||||
|
||||
static void SetUpTestSuite();
|
||||
static void TearDownTestSuite();
|
||||
};
|
||||
Reference in New Issue
Block a user