reduce iteration

guangzhixie · Jun 27, 2018 · 635ff42 · 635ff42
1 parent 7c0a0f0
commit 635ff42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/saber/bm/test_saber_func_softmax_BM.cpp b/test/saber/bm/test_saber_func_softmax_BM.cpp
@@ -16,7 +16,7 @@ TEST(TestSaberFuncSoftmaxBM, test_func_softmax_BM) {
 
     typedef TensorDf4::Dtype dtype;
 
-    int test_iter = 1000;
+    int test_iter = 10;
 
     int softmax_axis = 3; // channel
     int w_in = 3;