diff options
Diffstat (limited to 'src/database/kernels/xconvgemm')
| -rw-r--r-- | src/database/kernels/xconvgemm/xconvgemm_16.hpp | 10 | ||||
| -rw-r--r-- | src/database/kernels/xconvgemm/xconvgemm_32.hpp | 18 | ||||
| -rw-r--r-- | src/database/kernels/xconvgemm/xconvgemm_3232.hpp | 2 | ||||
| -rw-r--r-- | src/database/kernels/xconvgemm/xconvgemm_64.hpp | 4 | ||||
| -rw-r--r-- | src/database/kernels/xconvgemm/xconvgemm_6464.hpp | 2 |
5 files changed, 32 insertions, 4 deletions
diff --git a/src/database/kernels/xconvgemm/xconvgemm_16.hpp b/src/database/kernels/xconvgemm/xconvgemm_16.hpp index 841b6b38..a1162572 100644 --- a/src/database/kernels/xconvgemm/xconvgemm_16.hpp +++ b/src/database/kernels/xconvgemm/xconvgemm_16.hpp @@ -15,7 +15,7 @@ const DatabaseEntry XconvgemmHalf = { { // AMD GPUs kDeviceTypeGPU, "AMD", { { "default", { - { kDeviceNameDefault , Params{ 1, 8, 16, 16, 8, 0, 0, 1, 2, 32, 0, 0, 0, 0, 0, 0 } }, + { kDeviceNameDefault , Params{ 1, 8, 16, 16, 8, 0, 0, 2, 1, 32, 0, 0, 0, 0, 0, 0 } }, } }, { "gfx1010:xnack-", { { Name{"AMD Radeon RX 5700 XT "}, Params{ 1, 8, 16, 16, 8, 0, 0, 1, 2, 32, 0, 0, 0, 0, 0, 0 } }, @@ -37,6 +37,14 @@ const DatabaseEntry XconvgemmHalf = { { Name{"AMD Radeon Graphics "}, Params{ 1, 8, 16, 16, 8, 0, 0, 2, 1, 32, 0, 0, 0, 0, 0, 0 } }, { kDeviceNameDefault , Params{ 1, 8, 16, 16, 8, 0, 0, 2, 1, 32, 0, 0, 0, 0, 0, 0 } }, } }, + { "gfx902", { + { Name{"AMD Radeon(TM) RX Vega 10 Graphics "}, Params{ 1, 8, 16, 8, 8, 0, 0, 2, 4, 32, 0, 0, 0, 0, 0, 0 } }, + { kDeviceNameDefault , Params{ 1, 8, 16, 8, 8, 0, 0, 2, 4, 32, 0, 0, 0, 0, 0, 0 } }, + } }, + { "gfx906:sramecc+:xnack-", { + { Name{"AMD Radeon VII "}, Params{ 1, 8, 16, 8, 8, 0, 0, 2, 4, 32, 0, 0, 0, 0, 0, 0 } }, + { kDeviceNameDefault , Params{ 1, 8, 16, 8, 8, 0, 0, 2, 4, 32, 0, 0, 0, 0, 0, 0 } }, + } }, } }, { // Intel GPUs diff --git a/src/database/kernels/xconvgemm/xconvgemm_32.hpp b/src/database/kernels/xconvgemm/xconvgemm_32.hpp index 33cc5655..557841e2 100644 --- a/src/database/kernels/xconvgemm/xconvgemm_32.hpp +++ b/src/database/kernels/xconvgemm/xconvgemm_32.hpp @@ -47,6 +47,14 @@ const DatabaseEntry XconvgemmSingle = { { Name{"AMD Radeon Graphics "}, Params{ 1, 8, 8, 16, 8, 0, 0, 1, 2, 32, 0, 0, 0, 0, 0, 0 } }, { kDeviceNameDefault , Params{ 1, 8, 8, 16, 8, 0, 0, 1, 2, 32, 0, 0, 0, 0, 0, 0 } }, } }, + { "gfx902", { + { Name{"AMD Radeon(TM) RX Vega 10 Graphics "}, Params{ 1, 8, 8, 16, 8, 0, 0, 2, 1, 32, 0, 0, 0, 0, 0, 0 } }, + { kDeviceNameDefault , Params{ 1, 8, 8, 16, 8, 0, 0, 2, 1, 32, 0, 0, 0, 0, 0, 0 } }, + } }, + { "gfx906:sramecc+:xnack-", { + { Name{"AMD Radeon VII "}, Params{ 1, 16, 16, 8, 8, 0, 0, 1, 4, 32, 0, 0, 0, 0, 0, 0 } }, + { kDeviceNameDefault , Params{ 1, 16, 16, 8, 8, 0, 0, 1, 4, 32, 0, 0, 0, 0, 0, 0 } }, + } }, { "gfx90c", { { Name{"AMD Radeon(TM) Graphics "}, Params{ 1, 8, 8, 16, 8, 0, 0, 4, 1, 32, 0, 0, 0, 0, 0, 0 } }, { kDeviceNameDefault , Params{ 1, 8, 8, 16, 8, 0, 0, 4, 1, 32, 0, 0, 0, 0, 0, 0 } }, @@ -61,6 +69,14 @@ const DatabaseEntry XconvgemmSingle = { } }, } }, + { // Imagination Technologies GPUs + kDeviceTypeGPU, "Imagination Technologies", { + { "default", { + { Name{"PowerVR B-Series BXE-4-32 "}, Params{ 1, 16, 16, 16, 16, 0, 0, 1, 1, 16, 0, 0, 0, 0, 0, 0 } }, + { kDeviceNameDefault , Params{ 1, 16, 16, 16, 16, 0, 0, 1, 1, 16, 0, 0, 0, 0, 0, 0 } }, + } }, + } + }, { // Intel CPUs kDeviceTypeCPU, "Intel", { { "default", { @@ -121,7 +137,7 @@ const DatabaseEntry XconvgemmSingle = { { // Default kDeviceTypeAll, "default", { { "default", { - { kDeviceNameDefault , Params{ 1, 8, 32, 32, 8, 0, 0, 1, 1, 32, 0, 0, 0, 0, 0, 0 } }, + { kDeviceNameDefault , Params{ 1, 8, 16, 8, 8, 0, 0, 1, 1, 16, 0, 0, 0, 0, 0, 0 } }, } }, } }, diff --git a/src/database/kernels/xconvgemm/xconvgemm_3232.hpp b/src/database/kernels/xconvgemm/xconvgemm_3232.hpp index d884b1b3..3f5027ec 100644 --- a/src/database/kernels/xconvgemm/xconvgemm_3232.hpp +++ b/src/database/kernels/xconvgemm/xconvgemm_3232.hpp @@ -15,7 +15,7 @@ const DatabaseEntry XconvgemmComplexSingle = { { // Default kDeviceTypeAll, "default", { { "default", { - { kDeviceNameDefault , Params{ 1, 8, 32, 32, 8, 0, 0, 1, 1, 32, 0, 0, 0, 0, 0, 0 } }, + { kDeviceNameDefault , Params{ 1, 8, 16, 8, 8, 0, 0, 1, 1, 16, 0, 0, 0, 0, 0, 0 } }, } }, } }, diff --git a/src/database/kernels/xconvgemm/xconvgemm_64.hpp b/src/database/kernels/xconvgemm/xconvgemm_64.hpp index 4ff7bd93..cc17bcd4 100644 --- a/src/database/kernels/xconvgemm/xconvgemm_64.hpp +++ b/src/database/kernels/xconvgemm/xconvgemm_64.hpp @@ -38,6 +38,10 @@ const DatabaseEntry XconvgemmDouble = { { Name{"AMD Radeon Graphics "}, Params{ 1, 8, 16, 8, 8, 0, 0, 1, 4, 32, 0, 0, 0, 0, 0, 0 } }, { kDeviceNameDefault , Params{ 1, 8, 16, 8, 8, 0, 0, 1, 4, 32, 0, 0, 0, 0, 0, 0 } }, } }, + { "gfx906:sramecc+:xnack-", { + { Name{"AMD Radeon VII "}, Params{ 1, 16, 32, 8, 8, 0, 0, 1, 1, 32, 0, 0, 0, 0, 0, 0 } }, + { kDeviceNameDefault , Params{ 1, 16, 32, 8, 8, 0, 0, 1, 1, 32, 0, 0, 0, 0, 0, 0 } }, + } }, } }, { // Intel CPUs diff --git a/src/database/kernels/xconvgemm/xconvgemm_6464.hpp b/src/database/kernels/xconvgemm/xconvgemm_6464.hpp index 3b40433c..cdcd52e6 100644 --- a/src/database/kernels/xconvgemm/xconvgemm_6464.hpp +++ b/src/database/kernels/xconvgemm/xconvgemm_6464.hpp @@ -15,7 +15,7 @@ const DatabaseEntry XconvgemmComplexDouble = { { // Default kDeviceTypeAll, "default", { { "default", { - { kDeviceNameDefault , Params{ 1, 8, 32, 32, 8, 0, 0, 1, 1, 32, 0, 0, 0, 0, 0, 0 } }, + { kDeviceNameDefault , Params{ 1, 8, 16, 8, 8, 0, 0, 1, 1, 16, 0, 0, 0, 0, 0, 0 } }, } }, } }, |