summaryrefslogtreecommitdiff
path: root/src/database/kernels/gemm_routine/gemm_routine_32.hpp
diff options
context:
space:
mode:
authorCedric Nugteren <web@cedricnugteren.nl>2019-02-09 16:31:48 +0100
committerCedric Nugteren <web@cedricnugteren.nl>2019-02-09 16:31:48 +0100
commit7084311e454f312094195df06fba12996e64fb05 (patch)
tree9c36d3eac8587ffaa669c896ac4cd26d1fbe6189 /src/database/kernels/gemm_routine/gemm_routine_32.hpp
parent1035e533cd5fb42ae9309aeb7a65a6b7833323a6 (diff)
Added tuning parameters for Tesla P100 16GB
Diffstat (limited to 'src/database/kernels/gemm_routine/gemm_routine_32.hpp')
-rw-r--r--src/database/kernels/gemm_routine/gemm_routine_32.hpp8
1 files changed, 6 insertions, 2 deletions
diff --git a/src/database/kernels/gemm_routine/gemm_routine_32.hpp b/src/database/kernels/gemm_routine/gemm_routine_32.hpp
index 0270d687..97b9c5ed 100644
--- a/src/database/kernels/gemm_routine/gemm_routine_32.hpp
+++ b/src/database/kernels/gemm_routine/gemm_routine_32.hpp
@@ -51,6 +51,10 @@ const DatabaseEntry GemmRoutineSingle = {
{ Name{"GeForce GTX 970 "}, Params{ 1984, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } },
{ kDeviceNameDefault , Params{ 1984, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } },
} },
+ { "SM6.0", {
+ { Name{"Tesla P100-PCIE-16GB "}, Params{ 1984, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } },
+ { kDeviceNameDefault , Params{ 1984, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } },
+ } },
{ "SM6.1", {
{ Name{"GeForce GTX 1070 Ti "}, Params{ 1472, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } },
{ Name{"GeForce GTX 1080 Ti "}, Params{ 1792, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } },
@@ -58,14 +62,14 @@ const DatabaseEntry GemmRoutineSingle = {
{ kDeviceNameDefault , Params{ 1536, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } },
} },
{ "default", {
- { kDeviceNameDefault , Params{ 1472, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } },
+ { kDeviceNameDefault , Params{ 1536, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } },
} },
}
},
{ // Default
kDeviceTypeAll, "default", {
{ "default", {
- { kDeviceNameDefault , Params{ 704, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } },
+ { kDeviceNameDefault , Params{ 832, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } },
} },
}
},