From 1d47593bb7ac3b075829739360b3d2897978e419 Mon Sep 17 00:00:00 2001
From: gpetters-amd <159576198+gpetters-amd@users.noreply.github.com>
Date: Tue, 17 Sep 2024 04:45:37 -0400
Subject: [PATCH] Fix a bug in SD vmfb naming

---
 models/turbine_models/custom_models/sd_inference/utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/models/turbine_models/custom_models/sd_inference/utils.py b/models/turbine_models/custom_models/sd_inference/utils.py
index cc8591b9..69284e67 100644
--- a/models/turbine_models/custom_models/sd_inference/utils.py
+++ b/models/turbine_models/custom_models/sd_inference/utils.py
@@ -299,7 +299,8 @@ def compile_to_vmfb(
         safe_mlir_name = safe_name
     else:
         safe_vmfb_name = safe_name
-        safe_mlir_name = "".join(safe_name.split(target_triple))
+        (dir, file) = os.path.split(safe_name)
+        safe_mlir_name = os.path.join(dir, "".join(file.split(target_triple)))
 
     if mlir_source == "file":
         flatbuffer_blob = ireec.compile_file(