fix depthconv test

dw_conv can also use workspace that results in a non-zero workspace size. changing it to ASSERT_NE PiperOrigin-RevId: 593797013
phoebesv · Dec 26, 2023 · 9547e12 · 9547e12
1 parent 4178a8e
commit 9547e12
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/test/depthwise-convolution-2d.cc b/test/depthwise-convolution-2d.cc
@@ -906,14 +906,14 @@ TEST_F(DepthwiseConvolutionTestF32, matches_operator_api)
   ASSERT_NE(nullptr, op);
   size_t workspace_size = SIZE_MAX;
   size_t workspace_alignment = SIZE_MAX;
-  ASSERT_EQ(
-    xnn_status_success, xnn_reshape_convolution2d_nhwc_f32(
-                          op, batch_size, input_height, input_width,
-                           &workspace_size, &workspace_alignment,
-                          /*output_height_out=*/nullptr, /*output_width_out=*/nullptr,
-                          /*threadpool=*/nullptr));
-  ASSERT_EQ(workspace_size, 0);
-  ASSERT_EQ(workspace_alignment, 1);
+  ASSERT_EQ(xnn_status_success,
+            xnn_reshape_convolution2d_nhwc_f32(
+                op, batch_size, input_height, input_width, &workspace_size,
+                &workspace_alignment,
+                /*output_height_out=*/nullptr, /*output_width_out=*/nullptr,
+                /*threadpool=*/nullptr));
+  ASSERT_NE(workspace_size, SIZE_MAX);
+  ASSERT_NE(workspace_alignment, SIZE_MAX);
   ASSERT_EQ(xnn_status_success, xnn_setup_convolution2d_nhwc_f32(op, /*workspace=*/nullptr, input.data(), operator_output.data()));
 
   ASSERT_EQ(xnn_status_success, xnn_run_operator(op, /*threadpool=*/nullptr));