ml/ml-models-public: changes to accommodate IREE API for the 12052024 revision
Context: due to a long-standing bug on IREE importer/compiler (https://github.com/iree-org/iree/issues/19402) and a latest change on IREE compiler/runtime (https://github.com/iree-org/iree/commit/487c5217e4b7a931eb288b35b11c043c7567a7e3), we have to pin IREE to a 12/05/2024 revision to
avoid breakage.
This change makes changes to accommodate IREE API for this specific revision
Change-Id: I5d916ec31c8add20db82c8f92ef71f8dc32ab069
diff --git a/float_models/iree_exec/mnist.c b/float_models/iree_exec/mnist.c
index 7e7d519..76e8738 100644
--- a/float_models/iree_exec/mnist.c
+++ b/float_models/iree_exec/mnist.c
@@ -49,7 +49,7 @@
}
iree_hal_executable_library_query_fn_t library_query(void) {
- return &mnist_linked_llvm_cpu_library_query;
+ return &mnist_linked_library_query;
}
iree_status_t load_input_data(const MlModel *model, void **buffer,
diff --git a/float_models/iree_exec/mobilenet_v1.c b/float_models/iree_exec/mobilenet_v1.c
index e0e8e2d..f75fe1f 100644
--- a/float_models/iree_exec/mobilenet_v1.c
+++ b/float_models/iree_exec/mobilenet_v1.c
@@ -49,7 +49,7 @@
}
iree_hal_executable_library_query_fn_t library_query(void) {
- return &llvm_module_linked_llvm_cpu_library_query;
+ return &module_linked_library_query;
}
iree_status_t load_input_data(const MlModel *model, void **buffer,
diff --git a/quant_models/iree_exec/hps.c b/quant_models/iree_exec/hps.c
index 7353110..56d8fb5 100644
--- a/quant_models/iree_exec/hps.c
+++ b/quant_models/iree_exec/hps.c
@@ -53,7 +53,7 @@
}
iree_hal_executable_library_query_fn_t library_query(void) {
- return &llvm_module_linked_llvm_cpu_library_query;
+ return &module_linked_library_query;
}
iree_status_t load_input_data(const MlModel *model, void **buffer,
diff --git a/quant_models/iree_exec/mobilenet_v1.c b/quant_models/iree_exec/mobilenet_v1.c
index 4c4f3be..543ef7e 100644
--- a/quant_models/iree_exec/mobilenet_v1.c
+++ b/quant_models/iree_exec/mobilenet_v1.c
@@ -49,7 +49,7 @@
}
iree_hal_executable_library_query_fn_t library_query(void) {
- return &llvm_module_linked_llvm_cpu_library_query;
+ return &module_linked_library_query;
}
iree_status_t load_input_data(const MlModel *model, void **buffer,
diff --git a/quant_models/iree_exec/mobilenet_v2.c b/quant_models/iree_exec/mobilenet_v2.c
index 4061f54..2a27c5b 100644
--- a/quant_models/iree_exec/mobilenet_v2.c
+++ b/quant_models/iree_exec/mobilenet_v2.c
@@ -49,7 +49,7 @@
}
iree_hal_executable_library_query_fn_t library_query(void) {
- return &llvm_module_linked_llvm_cpu_library_query;
+ return &module_linked_library_query;
}
iree_status_t load_input_data(const MlModel *model, void **buffer,
diff --git a/quant_models/iree_exec/person_detection.c b/quant_models/iree_exec/person_detection.c
index ebbd98c..911d850 100644
--- a/quant_models/iree_exec/person_detection.c
+++ b/quant_models/iree_exec/person_detection.c
@@ -63,7 +63,7 @@
#if !defined(BUILD_VMVX)
iree_hal_executable_library_query_fn_t library_query(void) {
- return &llvm_module_linked_llvm_cpu_library_query;
+ return &module_linked_library_query;
}
#endif
