diff --git a/MLIR.cmake b/MLIR.cmake
index d16b951f19..1a66f6e41e 100644
--- a/MLIR.cmake
+++ b/MLIR.cmake
@@ -83,9 +83,15 @@ add_custom_target(onnx-mlir-docs)
function(add_onnx_mlir_supported_ops input_file arch)
set(GEN_DOC_FILE ${ONNX_MLIR_SRC_ROOT}/docs/SupportedONNXOps-${arch}.md)
set(supported_ops_cmd ${Python3_EXECUTABLE} ${ONNX_MLIR_SRC_ROOT}/utils/documentOps.py -a ${arch} -i ${input_file} -p ${ONNX_MLIR_SRC_ROOT}/utils)
+ # For generating docs/SupportedONNXOps-NNPA.md we only want to use the "--notes" flag to bypass unnecessary content.
+ set(FLAG --notes --unsupported)
+ if (${arch} STREQUAL "NNPA")
+ unset(FLAG)
+ set(FLAG --notes)
+ endif()
add_custom_command(
OUTPUT ${GEN_DOC_FILE}
- COMMAND ${supported_ops_cmd} --notes --unsupported > ${GEN_DOC_FILE}
+ COMMAND ${supported_ops_cmd} ${FLAG} > ${GEN_DOC_FILE}
DEPENDS ${input_file})
add_custom_target(onnx_mlir_supported_ops_${arch} DEPENDS ${GEN_DOC_FILE})
add_dependencies(onnx_mlir_supported_ops onnx_mlir_supported_ops_${arch})
diff --git a/docs/SupportedONNXOps-NNPA.md b/docs/SupportedONNXOps-NNPA.md
index f73b650f04..ffe44411e2 100644
--- a/docs/SupportedONNXOps-NNPA.md
+++ b/docs/SupportedONNXOps-NNPA.md
@@ -1,5 +1,5 @@
-
+
# Supported ONNX Operation for Target *NNPA*.
diff --git a/docs/SupportedONNXOps-cpu.md b/docs/SupportedONNXOps-cpu.md
index 1200e38810..0a57f0e130 100644
--- a/docs/SupportedONNXOps-cpu.md
+++ b/docs/SupportedONNXOps-cpu.md
@@ -1,5 +1,5 @@
-
+
# Supported ONNX Operation for Target *cpu*.
diff --git a/test/accelerators/NNPA/backend/CMakeLists.txt b/test/accelerators/NNPA/backend/CMakeLists.txt
index 41f1fa6b65..bd79477bfa 100644
--- a/test/accelerators/NNPA/backend/CMakeLists.txt
+++ b/test/accelerators/NNPA/backend/CMakeLists.txt
@@ -187,7 +187,7 @@ set(NNPA_TEST_LIST
# ==OP== GRU
# ==MIN== 7
- # ==LIM== - `direction` and `hidden_size` in `W` must have static dimensions.
- `R` must have static dimensions.
- If `B` and `initial_h` are given, they must have static dimensions.
- `sequence_lens` is not supported.
- `activations` must be `["Sigmoid", "Tanh", "Tanh"]`.
- `clip` is not supported.
- `linear_before_reset` must be 1.
- `layout` is not supported.
+ # ==LIM== - `direction` and `hidden_size` in `W` must have static dimensions.
- `R` must have static dimensions.
- If `B` and `initial_h` are given, they must have static dimensions.
- `sequence_lens` is not supported for bidirectional GRU.
- `activations` must be `["Sigmoid", "Tanh", "Tanh"]`.
- `clip` is not supported.
- `linear_before_reset` must be 1.
- `layout` is not supported.
# test_gru_defaults_cpu
# test_gru_seq_length_cpu
# test_gru_with_initial_bias_cpu
diff --git a/utils/documentOps.py b/utils/documentOps.py
index 46b0b25992..5c38cc7c45 100755
--- a/utils/documentOps.py
+++ b/utils/documentOps.py
@@ -205,7 +205,7 @@ def print_row(array):
def print_md():
# Header.
print("")
- print("")
+ print("")
# Title
print("\n# Supported ONNX Operation for Target *" + target_arch + "*.\n")
# Top paragraph.