Skip to content

GitLab

Commit 66fd0f8d authored by Wen-Heng (Jack) Chung's avatar Wen-Heng (Jack) Chung
Browse files

XXX only enable gemm.

parent 4221505d
Show whitespace changes
Inline Side-by-side
Showing with 151 additions and 151 deletions
example/CMakeLists.txt
View file @ 66fd0f8d
......@@ -27,18 +27,18 @@ function(add_example_executable EXAMPLE_NAME)
endfunction(add_example_executable EXAMPLE_NAME)
add_subdirectory(01_gemm)
add_subdirectory(02_gemm_alpha_beta)
add_subdirectory(03_gemm_bias_relu)
add_subdirectory(04_gemm_bias_relu_add)
add_subdirectory(06_conv2d_fwd_bias_relu)
add_subdirectory(07_conv2d_fwd_bias_relu_add)
add_subdirectory(09_convnd_fwd)
add_subdirectory(10_conv2d_bwd_data)
add_subdirectory(11_conv2d_bwd_weight)
add_subdirectory(12_reduce)
add_subdirectory(13_pool2d_fwd)
add_subdirectory(14_gemm_xdl_requant_relu_requant)
add_subdirectory(17_convnd_bwd_data_xdl)
add_subdirectory(15_grouped_gemm)
add_subdirectory(16_gemm_reduce)
add_subdirectory(18_batched_gemm_reduce)
#add_subdirectory(02_gemm_alpha_beta)
#add_subdirectory(03_gemm_bias_relu)
#add_subdirectory(04_gemm_bias_relu_add)
#add_subdirectory(06_conv2d_fwd_bias_relu)
#add_subdirectory(07_conv2d_fwd_bias_relu_add)
#add_subdirectory(09_convnd_fwd)
#add_subdirectory(10_conv2d_bwd_data)
#add_subdirectory(11_conv2d_bwd_weight)
#add_subdirectory(12_reduce)
#add_subdirectory(13_pool2d_fwd)
#add_subdirectory(14_gemm_xdl_requant_relu_requant)
#add_subdirectory(17_convnd_bwd_data_xdl)
#add_subdirectory(15_grouped_gemm)
#add_subdirectory(16_gemm_reduce)
#add_subdirectory(18_batched_gemm_reduce)
library/src/tensor_operation_instance/gpu/CMakeLists.txt
View file @ 66fd0f8d
......@@ -24,20 +24,20 @@ function(add_instance_library INSTANCE_NAME)
endfunction(add_instance_library INSTANCE_NAME)
add_subdirectory(gemm)
add_subdirectory(gemm_bias2d)
add_subdirectory(gemm_bias_relu)
add_subdirectory(gemm_bias_relu_add)
add_subdirectory(gemm_reduce)
add_subdirectory(batched_gemm)
add_subdirectory(conv1d_fwd)
add_subdirectory(conv2d_fwd)
add_subdirectory(conv3d_fwd)
add_subdirectory(conv2d_fwd_bias_relu)
add_subdirectory(conv2d_fwd_bias_relu_add)
add_subdirectory(conv2d_fwd_bias_relu_atomic_add)
add_subdirectory(conv2d_bwd_data)
add_subdirectory(reduce)
add_subdirectory(convnd_bwd_data)
add_subdirectory(grouped_gemm)
add_subdirectory(conv2d_bwd_weight)
add_subdirectory(batched_gemm_reduce)
#add_subdirectory(gemm_bias2d)
#add_subdirectory(gemm_bias_relu)
#add_subdirectory(gemm_bias_relu_add)
#add_subdirectory(gemm_reduce)
#add_subdirectory(batched_gemm)
#add_subdirectory(conv1d_fwd)
#add_subdirectory(conv2d_fwd)
#add_subdirectory(conv3d_fwd)
#add_subdirectory(conv2d_fwd_bias_relu)
#add_subdirectory(conv2d_fwd_bias_relu_add)
#add_subdirectory(conv2d_fwd_bias_relu_atomic_add)
#add_subdirectory(conv2d_bwd_data)
#add_subdirectory(reduce)
#add_subdirectory(convnd_bwd_data)
#add_subdirectory(grouped_gemm)
#add_subdirectory(conv2d_bwd_weight)
#add_subdirectory(batched_gemm_reduce)
profiler/CMakeLists.txt
View file @ 66fd0f8d
......@@ -24,38 +24,38 @@ include_directories(BEFORE
set(PROFILER_SOURCE
src/profiler.cpp
src/profile_gemm.cpp
src/profile_gemm_bias_2d.cpp
src/profile_gemm_bias_relu.cpp
src/profile_gemm_bias_relu_add.cpp
src/profile_gemm_reduce.cpp
src/profile_batched_gemm.cpp
src/profile_conv_fwd.cpp
src/profile_conv_fwd_bias_relu.cpp
src/profile_conv_fwd_bias_relu_add.cpp
src/profile_conv_fwd_bias_relu_atomic_add.cpp
src/profile_convnd_bwd_data.cpp
src/profile_reduce.cpp
src/profile_grouped_gemm.cpp
src/profile_conv_bwd_weight.cpp
src/profile_batched_gemm_reduce.cpp
# src/profile_gemm_bias_2d.cpp
# src/profile_gemm_bias_relu.cpp
# src/profile_gemm_bias_relu_add.cpp
# src/profile_gemm_reduce.cpp
# src/profile_batched_gemm.cpp
# src/profile_conv_fwd.cpp
# src/profile_conv_fwd_bias_relu.cpp
# src/profile_conv_fwd_bias_relu_add.cpp
# src/profile_conv_fwd_bias_relu_atomic_add.cpp
# src/profile_convnd_bwd_data.cpp
# src/profile_reduce.cpp
# src/profile_grouped_gemm.cpp
# src/profile_conv_bwd_weight.cpp
# src/profile_batched_gemm_reduce.cpp
)
add_executable(ckProfiler ${PROFILER_SOURCE})
target_link_libraries(ckProfiler PRIVATE host_tensor)
target_link_libraries(ckProfiler PRIVATE device_gemm_reduce_instance)
#target_link_libraries(ckProfiler PRIVATE device_gemm_reduce_instance)
target_link_libraries(ckProfiler PRIVATE device_gemm_instance)
target_link_libraries(ckProfiler PRIVATE device_gemm_bias2d_instance)
target_link_libraries(ckProfiler PRIVATE device_gemm_bias_relu_instance)
target_link_libraries(ckProfiler PRIVATE device_gemm_bias_relu_add_instance)
target_link_libraries(ckProfiler PRIVATE device_batched_gemm_instance)
target_link_libraries(ckProfiler PRIVATE device_conv2d_fwd_instance)
target_link_libraries(ckProfiler PRIVATE device_conv2d_fwd_bias_relu_instance)
target_link_libraries(ckProfiler PRIVATE device_conv2d_fwd_bias_relu_add_instance)
target_link_libraries(ckProfiler PRIVATE device_conv2d_fwd_bias_relu_atomic_add_instance)
target_link_libraries(ckProfiler PRIVATE device_convnd_bwd_data_instance)
target_link_libraries(ckProfiler PRIVATE device_reduce_instance)
target_link_libraries(ckProfiler PRIVATE device_reduce_instance)
target_link_libraries(ckProfiler PRIVATE device_grouped_gemm_instance)
target_link_libraries(ckProfiler PRIVATE device_conv2d_bwd_weight_instance)
target_link_libraries(ckProfiler PRIVATE device_batched_gemm_reduce_instance)
#target_link_libraries(ckProfiler PRIVATE device_gemm_bias2d_instance)
#target_link_libraries(ckProfiler PRIVATE device_gemm_bias_relu_instance)
#target_link_libraries(ckProfiler PRIVATE device_gemm_bias_relu_add_instance)
#target_link_libraries(ckProfiler PRIVATE device_batched_gemm_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv2d_fwd_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv2d_fwd_bias_relu_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv2d_fwd_bias_relu_add_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv2d_fwd_bias_relu_atomic_add_instance)
#target_link_libraries(ckProfiler PRIVATE device_convnd_bwd_data_instance)
#target_link_libraries(ckProfiler PRIVATE device_reduce_instance)
#target_link_libraries(ckProfiler PRIVATE device_reduce_instance)
#target_link_libraries(ckProfiler PRIVATE device_grouped_gemm_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv2d_bwd_weight_instance)
#target_link_libraries(ckProfiler PRIVATE device_batched_gemm_reduce_instance)
profiler/src/profiler.cpp
View file @ 66fd0f8d
......@@ -5,20 +5,20 @@
#include <cstring>
int profile_gemm(int, char*[]);
int profile_gemm_bias_2d(int, char*[]);
int profile_gemm_bias_relu(int, char*[]);
int profile_gemm_bias_relu_add(int, char*[]);
int profile_gemm_reduce(int, char*[]);
int profile_batched_gemm(int, char*[]);
int profile_grouped_gemm(int, char*[]);
int profile_conv_fwd(int, char*[]);
int profile_conv_fwd_bias_relu(int, char*[]);
int profile_conv_fwd_bias_relu_add(int, char*[]);
int profile_conv_fwd_bias_relu_atomic_add(int, char*[]);
int profile_convnd_bwd_data(int, char*[], int);
int profile_reduce(int, char*[]);
int profile_conv_bwd_weight(int, char*[]);
int profile_batched_gemm_reduce(int, char*[]);
//int profile_gemm_bias_2d(int, char*[]);
//int profile_gemm_bias_relu(int, char*[]);
//int profile_gemm_bias_relu_add(int, char*[]);
//int profile_gemm_reduce(int, char*[]);
//int profile_batched_gemm(int, char*[]);
//int profile_grouped_gemm(int, char*[]);
//int profile_conv_fwd(int, char*[]);
//int profile_conv_fwd_bias_relu(int, char*[]);
//int profile_conv_fwd_bias_relu_add(int, char*[]);
//int profile_conv_fwd_bias_relu_atomic_add(int, char*[]);
//int profile_convnd_bwd_data(int, char*[], int);
//int profile_reduce(int, char*[]);
//int profile_conv_bwd_weight(int, char*[]);
//int profile_batched_gemm_reduce(int, char*[]);
int main(int argc, char* argv[])
{
......@@ -26,70 +26,70 @@ int main(int argc, char* argv[])
{
return profile_gemm(argc, argv);
}
else if(strcmp(argv[1], "gemm_bias_2d") == 0)
{
return profile_gemm_bias_2d(argc, argv);
}
else if(strcmp(argv[1], "gemm_bias_relu") == 0)
{
return profile_gemm_bias_relu(argc, argv);
}
else if(strcmp(argv[1], "gemm_bias_relu_add") == 0)
{
return profile_gemm_bias_relu_add(argc, argv);
}
else if(strcmp(argv[1], "gemm_reduce") == 0)
{
return profile_gemm_reduce(argc, argv);
}
else if(strcmp(argv[1], "batched_gemm") == 0)
{
return profile_batched_gemm(argc, argv);
}
else if(strcmp(argv[1], "batched_gemm_reduce") == 0)
{
return profile_batched_gemm_reduce(argc, argv);
}
else if(strcmp(argv[1], "grouped_gemm") == 0)
{
profile_grouped_gemm(argc, argv);
}
else if(strcmp(argv[1], "conv_fwd") == 0)
{
return profile_conv_fwd(argc, argv);
}
else if(strcmp(argv[1], "conv_fwd_bias_relu") == 0)
{
return profile_conv_fwd_bias_relu(argc, argv);
}
else if(strcmp(argv[1], "conv_fwd_bias_relu_add") == 0)
{
return profile_conv_fwd_bias_relu_add(argc, argv);
}
else if(strcmp(argv[1], "conv_fwd_bias_relu_atomic_add") == 0)
{
return profile_conv_fwd_bias_relu_atomic_add(argc, argv);
}
else if(strcmp(argv[1], "conv1d_bwd_data") == 0)
{
return profile_convnd_bwd_data(argc, argv, 1);
}
else if(strcmp(argv[1], "conv2d_bwd_data") == 0)
{
return profile_convnd_bwd_data(argc, argv, 2);
}
else if(strcmp(argv[1], "conv3d_bwd_data") == 0)
{
return profile_convnd_bwd_data(argc, argv, 3);
}
else if(strcmp(argv[1], "reduce") == 0)
{
return profile_reduce(argc, argv);
}
else if(strcmp(argv[1], "conv2d_bwd_weight") == 0)
{
return profile_conv_bwd_weight(argc, argv);
}
//else if(strcmp(argv[1], "gemm_bias_2d") == 0)
//{
// return profile_gemm_bias_2d(argc, argv);
//}
//else if(strcmp(argv[1], "gemm_bias_relu") == 0)
//{
// return profile_gemm_bias_relu(argc, argv);
//}
//else if(strcmp(argv[1], "gemm_bias_relu_add") == 0)
//{
// return profile_gemm_bias_relu_add(argc, argv);
//}
//else if(strcmp(argv[1], "gemm_reduce") == 0)
//{
// return profile_gemm_reduce(argc, argv);
//}
//else if(strcmp(argv[1], "batched_gemm") == 0)
//{
// return profile_batched_gemm(argc, argv);
//}
//else if(strcmp(argv[1], "batched_gemm_reduce") == 0)
//{
// return profile_batched_gemm_reduce(argc, argv);
//}
//else if(strcmp(argv[1], "grouped_gemm") == 0)
//{
// profile_grouped_gemm(argc, argv);
//}
//else if(strcmp(argv[1], "conv_fwd") == 0)
//{
// return profile_conv_fwd(argc, argv);
//}
//else if(strcmp(argv[1], "conv_fwd_bias_relu") == 0)
//{
// return profile_conv_fwd_bias_relu(argc, argv);
//}
//else if(strcmp(argv[1], "conv_fwd_bias_relu_add") == 0)
//{
// return profile_conv_fwd_bias_relu_add(argc, argv);
//}
//else if(strcmp(argv[1], "conv_fwd_bias_relu_atomic_add") == 0)
//{
// return profile_conv_fwd_bias_relu_atomic_add(argc, argv);
//}
//else if(strcmp(argv[1], "conv1d_bwd_data") == 0)
//{
// return profile_convnd_bwd_data(argc, argv, 1);
//}
//else if(strcmp(argv[1], "conv2d_bwd_data") == 0)
//{
// return profile_convnd_bwd_data(argc, argv, 2);
//}
//else if(strcmp(argv[1], "conv3d_bwd_data") == 0)
//{
// return profile_convnd_bwd_data(argc, argv, 3);
//}
//else if(strcmp(argv[1], "reduce") == 0)
//{
// return profile_reduce(argc, argv);
//}
//else if(strcmp(argv[1], "conv2d_bwd_weight") == 0)
//{
// return profile_conv_bwd_weight(argc, argv);
//}
else
{
// clang-format off
......
test/CMakeLists.txt
View file @ 66fd0f8d
......@@ -32,16 +32,16 @@ function(add_test_executable TEST_NAME)
add_dependencies(check ${TEST_NAME})
endfunction(add_test_executable TEST_NAME)
add_subdirectory(magic_number_division)
add_subdirectory(space_filling_curve)
add_subdirectory(conv_util)
add_subdirectory(reference_conv_fwd)
#add_subdirectory(magic_number_division)
#add_subdirectory(space_filling_curve)
#add_subdirectory(conv_util)
#add_subdirectory(reference_conv_fwd)
add_subdirectory(gemm)
add_subdirectory(gemm_split_k)
add_subdirectory(gemm_reduce)
add_subdirectory(batched_gemm)
add_subdirectory(batched_gemm_reduce)
add_subdirectory(grouped_gemm)
add_subdirectory(convnd_fwd)
add_subdirectory(reduce)
add_subdirectory(conv2d_bwd_weight)
#add_subdirectory(gemm_split_k)
#add_subdirectory(gemm_reduce)
#add_subdirectory(batched_gemm)
#add_subdirectory(batched_gemm_reduce)
#add_subdirectory(grouped_gemm)
#add_subdirectory(convnd_fwd)
#add_subdirectory(reduce)
#add_subdirectory(conv2d_bwd_weight)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment