Add unitest for factorization machine layer with sparse input

8 years ago · 09f4f92579
parent 0b6afb589c
commit 09f4f92579
1 changed files with 1 additions and 1 deletions
--- a/paddle/gserver/tests/test_LayerGrad.cpp
+++ b/paddle/gserver/tests/test_LayerGrad.cpp
@ -2444,8 +2444,8 @@ void testFactorizationMachineLayer(InputType type, bool useGpu) {
 TEST(Layer, FactorizationMachineLayer) {
  for (auto useGpu : {false, true}) {
    testFactorizationMachineLayer(INPUT_DATA, useGpu);
-    testFactorizationMachineLayer(INPUT_SPARSE_FLOAT_VALUE_DATA, useGpu);
  }
+  testFactorizationMachineLayer(INPUT_SPARSE_FLOAT_VALUE_DATA, false);
 }

 int main(int argc, char** argv) {