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};