From 3d85224be96844dd44217a52d2a1efd2a77872c2 Mon Sep 17 00:00:00 2001 From: "Tung D. Le" Date: Tue, 15 Aug 2023 16:39:50 +0900 Subject: [PATCH] Donot check sequence_lens in lowering ONNX to ZHigh Signed-off-by: Tung D. Le --- .../NNPA/Conversion/ONNXToZHigh/ONNXLegalityCheck.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/Accelerators/NNPA/Conversion/ONNXToZHigh/ONNXLegalityCheck.cpp b/src/Accelerators/NNPA/Conversion/ONNXToZHigh/ONNXLegalityCheck.cpp index 69ad591ad2..dbe088b600 100644 --- a/src/Accelerators/NNPA/Conversion/ONNXToZHigh/ONNXLegalityCheck.cpp +++ b/src/Accelerators/NNPA/Conversion/ONNXToZHigh/ONNXLegalityCheck.cpp @@ -661,8 +661,8 @@ bool isSuitableForZDNN( if (hidden_size > MAXIMUM_NUM_HIDDEN_SIZE_GRU) return false; // zDNN does not support sequence_lens. - if (!isNoneValue(op.getSequenceLens())) - return false; + // if (!isNoneValue(op.getSequenceLens())) + // return false; // check if B and initial_h have static dimensions if given. if (!isNoneValue(B) && !B.getType().cast().hasStaticShape()) return false;