only do test_ConvUnify when there is a gpu

avx_docs
wangyang59 9 years ago
parent 60bf1a476a
commit ba786b2e9e

@ -94,6 +94,7 @@ MatrixPtr doOneConvTest(size_t imgSize, size_t output_x, size_t stride,
}
TEST(Layer, convParaUnified) {
#ifndef PADDLE_ONLY_CPU
MatrixPtr input, resultCpu, resultGpu;
input = Matrix::create(1, 4 * 4, false, false);
float inputData[] = {1, 2, 3, 4,
@ -155,6 +156,7 @@ TEST(Layer, convParaUnified) {
/*numfilters*/ 2,
input, param2, true);
checkMatrixEqual(resultCpu, resultGpu);
#endif
}
int main(int argc, char** argv) {

Loading…
Cancel
Save