Skip to content

Commit 5db2668

Browse files
committed
fix(agentic): avoid partial LMCache import patching
1 parent 0103241 commit 5db2668

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

benchmarks/single_node/agentic/kimik2.5_fp4_mi355x.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ if os.environ.get("LMCACHE_ROCM_DEMAND_PINNED_ALLOCATOR") == "1":
158158
159159
def _maybe_patch_lazy_memory_allocator() -> None:
160160
module = sys.modules.get("lmcache.v1.lazy_memory_allocator")
161-
if module is not None:
161+
if module is not None and hasattr(module, "LazyMemoryAllocator"):
162162
_patch_lazy_memory_allocator(module)
163163
164164
def _agentic_rocm_import(name, globals=None, locals=None, fromlist=(), level=0):

0 commit comments

Comments
 (0)