Changes to accommodate upstream IREE ukernel changes
Most changes are simply renaming.
Change-Id: I6627dfda5983278d66139951b6493c8fdf4e27ad
diff --git a/vmvx_ukernel/CMakeLists.txt b/vmvx_ukernel/CMakeLists.txt
index 67e0e6c..4272c4b 100644
--- a/vmvx_ukernel/CMakeLists.txt
+++ b/vmvx_ukernel/CMakeLists.txt
@@ -5,9 +5,6 @@
ukernel
HDRS
"${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/api.h"
- "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/mmt4d_tile.h"
- "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/pack_tile.h"
- "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/unpack_tile.h"
SRCS
"elementwise.c"
"elementwise_impl.c.inc"
@@ -19,7 +16,7 @@
"${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/unpack.c"
"${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/unpack_tile.c"
DEPS
- iree::builtins::ukernel::headers
+ iree::builtins::ukernel::internal_headers
PUBLIC
)
diff --git a/vmvx_ukernel/mmt4d_tile.c b/vmvx_ukernel/mmt4d_tile.c
index a591c66..05730fb 100644
--- a/vmvx_ukernel/mmt4d_tile.c
+++ b/vmvx_ukernel/mmt4d_tile.c
@@ -14,11 +14,11 @@
* limitations under the License.
*/
-#include "iree/builtins/ukernel/mmt4d_tile.h"
-
#include <riscv_vector.h>
#include <string.h>
+#include "iree/builtins/ukernel/mmt4d_internal.h"
+
// Calculate the dot product of two int8 vectors using RVV
static iree_uk_int32_t dot_product_rvv(const iree_uk_int8_t* u,
const iree_uk_int8_t* w, int n) {
diff --git a/vmvx_ukernel/query_tile_sizes.c b/vmvx_ukernel/query_tile_sizes.c
index 4bab96e..e3c9222 100644
--- a/vmvx_ukernel/query_tile_sizes.c
+++ b/vmvx_ukernel/query_tile_sizes.c
@@ -14,7 +14,7 @@
* limitations under the License.
*/
-#include "iree/builtins/ukernel/query_tile_sizes.h"
+#include "iree/builtins/ukernel/query_tile_sizes_internal.h"
static bool iree_uk_query_tile_sizes_operation_is_matmul(
iree_uk_uint32_t flags) {
