Lerking/Hands-On-GPU-Programming-with-CUDA-C-and-Python-3.x-Second-Edition
1
0
Fork 0
mirror of https://github.com/PacktPublishing/Hands-On-GPU-Programming-with-CUDA-C-and-Python-3.x-Second-Edition.git synced 2025-07-21 21:01:06 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update deviceQuery.py

Browse Source
This commit is contained in:
Brian Tuomanen
2020-02-13 21:57:53 -08:00
committed by GitHub
parent 6d65cde154
commit 1801ffc391
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
Chapter03/deviceQuery.py
View File

@@ -32,7 +32,8 @@ for i in range(drv.Device.count()):
# See the following:
# http://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#compute-capabilities
cuda_cores_per_mp = { 5.0 : 128, 5.1 : 128, 5.2 : 128, 6.0 : 64, 6.1 : 128, 6.2 : 128, 7.0 : 64, 7.1 : 64, 7.2 : 64, 7.3 : 64, 7.4 : 64, 7.5 : 64}[compute_capability]
cuda_cores_per_mp = { 3.0 : 192, 3.2 : 192, 3.5 : 192, 3.7 : 192, 5.0 : 128, 5.1 : 128, 5.2 : 128, 5.3 : 128, 6.0 : 64, 6.1 : 128,\
6.2 : 128, 7.0 : 64, 7.1 : 64, 7.2 : 64, 7.3 : 64, 7.4 : 64, 7.5 : 64}[compute_capability]
print ('\t ({}) Multiprocessors, ({}) CUDA Cores / Multiprocessor: {} CUDA Cores'.format(num_mp, cuda_cores_per_mp, num_mp*cuda_cores_per_mp))