From 3b0ec8ce4e4a67ade03528d54df5a419a53a201b Mon Sep 17 00:00:00 2001
From: Anush Elangovan <anush@nod-labs.com>
Date: Wed, 14 Sep 2022 16:56:20 -0700
Subject: [PATCH] Update resnet paths

---
 cpp/{vulkan_gui => }/dog_imagenet.jpg  | Bin
 cpp/vulkan_gui/vulkan_inference_gui.cc |  14 +++++++-------
 2 files changed, 7 insertions(+), 7 deletions(-)
 rename cpp/{vulkan_gui => }/dog_imagenet.jpg (100%)

diff --git a/cpp/vulkan_gui/dog_imagenet.jpg b/cpp/dog_imagenet.jpg
similarity index 100%
rename from cpp/vulkan_gui/dog_imagenet.jpg
rename to cpp/dog_imagenet.jpg
diff --git a/cpp/vulkan_gui/vulkan_inference_gui.cc b/cpp/vulkan_gui/vulkan_inference_gui.cc
index 234600a2..6286bedd 100644
--- a/cpp/vulkan_gui/vulkan_inference_gui.cc
+++ b/cpp/vulkan_gui/vulkan_inference_gui.cc
@@ -723,7 +723,7 @@ namespace iree {
 extern "C" int iree_main(int argc, char** argv) {
 
   fprintf(stdout, "starting yo\n");
-  
+
   // --------------------------------------------------------------------------
   // Create a window.
   if (SDL_Init(SDL_INIT_VIDEO | SDL_INIT_TIMER) != 0) {
@@ -747,7 +747,7 @@ extern "C" int iree_main(int argc, char** argv) {
     abort();
     return 1;
   }
-  
+
   // Setup Vulkan
   iree_hal_vulkan_features_t iree_vulkan_features =
       static_cast<iree_hal_vulkan_features_t>(
@@ -898,10 +898,10 @@ extern "C" int iree_main(int argc, char** argv) {
                                        iree_allocator_system(), &hal_module));
 
 
-  // Load bytecode module 
+  // Load bytecode module
   iree_file_toc_t module_file_toc;
-  const char network_model[] = "amd-resnet50.vmfb";
-  fprintf(stdout, "Loading: %s\n", network_model); 
+  const char network_model[] = "resnet50_tf.vmfb";
+  fprintf(stdout, "Loading: %s\n", network_model);
   if (load_file(network_model, &module_file_toc.data, &module_file_toc.size) == false)
   {
       abort();
@@ -967,7 +967,7 @@ extern "C" int iree_main(int argc, char** argv) {
 
   // Lookup the entry point function.
   iree_vm_function_t main_function;
-  const char kMainFunctionName[] = "module.predict";
+  const char kMainFunctionName[] = "module.forward";
   IREE_CHECK_OK(iree_vm_context_resolve_function(
       iree_context,
       iree_string_view_t{kMainFunctionName, sizeof(kMainFunctionName) - 1},
@@ -975,7 +975,7 @@ extern "C" int iree_main(int argc, char** argv) {
   iree_string_view_t main_function_name = iree_vm_function_name(&main_function);
   fprintf(stdout, "Resolved main function named '%.*s'\n",
           (int)main_function_name.size, main_function_name.data);
-          
+
   // --------------------------------------------------------------------------
 
         // Write inputs into mappable buffers.