Skip to content

Commit 41d7268

Browse files
committed
Disable additional tests for platform-dependent behavior.
1 parent 45db4ff commit 41d7268

2 files changed

Lines changed: 5 additions & 3 deletions

File tree

cuda_core/tests/test_graph_mem.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -200,8 +200,10 @@ def test_graph_mem_set_attributes(init_cuda, mode):
200200
gmr.attributes.reserved_mem_high = 0
201201
gmr.attributes.used_mem_high = 0
202202

203-
assert gmr.attributes.reserved_mem_high == 0
204-
assert gmr.attributes.used_mem_high == 0
203+
# High-water marks are not guaranteed to be reset. This is
204+
# platform-dependent behavior.
205+
# assert gmr.attributes.reserved_mem_high == 0
206+
# assert gmr.attributes.used_mem_high == 0
205207

206208
mman.reset()
207209

cuda_core/tests/test_memory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -664,7 +664,7 @@ def test_mempool_attributes_repr(mempool_device):
664664
assert re.match(
665665
r"DeviceMemoryResourceAttributes\(release_threshold=\d+, reserved_mem_current=\d+, reserved_mem_high=\d+, "
666666
r"reuse_allow_internal_dependencies=(True|False), reuse_allow_opportunistic=(True|False), "
667-
r"reuse_follow_event_dependencies=(True|False), used_mem_current=64, used_mem_high=128\)",
667+
r"reuse_follow_event_dependencies=(True|False), used_mem_current=\d+, used_mem_high=\d+\)",
668668
str(mr.attributes),
669669
)
670670
buffer2.close()

0 commit comments

Comments
 (0)