Check strides in spatialfilter.

Bad strides were generated by pythorch - issue kraiskil#41
GorrG · Mar 9, 2024 · 196f9f5 · 196f9f5
1 parent 1b557ba
commit 196f9f5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/nodes/spatialfilter.h b/src/nodes/spatialfilter.h
@@ -61,6 +61,11 @@ class SpatialFilter : public Node {
 		if( strides.size() == 0 )
 			for( unsigned i=0; i<get_numDataDim(); i++)
 				strides.push_back(1);
+		if( get_numDataDim() != strides.size() )
+			ERROR("Dimension of the stride do not match data dimensions");
+		for( uint64_t s : strides )
+			if( s == 0 )
+				ERROR("Stride of 0");
 	}
 
 	void resolve_kernel_shape(void)