From f793e4f72aacf15609035afd1c8350e61b982484 Mon Sep 17 00:00:00 2001 From: peixuanzuo Date: Fri, 29 Dec 2023 04:26:52 +0000 Subject: [PATCH] update python format --- .../kernel_explorer/kernels/gemm_softmax_gemm_permute_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/python/tools/kernel_explorer/kernels/gemm_softmax_gemm_permute_test.py b/onnxruntime/python/tools/kernel_explorer/kernels/gemm_softmax_gemm_permute_test.py index 2b506d95adc86..d3b1e820269b5 100644 --- a/onnxruntime/python/tools/kernel_explorer/kernels/gemm_softmax_gemm_permute_test.py +++ b/onnxruntime/python/tools/kernel_explorer/kernels/gemm_softmax_gemm_permute_test.py @@ -126,7 +126,7 @@ def _make_causal_mask( correct_mask = np.full((seqence_length, total_sequence_length), 1) for i in range(seqence_length): - correct_mask[i][:] = not (sum(mask[i]) == total_sequence_length) + correct_mask[i][:] = sum(mask[i]) != total_sequence_length return mask, correct_mask