Skip to content

Commit 3c4db87

Browse files
Merge branch 'ai-dynamo:main' into main
2 parents bf48cd9 + d744784 commit 3c4db87

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

examples/python/partial_md_example.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -114,8 +114,12 @@ def invalidate_target_metadata(
114114
target_strs2.append((addr1, buf_size, 0, "test"))
115115
malloc_addrs.append(addr1)
116116

117-
target_reg_descs1 = target_agent.get_reg_descs(target_strs1, "DRAM", is_sorted=True)
118-
target_reg_descs2 = target_agent.get_reg_descs(target_strs2, "DRAM", is_sorted=True)
117+
target_reg_descs1 = target_agent.get_reg_descs(
118+
target_strs1, "DRAM", is_sorted=False
119+
)
120+
target_reg_descs2 = target_agent.get_reg_descs(
121+
target_strs2, "DRAM", is_sorted=False
122+
)
119123
target_xfer_descs1 = target_reg_descs1.trim()
120124
target_xfer_descs2 = target_reg_descs2.trim()
121125

@@ -132,7 +136,7 @@ def invalidate_target_metadata(
132136
init_strs.append((addr1, buf_size, 0, "test"))
133137
malloc_addrs.append(addr1)
134138

135-
init_reg_descs = init_agent.get_reg_descs(init_strs, "DRAM", is_sorted=True)
139+
init_reg_descs = init_agent.get_reg_descs(init_strs, "DRAM", is_sorted=False)
136140
init_xfer_descs = init_reg_descs.trim()
137141

138142
assert init_agent.register_memory(init_reg_descs) is not None
@@ -207,7 +211,7 @@ def invalidate_target_metadata(
207211
try:
208212
# initialize transfer mode
209213
xfer_handle_2 = init_agent.initialize_xfer(
210-
"READ", init_xfer_descs, target_xfer_descs1, "target", b"UUID1"
214+
"READ", init_xfer_descs, target_xfer_descs2, "target", b"UUID1"
211215
)
212216
except nixlNotFoundError:
213217
ready = False

0 commit comments

Comments
 (0)