ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-12 01:10:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

implicit gemm v3 added lds double buffer

Browse Source
This commit is contained in:
Chao Liu
2019-05-24 17:17:22 -05:00
parent e17b495dff
commit 8c385cf5cf
3 changed files with 425 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
driver/driver.hip.cpp
View File

@@ -422,7 +422,7 @@ int main(int argc, char* argv[])
constexpr index_t HPad = 0;
constexpr index_t WPad = 0;
#elif 1
#elif 0
// 3x3, 34x34
constexpr index_t N = 64;
constexpr index_t C = 256;
@@ -446,7 +446,7 @@ int main(int argc, char* argv[])
constexpr index_t HPad = 0;
constexpr index_t WPad = 0;
#elif 0
#elif 1
// 3x3 filter, 28x28 image
constexpr index_t N = 128;
constexpr index_t C = 256;