mirror of
https://github.com/NVIDIA/cuda-samples.git
synced 2026-10-13 00:08:26 +08:00
CUDA 13.4 samples update - v13.4-public
Release 13.4 of the CUDA Samples supported by CUDA Toolkit 13.4. See Changelog for more information.
This commit is contained in:
@@ -96,6 +96,13 @@ def main():
|
||||
print()
|
||||
print_gpu_info(device)
|
||||
|
||||
# This sample stages transfers through pinned memory, which is backed by a
|
||||
# host memory pool and requires CUDA memory-pool support. This is not
|
||||
# available on every platform (for example, Windows in TCC mode).
|
||||
if not device.properties.host_memory_pools_supported:
|
||||
print("Host pinned memory pools are not supported on this platform.")
|
||||
return 2
|
||||
|
||||
# Compile kernel
|
||||
arch = f"sm_{device.arch}"
|
||||
program = Program(
|
||||
@@ -306,7 +313,8 @@ def main():
|
||||
print("\nNote: Speedup depends on hardware characteristics. This technique")
|
||||
print("benefits most when transfer time is significant relative to compute.")
|
||||
print("=" * 60)
|
||||
return 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
sys.exit(main())
|
||||
|
||||
Reference in New Issue
Block a user