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

add conv1d/3d bwd weight instances (#318)

Browse Source 
* add conv1d/3d bwd weight instances

* add profiler code
This commit is contained in:
Shaojie WANG
2022-07-09 04:42:20 +08:00
committed by GitHub
parent 4fe9c393b8
commit 763ca61581
19 changed files with 1907 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
profiler/src/profiler.cpp
View File

@@ -20,6 +20,7 @@ int profile_convnd_bwd_data(int, char*[], int);
int profile_conv_bwd_weight(int, char*[]);
int profile_normalization(int, char*[]);
int profile_reduce(int, char*[]);
int profile_convnd_bwd_weight(int, char*[], int);
static void print_helper_message()
{
@@ -117,6 +118,18 @@ int main(int argc, char* argv[])
{
return profile_conv_bwd_weight(argc, argv);
}
else if(strcmp(argv[1], "convnd1d_bwd_weight") == 0)
{
return profile_convnd_bwd_weight(argc, argv, 1);
}
else if(strcmp(argv[1], "convnd2d_bwd_weight") == 0)
{
return profile_convnd_bwd_weight(argc, argv, 2);
}
else if(strcmp(argv[1], "convnd3d_bwd_weight") == 0)
{
return profile_convnd_bwd_weight(argc, argv, 3);
}
else if(strcmp(argv[1], "reduce") == 0)
{
return profile_reduce(argc, argv);