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.