diff --git a/samples/cuda_core/streamingCopyComputeOverlap/streamingCopyComputeOverlap.py b/samples/cuda_core/streamingCopyComputeOverlap/streamingCopyComputeOverlap.py index e4ea6530133..d28bfb28fdc 100644 --- a/samples/cuda_core/streamingCopyComputeOverlap/streamingCopyComputeOverlap.py +++ b/samples/cuda_core/streamingCopyComputeOverlap/streamingCopyComputeOverlap.py @@ -103,7 +103,11 @@ def main(): device.set_current() if not device.properties.memory_pools_supported: - print("PinnedMemoryResource requires CUDA memory pools, which are not supported on this device. Waiving.") + print("DeviceMemoryResource requires CUDA memory pools, which are not supported on this device. Waiving.") + sys.exit(EXIT_WAIVED) + + if not device.properties.host_memory_pools_supported: + print("PinnedMemoryResource requires host memory pools, which are not supported on this device. Waiving.") sys.exit(EXIT_WAIVED) print()