Fix stride2x2 validity check for out channel to check stride evenness

pulp-platform · Jan 27, 2024 · c436ea4 · c436ea4
1 parent a6f142a
commit c436ea4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/Ne16TestConf.py b/test/Ne16TestConf.py
@@ -85,9 +85,10 @@ def check_valid_bias_type(cls, v: Optional[IntegerType]) -> Optional[IntegerType
         return v
 
     @model_validator(mode="after")  # type: ignore
-    def check_valid_out_channel_with_stride_2x2(self) -> Ne16TestConf:
+    def check_valid_out_channel_stride_with_stride_2x2(self) -> Ne16TestConf:
         assert implies(
-            self.stride == Stride(height=2, width=2), self.out_channel % 2 == 0
+            self.stride == Stride(height=2, width=2),
+            self.out_channel * (self.out_type._bits // 8) % 2 == 0,
         ), f"With stride 2x2 supported only even output channel sizes. Given output channel {self.out_channel}"
         return self