commit 9f2cae1e7bac26e3258f6437f58496c8f7e08e20
author Cindy Liu <hcindyl@google.com> Thu Jan 12 10:58:08 2023 -0800
committer Cindy Liu <hcindyl@google.com> Thu Jan 12 11:12:35 2023 -0800
tree b4e436ea22e57f88e2835587f7015269e88bab86
parent d999794f8dd47bdbef60f97e6de8ac55970fb1a7

sw:vec_iree: Add ukernel tiling query support

Match API changes in https://github.com/iree-org/iree/pull/11640

Fix: 265310678
Change-Id: I1a1e675bac13aadea155eb933fb2d395d6d69493

vmvx_ukernel/CMakeLists.txt

diff --git a/vmvx_ukernel/CMakeLists.txt b/vmvx_ukernel/CMakeLists.txt
index a18e016..537e9e7 100644
--- a/vmvx_ukernel/CMakeLists.txt
+++ b/vmvx_ukernel/CMakeLists.txt
@@ -6,9 +6,11 @@
   HDRS
     "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/arch/mmt4d_arch.h"
     "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/arch/pack_arch.h"
+    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/arch/query_tile_sizes_arch.h"
   SRCS
     "mmt4d_arch.c"
     "pack_arch.c"
+    "query_tile_sizes_arch.c"
   DEPS
     iree::builtins::ukernel::common
     iree::builtins::ukernel::generic
@@ -22,10 +24,12 @@
     "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/elementwise.h"
     "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/mmt4d.h"
     "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/pack.h"
+    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/query_tile_sizes.h"
     "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/unpack.h"
   SRCS
     "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/mmt4d.c"
     "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/pack.c"
+    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/query_tile_sizes.c"
     "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/unpack.c"
   DEPS
     ::arch

vmvx_ukernel/query_tile_sizes_arch.c

diff --git a/vmvx_ukernel/query_tile_sizes_arch.c b/vmvx_ukernel/query_tile_sizes_arch.c
new file mode 100644
index 0000000..90b3502
--- /dev/null
+++ b/vmvx_ukernel/query_tile_sizes_arch.c
@@ -0,0 +1,15 @@
+// Copyright 2023 Google LLC.
+// Copyright 2022 The IREE Authors
+//
+// Licensed under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+#include "iree/builtins/ukernel/arch/query_tile_sizes_arch.h"
+
+bool iree_uk_query_matmul_tile_sizes_arch(
+    const iree_uk_query_tile_sizes_2d_params_t* params,
+    iree_uk_matmul_tile_sizes_t* out_matmul_tile_sizes) {
+  // TODO(lundong): to be replaced with Kelvin logic.
+  return false;
+}