1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
#pragma once
#include <unittest/testframework.h>
#include <thrust/system/cuda/memory.h>
#include <thrust/system_error.h>
#include <vector>
class CUDATestDriver
: public UnitTestDriver
{
public:
int current_device_architecture() const;
private:
std::vector<int> target_devices(const ArgumentMap &kwargs);
bool check_cuda_error(bool concise);
virtual bool post_test_smoke_check(const UnitTest &test, bool concise);
virtual bool run_tests(const ArgumentSet &args, const ArgumentMap &kwargs);
};
UnitTestDriver &driver_instance(thrust::system::cuda::tag);
|
