FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
CaffeOnACL/unit_tests/test_common.cpp at master · tonylt/CaffeOnACL · GitHub
tonylt
CaffeOnACL
Repository navigation
Code
Pull requests
Actions
Projects
Wiki
Security and quality
Insights
Expand file tree
Breadcrumbs
CaffeOnACL
/
unit_tests
/
test_common.cpp
Copy path
More file actions
More file actions
Latest commit
History
History
History
64 lines (49 loc) · 1.77 KB
Breadcrumbs
CaffeOnACL
/
unit_tests
/
test_common.cpp
Copy path
File metadata and controls
64 lines (49 loc) · 1.77 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#
include
"
gtest/gtest.h
"
#
include
"
caffe/common.hpp
"
#
include
"
caffe/syncedmem.hpp
"
#
include
"
caffe/util/math_functions.hpp
"
#
include
"
caffe/test/test_caffe_main.hpp
"
namespace
caffe
{
class
CommonTest
: public ::testing::Test {};
#
ifndef
CPU_ONLY
//
GPU Caffe singleton test.
TEST_F
(CommonTest, TestCublasHandlerGPU) {
int
cuda_device_id;
CUDA_CHECK
(
cudaGetDevice
(&cuda_device_id));
EXPECT_TRUE
(
Caffe::cublas_handle
());
}
#
endif
TEST_F
(CommonTest, TestBrewMode) {
Caffe::set_mode
(Caffe::
CPU
);
EXPECT_EQ
(
Caffe::mode
(), Caffe::
CPU
);
Caffe::set_mode
(Caffe::
GPU
);
EXPECT_EQ
(
Caffe::mode
(), Caffe::
GPU
);
}
TEST_F
(CommonTest, TestRandSeedCPU) {
SyncedMemory
data_a
(
10
*
sizeof
(
int
));
SyncedMemory
data_b
(
10
*
sizeof
(
int
));
Caffe::set_random_seed
(
1701
);
caffe_rng_bernoulli
(
10
,
0.5
,
static_cast
<
int
*>(data_a.
mutable_cpu_data
()));
Caffe::set_random_seed
(
1701
);
caffe_rng_bernoulli
(
10
,
0.5
,
static_cast
<
int
*>(data_b.
mutable_cpu_data
()));
for
(
int
i =
0
; i <
10
; ++i) {
EXPECT_EQ
(
static_cast
<
const
int
*>(data_a.
cpu_data
())[i],
static_cast
<
const
int
*>(data_b.
cpu_data
())[i]);
}
}
#
ifndef
CPU_ONLY
//
GPU Caffe singleton test.
TEST_F
(CommonTest, TestRandSeedGPU) {
SyncedMemory
data_a
(
10
*
sizeof
(
unsigned
int
));
SyncedMemory
data_b
(
10
*
sizeof
(
unsigned
int
));
Caffe::set_random_seed
(
1701
);
CURAND_CHECK
(
curandGenerate
(
Caffe::curand_generator
(),
static_cast
<
unsigned
int
*>(data_a.
mutable_gpu_data
()),
10
));
Caffe::set_random_seed
(
1701
);
CURAND_CHECK
(
curandGenerate
(
Caffe::curand_generator
(),
static_cast
<
unsigned
int
*>(data_b.
mutable_gpu_data
()),
10
));
for
(
int
i =
0
; i <
10
; ++i) {
EXPECT_EQ
(((
const
unsigned
int
*)(data_a.
cpu_data
()))[i],
((
const
unsigned
int
*)(data_b.
cpu_data
()))[i]);
}
}
#
endif
}
//
namespace caffe
Back
|
FazBrowse Home
|
New Git URL