Skip to content

Commit 356b6ee

Browse files
author
Vahid Tavanashad
committed
add assert distinct_core != 1
1 parent d0168a9 commit 356b6ee

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

dpnp/dpnp_utils/dpnp_utils_linearalgebra.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -853,6 +853,7 @@ def dpnp_multiplication(
853853
# "matmul," "matvec," and "vecmat" always have multiple distinct cores,
854854
# and `axis` is not supported for these functions.
855855
# Therefore, raise an error in all cases where `axis` is provided.
856+
assert distinct_core != 1
856857
raise TypeError(
857858
f"{func}: axis can only be used with a single shared core "
858859
f"dimension, not with the {distinct_core} distinct ones implied "

0 commit comments

Comments
 (0)