Skip to content

Commit b18bf62

Browse files
committed
Adjust VMM rollback tests for release-fail retries
1 parent 23647d8 commit b18bf62

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

cuda_core/tests/test_memory.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1205,6 +1205,7 @@ def fake_from_handle(ptr, size, mr):
12051205
"set_access",
12061206
"release",
12071207
"unmap",
1208+
"release",
12081209
"addr_free",
12091210
]
12101211
assert calls.count(("release", HANDLE)) == 2
@@ -1287,6 +1288,7 @@ def __init__(self, size):
12871288
"set_access",
12881289
"release",
12891290
"unmap",
1291+
"release",
12901292
"addr_free",
12911293
]
12921294

@@ -1508,6 +1510,7 @@ def __init__(self, ptr, size, mr):
15081510
"create",
15091511
"map",
15101512
"set_access",
1513+
"release",
15111514
"unmap",
15121515
"release",
15131516
"unmap",

0 commit comments

Comments
 (0)