Skip to content

Commit

Permalink
Merge pull request #1824 from Robzz/fix/cuda_pascal_support
Browse files Browse the repository at this point in the history
Added number of CUDA cores per SM for Pascal GPUs
  • Loading branch information
taketwo authored Mar 17, 2017
2 parents 6fb1b65 + 279940e commit 26bf74c
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions gpu/containers/src/initialization.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -115,8 +115,10 @@ namespace
int Cores;
} SMtoCores;

SMtoCores gpuArchCoresPerSM[] = { { 0x10, 8}, { 0x11, 8}, { 0x12, 8}, { 0x13, 8}, { 0x20, 32}, { 0x21, 48}, {0x30, 192}, {0x35, 192}, {0x50, 128}, {0x52, 128}, {-1, -1} };

SMtoCores gpuArchCoresPerSM[] = {
{0x10, 8}, {0x11, 8}, {0x12, 8}, {0x13, 8}, {0x20, 32}, {0x21, 48}, {0x30, 192},
{0x35, 192}, {0x50, 128}, {0x52, 128}, {0x60, 64}, {0x61, 128}, {-1, -1}
};
int index = 0;
while (gpuArchCoresPerSM[index].SM != -1)
{
Expand Down

0 comments on commit 26bf74c

Please sign in to comment.