github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-02-19 11:56:43 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix resnet50 vulkan_gui to work with tank models

Browse Source
This commit is contained in:
Anush Elangovan
2022-09-13 19:21:53 -07:00
parent 8d21292d34
commit e212ff2071
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cpp/CMakeLists.txt
View File

@@ -27,8 +27,8 @@ include(FetchContent)
FetchContent_Declare(
iree
GIT_REPOSITORY https://github.com/iree-org/iree.git
GIT_TAG c4624870a8de85224c3f02f4ea8e26837cd99e2f # 2022-08-12
GIT_REPOSITORY https://github.com/nod-ai/shark-runtime.git
GIT_TAG f5d84869cf4b745cc2461396978886e503861a88
GIT_SUBMODULES_RECURSE OFF
GIT_SHALLOW OFF
GIT_PROGRESS ON

6
cpp/vulkan_gui/vulkan_inference_gui.cc
View File

@@ -895,13 +895,13 @@ extern "C" int iree_main(int argc, char** argv) {
*/
iree_file_toc_t module_file_toc;
load_file("amd-resnet50.vmfb", &module_file_toc.data, &module_file_toc.size);
load_file("resnet50_tf.vmfb", &module_file_toc.data, &module_file_toc.size);
fprintf(stdout, "module size: %lu\n", module_file_toc.size);
static float input_res50[224*224*3];
static float output_res50[1000];
char filename[] = "dog_imagenet.jpg";
char filename[] = "vulkan_gui/dog_imagenet.jpg";
fprintf(stdout, "loading: %s\n", filename);
int x,y,n;
unsigned char *image_raw = stbi_load(filename, &x, &y, &n, 0);
@@ -954,7 +954,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},