Skip to content

Commit 4bdbfca

Browse files
rwgkrwgk
committed
fix(samples): waive streaming overlap when host pools are unsupported
1 parent 2edf629 commit 4bdbfca

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

‎samples/cuda_core/streamingCopyComputeOverlap/streamingCopyComputeOverlap.py‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,11 @@ def main():
103103
device.set_current()
104104

105105
if not device.properties.memory_pools_supported:
106-
print("PinnedMemoryResource requires CUDA memory pools, which are not supported on this device. Waiving.")
106+
print("DeviceMemoryResource requires CUDA memory pools, which are not supported on this device. Waiving.")
107+
sys.exit(EXIT_WAIVED)
108+
109+
if not device.properties.host_memory_pools_supported:
110+
print("PinnedMemoryResource requires host memory pools, which are not supported on this device. Waiving.")
107111
sys.exit(EXIT_WAIVED)
108112

109113
print()

0 commit comments

Comments
 (0)