assert instead of raise

leofang · leofang · commit 73c8b3cac309 · 2019-11-26T00:34:48.000-05:00
diff --git a/cupy/cuda/cub.pyx b/cupy/cuda/cub.pyx
@@ -74,8 +74,7 @@ cpdef _preprocess_array(ndarray arr, axis, bint keepdims, str order):
         axis_permutes = out_axis + reduce_axis
     elif order == 'F':
         axis_permutes = reduce_axis + out_axis
-    if axis_permutes != tuple(range(len(arr.shape))):
-        raise ValueError('should not happen')
+    assert axis_permutes == tuple(range(len(arr.shape)))
 
     for axis in reduce_axis:
         contiguous_size *= arr.shape[axis]
