Update SupportedONNXOps-NNPA.md to include the correct command (#2549)

* Update SupportedONNXOps-NNPA.md --------- Co-authored-by: Megan Hampton <[email protected]>
onnx · Oct 4, 2023 · 49a98ec · 49a98ec
1 parent 31f6f59
commit 49a98ec
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 5 deletions.
diff --git 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
@@ -1,5 +1,5 @@
 <!--- Automatically generated, do not edit. -->
-<!--- To update, run `make onnx_mlir_supported_ops` -->
+<!--- To update, run `make onnx_mlir_supported_ops_NNPA' -->
 
 # Supported ONNX Operation for Target *NNPA*.
 

diff --git a/docs/SupportedONNXOps-cpu.md b/docs/SupportedONNXOps-cpu.md
@@ -1,5 +1,5 @@
 <!--- Automatically generated, do not edit. -->
-<!--- To update, run `make onnx_mlir_supported_ops` -->
+<!--- To update, run `make onnx_mlir_supported_ops_cpu' -->
 
 # Supported ONNX Operation for Target *cpu*.
 

diff --git 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.<br>- `R` must have static dimensions.<br>- If `B` and `initial_h` are given, they must have static dimensions.<br>- `sequence_lens` is not supported.<br>- `activations` must be `["Sigmoid", "Tanh", "Tanh"]`.<br>- `clip` is not supported.<br>- `linear_before_reset` must be 1.<br>- `layout` is not supported.
+    # ==LIM== - `direction` and `hidden_size` in `W` must have static dimensions.<br>- `R` must have static dimensions.<br>- If `B` and `initial_h` are given, they must have static dimensions.<br>- `sequence_lens` is not supported for bidirectional GRU.<br>- `activations` must be `["Sigmoid", "Tanh", "Tanh"]`.<br>- `clip` is not supported.<br>- `linear_before_reset` must be 1.<br>- `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
@@ -205,7 +205,7 @@ def print_row(array):
 def print_md():
     # Header.
     print("<!--- Automatically generated, do not edit. -->")
-    print("<!--- To update, run `make onnx_mlir_supported_ops` -->")
+    print("<!--- To update, run `make onnx_mlir_supported_ops_" + target_arch + "' -->")
     # Title
     print("\n# Supported ONNX Operation for Target *" + target_arch + "*.\n")
     # Top paragraph.