Add XLA ops coverage to docs publication.
The table will be updated automatically, so also removes the manually table.
PiperOrigin-RevId: 306991242
diff --git a/build_tools/cmake/build_docs.sh b/build_tools/cmake/build_docs.sh
index 1eba219..77195b5 100755
--- a/build_tools/cmake/build_docs.sh
+++ b/build_tools/cmake/build_docs.sh
@@ -49,6 +49,9 @@
cd ${ROOT_DIR?}
+# Update op_coverage.md
+scripts/update_op_coverage.py ${BUILD_DIR}
+
# Copy a curated list of docs to publish. This is expected to cover all docs
# under docs/ after they are refreshed.
diff --git a/docs/op_coverage.md b/docs/op_coverage.md
deleted file mode 100644
index 3078511..0000000
--- a/docs/op_coverage.md
+++ /dev/null
@@ -1,60 +0,0 @@
-# Op Coverage
-
-There are four backend
-[targets](https://github.com/google/iree/tree/master/iree/compiler/Dialect/HAL/Target)
-in IREE:
-
-- vmla
-- llvm-ir
-- vulkan (direct path)
-- vulkan (structured ops path)
-
-IREE has two path for Vulkan backend, one is using direct path (XLA -> SPIR-V),
-and another is using structured ops path (XLA -> Linalg -> ... -> SPIR-V).
-
-The table shows the supported XLA HLO ops on each backend.
-
-| op | vmla | vulkan (direct | vulkan (structured | llvm-ir |
-: : : path) : ops path) : :
-| :------------------: | :--: | :------------: | :----------------: | :-----: |
-| abs | ✓ | ✓ | ✓ | ✓ |
-| add | ✓ | ✓ | ✓ | ✓ |
-| batch_norm_inference | ✓ | ✗ | ✓ | ✓ |
-| broadcast | ✓ | ✓ | ✗ | ✗ |
-| broadcast_in_dim | ✓ | ✓ | ✓ | ✓ |
-| compare | ✓ | ✓ | ✗ | ✓ |
-| concatenate | ✓ | ✓ | ✗ | ✗ |
-| constant | ✓ | ✓ | ✓ | ✓ |
-| conv | ✓ | ✗ | ✗ | ✓ |
-| convert_int | ✓ | ✓ | ✗ | ✗ |
-| cos | ✓ | ✓ | ✓ | ✓ |
-| dot | ✓ | ✓ | ✗ | ✓ |
-| dot_general | ✓ | ✗ | ✗ | ✗ |
-| exp | ✓ | ✓ | ✓ | ✓ |
-| floor | ✓ | ✓ | ✗ | ✗ |
-| gather | ✓ | ✓ | ✗ | ✗ |
-| gather_concat | ✓ | ✓ | ✗ | ✗ |
-| gemm | ✓ | ✓ | ✓ | ✓ |
-| gemm_large | ✓ | ✓ | ✓ | ✓ |
-| log | ✓ | ✓ | ✓ | ✓ |
-| max_float | ✓ | ✓ | ✓ | ✓ |
-| max_int | ✓ | ✓ | ✓ | ✓ |
-| min_float | ✓ | ✓ | ✓ | ✓ |
-| min_int | ✓ | ✓ | ✓ | ✓ |
-| multiply | ✓ | ✓ | ✗ | ✓ |
-| negate | ✓ | ✗ | ✗ | ✓ |
-| pad | ✓ | ✓ | ✗ | ✗ |
-| reduce_float | ✓ | ✓ | ✗ | ✓ |
-| reduce_int | ✓ | ✓ | ✗ | ✓ |
-| reduce_window | ✓ | ✗ | ✗ | ✓ |
-| rem | ✓ | ✓ | ✓ | ✓ |
-| reshape | ✓ | ✓ | ✗ | ✓ |
-| reshape_adddims | ✓ | ✓ | ✓ | ✓ |
-| reshape_dropdims | ✓ | ✓ | ✓ | ✓ |
-| reverse | ✓ | ✓ | ✗ | ✗ |
-| rsqrt | ✓ | ✓ | ✓ | ✓ |
-| select | ✓ | ✓ | ✓ | ✓ |
-| sin | ✓ | ✓ | ✗ | ✗ |
-| slice | ✓ | ✓ | ✗ | ✗ |
-| sqrt | ✓ | ✓ | ✓ | ✓ |
-| while | ✓ | ✓ | ✗ | ✓ |
diff --git a/scripts/prepare_doc_publication.py b/scripts/prepare_doc_publication.py
index b17cb6f..fc10aa5 100755
--- a/scripts/prepare_doc_publication.py
+++ b/scripts/prepare_doc_publication.py
@@ -54,6 +54,7 @@
'getting_started_windows_bazel.md': 'Windows with Bazel',
'getting_started_windows_cmake.md': 'Windows with CMake',
'getting_started_windows_vulkan.md': 'Windows with Vulkan',
+ 'op_coverage.md': 'XLA HLO Operation Coverage',
'roadmap.md': 'Short-term Focus Areas',
'roadmap_design.md': 'Long-term Design Roadmap',
}
@@ -73,6 +74,7 @@
'getting_started_windows_cmake.md': 'GetStarted/WindowsCMake',
'getting_started_windows_vulkan.md': 'GetStarted/WindowsVulkan',
'developer_overview.md': 'DeveloperOverview',
+ 'op_coverage.md': 'HLOOpCoverage',
'roadmap.md': 'FocusAreas',
'roadmap_design.md': 'DesignRoadmap',
}
@@ -88,6 +90,7 @@
'developer_overview.md': 3,
'roadmap_design.md': 4,
'roadmap.md': 5,
+ 'op_coverage.md': 6,
}
# A dictionary containing source directory to section tile mappings.
diff --git a/scripts/update_op_coverage.py b/scripts/update_op_coverage.py
index 5789f41..8e3f78b 100755
--- a/scripts/update_op_coverage.py
+++ b/scripts/update_op_coverage.py
@@ -29,7 +29,7 @@
E2E_XLA_OPS_PATH = 'iree/test/e2e/xla_ops'
-OP_COVERAGE_DESCRIPTION = """# Op Coverage
+OP_COVERAGE_DESCRIPTION = """# HLO Op Coverage
There are four backend [targets](https://github.com/google/iree/tree/master/iree/compiler/Dialect/HAL/Target) in IREE:
- vmla
