Fixing check module factory fn name.
diff --git a/iree/modules/check/check_test.cc b/iree/modules/check/check_test.cc
index 30c5e9c..daab092 100644
--- a/iree/modules/check/check_test.cc
+++ b/iree/modules/check/check_test.cc
@@ -49,7 +49,7 @@
iree_vm_instance_create(iree_allocator_system(), &instance_));
IREE_ASSERT_OK(
- check_native_module_create(iree_allocator_system(), &check_module_))
+ iree_check_module_create(iree_allocator_system(), &check_module_))
<< "Native module failed to init";
}
diff --git a/iree/modules/check/module.cc b/iree/modules/check/module.cc
index 2a14530..19348eb 100644
--- a/iree/modules/check/module.cc
+++ b/iree/modules/check/module.cc
@@ -4,6 +4,8 @@
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+#include "iree/modules/check/module.h"
+
#include <cassert>
#include <cmath>
#include <cstdint>
@@ -19,7 +21,6 @@
#include "iree/base/internal/math.h"
#include "iree/base/status.h"
#include "iree/hal/api.h"
-#include "iree/modules/check/module.h"
#include "iree/modules/hal/module.h"
#include "iree/testing/gtest.h"
#include "iree/vm/native_module_cc.h"
@@ -407,7 +408,7 @@
// Note that while we are using C++ bindings internally we still expose the
// module as a C instance. This hides the details of our implementation.
-extern "C" iree_status_t check_native_module_create(
+extern "C" iree_status_t iree_check_module_create(
iree_allocator_t allocator, iree_vm_module_t** out_module) {
IREE_ASSERT_ARGUMENT(out_module);
*out_module = NULL;
diff --git a/iree/modules/check/module.h b/iree/modules/check/module.h
index 5264b49..24d29ba 100644
--- a/iree/modules/check/module.h
+++ b/iree/modules/check/module.h
@@ -17,8 +17,8 @@
#endif // __cplusplus
// Creates a native custom module.
-iree_status_t check_native_module_create(iree_allocator_t allocator,
- iree_vm_module_t** out_module);
+iree_status_t iree_check_module_create(iree_allocator_t allocator,
+ iree_vm_module_t** out_module);
#ifdef __cplusplus
} // extern "C"
diff --git a/iree/tools/iree-check-module-main.cc b/iree/tools/iree-check-module-main.cc
index b2a65b5..9d58f56 100644
--- a/iree/tools/iree-check-module-main.cc
+++ b/iree/tools/iree-check-module-main.cc
@@ -110,7 +110,8 @@
IREE_RETURN_IF_ERROR(
iree_hal_module_create(device, iree_allocator_system(), &hal_module));
iree_vm_module_t* check_module = nullptr;
- check_native_module_create(iree_allocator_system(), &check_module);
+ IREE_RETURN_IF_ERROR(
+ iree_check_module_create(iree_allocator_system(), &check_module));
std::array<iree_vm_module_t*, 3> modules = {hal_module, check_module,
input_module};