fix: Strengthen join condition between kernels and images (#2993) (#2996

) Co-authored-by: Sanghun Lee <[email protected]>
lablup · Oct 29, 2024 · a8c6e9c · a8c6e9c
1 parent 8b8ea76
commit a8c6e9c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/changes/2993.fix.md b/changes/2993.fix.md
@@ -0,0 +1 @@
+Strengthen join condition between kernels and images to prevent incorrect matches
diff --git a/src/ai/backend/manager/models/kernel.py b/src/ai/backend/manager/models/kernel.py
@@ -536,7 +536,7 @@ class KernelRow(Base):
     image_row = relationship(
         "ImageRow",
         foreign_keys="KernelRow.image",
-        primaryjoin="KernelRow.image == ImageRow.name",
+        primaryjoin="and_(KernelRow.image == ImageRow.name, KernelRow.architecture == ImageRow.architecture)",
     )
     agent_row = relationship("AgentRow", back_populates="kernels")
     group_row = relationship("GroupRow", back_populates="kernels")