Skip to content

Commit d9e8dae

Browse files
author
Vahid Tavanashad
committed
update a TODO
1 parent f2f5f13 commit d9e8dae

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dpnp/dpnp_utils/dpnp_utils_linearalgebra.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -509,8 +509,8 @@ def _gemm_special_case(x1, x2, res_dtype, call_flag):
509509
while `gemv` does not.
510510
511511
"""
512-
# TODO: replace with dpnp.int8 when it is added
513-
is_int8 = x1.dtype == numpy.int8 and x2.dtype == numpy.int8
512+
513+
is_int8 = x1.dtype == dpnp.int8 and x2.dtype == dpnp.int8
514514
is_int32_or_f32 = res_dtype in [dpnp.int32, dpnp.float32]
515515
flag = is_int8 and is_int32_or_f32 and call_flag in ["gemm", "gemm_batch"]
516516

0 commit comments

Comments
 (0)