Skip to content

Commit 09f4f92

Browse files
committed
Add unitest for factorization machine layer with sparse input
1 parent 0b6afb5 commit 09f4f92

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/gserver/tests/test_LayerGrad.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2444,8 +2444,8 @@ void testFactorizationMachineLayer(InputType type, bool useGpu) {
24442444
TEST(Layer, FactorizationMachineLayer) {
24452445
for (auto useGpu : {false, true}) {
24462446
testFactorizationMachineLayer(INPUT_DATA, useGpu);
2447-
testFactorizationMachineLayer(INPUT_SPARSE_FLOAT_VALUE_DATA, useGpu);
24482447
}
2448+
testFactorizationMachineLayer(INPUT_SPARSE_FLOAT_VALUE_DATA, false);
24492449
}
24502450

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

0 commit comments

Comments
 (0)