@@ -42,8 +42,10 @@ class TestSoftmaxCrossEntropy : public ::testing::Test {
42
42
43
43
TEST_F (TestSoftmaxCrossEntropy, CppForward) {
44
44
p.CopyDataFromHostPtr (pdat, 8 );
45
- t. AsType (singa:: kInt );
45
+ EXPECT_TRUE (p. block ()-> initialized () );
46
46
t.CopyDataFromHostPtr (tdat, 2 );
47
+ t.AsType (singa::kInt );
48
+
47
49
48
50
singa::SoftmaxCrossEntropy cross_entropy;
49
51
const Tensor& loss = cross_entropy.Forward (singa::kEval , p, t);
@@ -56,8 +58,8 @@ TEST_F(TestSoftmaxCrossEntropy, CppForward) {
56
58
57
59
TEST_F (TestSoftmaxCrossEntropy, CppForwardAryTarget) {
58
60
p.CopyDataFromHostPtr (pdat, 8 );
59
- ta.AsType (singa::kInt );
60
61
ta.CopyDataFromHostPtr (tary, 8 );
62
+ ta.AsType (singa::kInt );
61
63
62
64
singa::SoftmaxCrossEntropy cross_entropy;
63
65
const Tensor& loss = cross_entropy.Forward (singa::kEval , p, ta);
@@ -70,8 +72,8 @@ TEST_F(TestSoftmaxCrossEntropy, CppForwardAryTarget) {
70
72
71
73
TEST_F (TestSoftmaxCrossEntropy, CppBackward) {
72
74
p.CopyDataFromHostPtr (pdat, 8 );
73
- t.AsType (singa::kInt );
74
75
t.CopyDataFromHostPtr (tdat, 2 );
76
+ t.AsType (singa::kInt );
75
77
76
78
singa::SoftmaxCrossEntropy cross_entropy;
77
79
cross_entropy.Forward (singa::kTrain , p, t);
@@ -90,8 +92,8 @@ TEST_F(TestSoftmaxCrossEntropy, CppBackward) {
90
92
91
93
TEST_F (TestSoftmaxCrossEntropy, CppBackwardAryTarget) {
92
94
p.CopyDataFromHostPtr (pdat, 8 );
93
- ta.AsType (singa::kInt );
94
95
ta.CopyDataFromHostPtr (tary, 8 );
96
+ ta.AsType (singa::kInt );
95
97
96
98
singa::SoftmaxCrossEntropy cross_entropy;
97
99
cross_entropy.Forward (singa::kTrain , p, ta);
0 commit comments