Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel_ROCM
Commits
9f1b4276
Commit
9f1b4276
authored
Apr 04, 2024
by
Jakub Piasecki
Browse files
resolved conflicts
parents
711857c4
c7010716
Changes
198
Hide whitespace changes
Inline
Side-by-side
Showing
18 changed files
with
61 additions
and
90 deletions
+61
-90
test/gemm_split_k/test_gemm_splitk_xdl.cpp
test/gemm_split_k/test_gemm_splitk_xdl.cpp
+0
-0
test/grouped_convnd_bwd_data/CMakeLists.txt
test/grouped_convnd_bwd_data/CMakeLists.txt
+12
-19
test/grouped_convnd_bwd_data/test_grouped_convnd_bwd_data_xdl_wmma.cpp
...convnd_bwd_data/test_grouped_convnd_bwd_data_xdl_wmma.cpp
+0
-0
test/grouped_convnd_bwd_weight/CMakeLists.txt
test/grouped_convnd_bwd_weight/CMakeLists.txt
+9
-17
test/grouped_convnd_bwd_weight/test_grouped_convnd_bwd_weight_xdl_wmma.cpp
...nd_bwd_weight/test_grouped_convnd_bwd_weight_xdl_wmma.cpp
+0
-0
test/grouped_convnd_fwd/CMakeLists.txt
test/grouped_convnd_fwd/CMakeLists.txt
+11
-5
test/grouped_convnd_fwd/test_grouped_convnd_fwd_multi_d_interface_compatibility_xdl_wmma.cpp
...d_convnd_fwd_multi_d_interface_compatibility_xdl_wmma.cpp
+0
-0
test/grouped_convnd_fwd/test_grouped_convnd_fwd_xdl_wmma.cpp
test/grouped_convnd_fwd/test_grouped_convnd_fwd_xdl_wmma.cpp
+0
-0
test/grouped_gemm/CMakeLists.txt
test/grouped_gemm/CMakeLists.txt
+13
-14
test/grouped_gemm/test_grouped_gemm_interface_xdl.cpp
test/grouped_gemm/test_grouped_gemm_interface_xdl.cpp
+0
-0
test/grouped_gemm/test_grouped_gemm_splitk_xdl.cpp
test/grouped_gemm/test_grouped_gemm_splitk_xdl.cpp
+0
-0
test/normalization_bwd_data/CMakeLists.txt
test/normalization_bwd_data/CMakeLists.txt
+4
-9
test/normalization_bwd_gamma_beta/CMakeLists.txt
test/normalization_bwd_gamma_beta/CMakeLists.txt
+4
-9
test/permute_scale/CMakeLists.txt
test/permute_scale/CMakeLists.txt
+2
-4
test/transpose/CMakeLists.txt
test/transpose/CMakeLists.txt
+4
-9
test/transpose/test_transpose_xdl.cpp
test/transpose/test_transpose_xdl.cpp
+0
-0
test/wrapper/CMakeLists.txt
test/wrapper/CMakeLists.txt
+2
-4
test/wrapper/test_wrapper_gemm_xdl.cpp
test/wrapper/test_wrapper_gemm_xdl.cpp
+0
-0
No files found.
test/gemm_split_k/test_gemm_splitk.cpp
→
test/gemm_split_k/test_gemm_splitk
_xdl
.cpp
View file @
9f1b4276
File moved
test/grouped_convnd_bwd_data/CMakeLists.txt
View file @
9f1b4276
list
(
APPEND gpu_list_xdl gfx908 gfx90a gfx940
)
list
(
APPEND gpu_list_wmma gfx1100 gfx1101 gfx1102 gfx1103
)
set
(
target 0
)
foreach
(
gpu IN LISTS GPU_TARGETS
)
if
(
gpu IN_LIST gpu_list_xdl AND target EQUAL 0
)
add_gtest_executable
(
test_grouped_convnd_bwd_data test_grouped_convnd_bwd_data.cpp
)
target_link_libraries
(
test_grouped_convnd_bwd_data PRIVATE utility device_grouped_conv2d_bwd_data_instance device_grouped_conv3d_bwd_data_instance
)
add_gtest_executable
(
test_grouped_convnd_bwd_data_interface test_grouped_convnd_bwd_data_interface_xdl.cpp
)
target_link_libraries
(
test_grouped_convnd_bwd_data_interface PRIVATE utility device_grouped_conv2d_bwd_data_instance
)
set
(
target 1
)
endif
()
if
(
gpu IN_LIST gpu_list_wmma AND target EQUAL 0
)
add_gtest_executable
(
test_grouped_convnd_bwd_data test_grouped_convnd_bwd_data.cpp
)
target_link_libraries
(
test_grouped_convnd_bwd_data PRIVATE utility device_grouped_conv2d_bwd_data_instance device_grouped_conv3d_bwd_data_instance
)
add_gtest_executable
(
test_grouped_convnd_bwd_data_interface test_grouped_convnd_bwd_data_interface_wmma.cpp
)
target_link_libraries
(
test_grouped_convnd_bwd_data_interface PRIVATE utility device_grouped_conv2d_bwd_data_instance
)
set
(
target 1
)
endif
()
endforeach
()
\ No newline at end of file
add_gtest_executable
(
test_grouped_convnd_bwd_data test_grouped_convnd_bwd_data_xdl_wmma.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_grouped_convnd_bwd_data PRIVATE utility device_grouped_conv2d_bwd_data_instance device_grouped_conv3d_bwd_data_instance
)
endif
()
add_gtest_executable
(
test_grouped_convnd_bwd_data_interface test_grouped_convnd_bwd_data_interface_xdl.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_grouped_convnd_bwd_data_interface PRIVATE utility device_grouped_conv2d_bwd_data_instance
)
endif
()
add_gtest_executable
(
test_grouped_convnd_bwd_data_interface test_grouped_convnd_bwd_data_interface_wmma.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_grouped_convnd_bwd_data_interface PRIVATE utility device_grouped_conv2d_bwd_data_instance
)
endif
()
test/grouped_convnd_bwd_data/test_grouped_convnd_bwd_data.cpp
→
test/grouped_convnd_bwd_data/test_grouped_convnd_bwd_data
_xdl_wmma
.cpp
View file @
9f1b4276
File moved
test/grouped_convnd_bwd_weight/CMakeLists.txt
View file @
9f1b4276
list
(
APPEND gpu_list_xdl gfx908 gfx90a gfx940 gfx941 gfx942
)
list
(
APPEND gpu_list_wmma gfx1100 gfx1101 gfx1102 gfx1103
)
set
(
target 0
)
foreach
(
gpu IN LISTS GPU_TARGETS
)
if
(
gpu IN_LIST gpu_list_xdl AND target EQUAL 0
)
add_gtest_executable
(
test_grouped_convnd_bwd_weight test_grouped_convnd_bwd_weight.cpp
)
add_gtest_executable
(
test_grouped_convnd_bwd_weight test_grouped_convnd_bwd_weight_xdl_wmma.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_grouped_convnd_bwd_weight PRIVATE utility device_grouped_conv1d_bwd_weight_instance device_grouped_conv2d_bwd_weight_instance device_grouped_conv3d_bwd_weight_instance
)
add_gtest_executable
(
test_grouped_convnd_bwd_weight_interface test_grouped_convnd_bwd_weight_interface_xdl.cpp
)
endif
()
add_gtest_executable
(
test_grouped_convnd_bwd_weight_interface test_grouped_convnd_bwd_weight_interface_xdl.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_grouped_convnd_bwd_weight_interface PRIVATE utility
)
set
(
target 1
)
endif
()
if
(
gpu IN_LIST gpu_list_wmma AND target EQUAL 0
)
add_gtest_executable
(
test_grouped_convnd_bwd_weight test_grouped_convnd_bwd_weight.cpp
)
target_link_libraries
(
test_grouped_convnd_bwd_weight PRIVATE utility device_grouped_conv1d_bwd_weight_instance device_grouped_conv2d_bwd_weight_instance device_grouped_conv3d_bwd_weight_instance
)
add_gtest_executable
(
test_grouped_convnd_bwd_weight_interface test_grouped_convnd_bwd_weight_interface_wmma.cpp
)
endif
()
add_gtest_executable
(
test_grouped_convnd_bwd_weight_interface test_grouped_convnd_bwd_weight_interface_wmma.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_grouped_convnd_bwd_weight_interface PRIVATE utility
)
set
(
target 1
)
endif
()
endforeach
()
\ No newline at end of file
endif
()
test/grouped_convnd_bwd_weight/test_grouped_convnd_bwd_weight.cpp
→
test/grouped_convnd_bwd_weight/test_grouped_convnd_bwd_weight
_xdl_wmma
.cpp
View file @
9f1b4276
File moved
test/grouped_convnd_fwd/CMakeLists.txt
View file @
9f1b4276
add_gtest_executable
(
test_grouped_convnd_fwd test_grouped_convnd_fwd.cpp
)
target_link_libraries
(
test_grouped_convnd_fwd PRIVATE utility device_grouped_conv1d_fwd_instance device_grouped_conv2d_fwd_instance device_grouped_conv3d_fwd_instance
)
add_gtest_executable
(
test_grouped_convnd_fwd test_grouped_convnd_fwd_xdl_wmma.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_grouped_convnd_fwd PRIVATE utility device_grouped_conv1d_fwd_instance device_grouped_conv2d_fwd_instance device_grouped_conv3d_fwd_instance
)
endif
()
add_gtest_executable
(
test_grouped_convnd_fwd_multi_ab_interface test_grouped_convnd_fwd_multi_ab_interface.cpp
)
target_link_libraries
(
test_grouped_convnd_fwd_multi_ab_interface PRIVATE utility
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_grouped_convnd_fwd_multi_ab_interface PRIVATE utility
)
endif
()
add_gtest_executable
(
test_grouped_convnd_fwd_multi_d_interface_compatibility test_grouped_convnd_fwd_multi_d_interface_compatibility.cpp
)
target_link_libraries
(
test_grouped_convnd_fwd_multi_d_interface_compatibility PRIVATE utility device_grouped_conv3d_fwd_instance
)
add_gtest_executable
(
test_grouped_convnd_fwd_multi_d_interface_compatibility test_grouped_convnd_fwd_multi_d_interface_compatibility_xdl_wmma.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_grouped_convnd_fwd_multi_d_interface_compatibility PRIVATE utility device_grouped_conv3d_fwd_instance
)
endif
()
test/grouped_convnd_fwd/test_grouped_convnd_fwd_multi_d_interface_compatibility.cpp
→
test/grouped_convnd_fwd/test_grouped_convnd_fwd_multi_d_interface_compatibility
_xdl_wmma
.cpp
View file @
9f1b4276
File moved
test/grouped_convnd_fwd/test_grouped_convnd_fwd.cpp
→
test/grouped_convnd_fwd/test_grouped_convnd_fwd
_xdl_wmma
.cpp
View file @
9f1b4276
File moved
test/grouped_gemm/CMakeLists.txt
View file @
9f1b4276
list
(
APPEND gpu_list gfx908 gfx90a gfx940 gfx941 gfx942
)
set
(
target 0
)
foreach
(
gpu IN LISTS GPU_TARGETS
)
if
(
gpu IN_LIST gpu_list AND target EQUAL 0
)
add_custom_target
(
test_grouped_gemm
)
add_gtest_executable
(
test_grouped_gemm_splitk test_grouped_gemm_splitk.cpp
)
add_gtest_executable
(
test_grouped_gemm_interface test_grouped_gemm_interface.cpp
)
target_link_libraries
(
test_grouped_gemm_splitk PRIVATE utility device_grouped_gemm_instance
)
target_link_libraries
(
test_grouped_gemm_interface PRIVATE utility device_grouped_gemm_instance
)
add_dependencies
(
test_grouped_gemm test_grouped_gemm_splitk test_grouped_gemm_interface
)
set
(
target 1
)
endif
()
endforeach
()
add_custom_target
(
test_grouped_gemm
)
add_gtest_executable
(
test_grouped_gemm_splitk test_grouped_gemm_splitk_xdl.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_grouped_gemm_splitk PRIVATE utility device_grouped_gemm_instance
)
add_dependencies
(
test_grouped_gemm test_grouped_gemm_splitk
)
endif
()
add_gtest_executable
(
test_grouped_gemm_interface test_grouped_gemm_interface_xdl.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_grouped_gemm_interface PRIVATE utility device_grouped_gemm_instance
)
add_dependencies
(
test_grouped_gemm test_grouped_gemm_interface
)
endif
()
test/grouped_gemm/test_grouped_gemm_interface.cpp
→
test/grouped_gemm/test_grouped_gemm_interface
_xdl
.cpp
View file @
9f1b4276
File moved
test/grouped_gemm/test_grouped_gemm_splitk.cpp
→
test/grouped_gemm/test_grouped_gemm_splitk
_xdl
.cpp
View file @
9f1b4276
File moved
test/normalization_bwd_data/CMakeLists.txt
View file @
9f1b4276
add_custom_target
(
test_normalization_bwd_data
)
add_gtest_executable
(
test_layernorm2d_bwd_data_fp32 test_layernorm2d_bwd_data_fp32.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_layernorm2d_bwd_data_fp32 PRIVATE utility device_normalization_bwd_data_instance
)
add_dependencies
(
test_normalization_bwd_data test_layernorm2d_bwd_data_fp32
)
endif
()
target_link_libraries
(
test_layernorm2d_bwd_data_fp32 PRIVATE utility device_normalization_bwd_data_instance
)
add_dependencies
(
test_normalization_bwd_data test_layernorm2d_bwd_data_fp32
)
add_gtest_executable
(
test_groupnorm_bwd_data_fp32 test_groupnorm_bwd_data_fp32.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_groupnorm_bwd_data_fp32 PRIVATE utility device_normalization_bwd_data_instance
)
add_dependencies
(
test_normalization_bwd_data test_groupnorm_bwd_data_fp32
)
endif
()
target_link_libraries
(
test_groupnorm_bwd_data_fp32 PRIVATE utility device_normalization_bwd_data_instance
)
add_dependencies
(
test_normalization_bwd_data test_groupnorm_bwd_data_fp32
)
test/normalization_bwd_gamma_beta/CMakeLists.txt
View file @
9f1b4276
add_custom_target
(
test_normalization_bwd_gamma_beta
)
add_gtest_executable
(
test_layernorm2d_bwd_gamma_beta_fp32 test_layernorm2d_bwd_gamma_beta_fp32.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_layernorm2d_bwd_gamma_beta_fp32 PRIVATE utility device_normalization_bwd_gamma_beta_instance
)
add_dependencies
(
test_normalization_bwd_gamma_beta test_layernorm2d_bwd_gamma_beta_fp32
)
endif
()
target_link_libraries
(
test_layernorm2d_bwd_gamma_beta_fp32 PRIVATE utility device_normalization_bwd_gamma_beta_instance
)
add_dependencies
(
test_normalization_bwd_gamma_beta test_layernorm2d_bwd_gamma_beta_fp32
)
add_gtest_executable
(
test_groupnorm_bwd_gamma_beta_fp32 test_groupnorm_bwd_gamma_beta_fp32.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_groupnorm_bwd_gamma_beta_fp32 PRIVATE utility device_normalization_bwd_gamma_beta_instance
)
add_dependencies
(
test_normalization_bwd_gamma_beta test_groupnorm_bwd_gamma_beta_fp32
)
endif
()
target_link_libraries
(
test_groupnorm_bwd_gamma_beta_fp32 PRIVATE utility device_normalization_bwd_gamma_beta_instance
)
add_dependencies
(
test_normalization_bwd_gamma_beta test_groupnorm_bwd_gamma_beta_fp32
)
test/permute_scale/CMakeLists.txt
View file @
9f1b4276
add_custom_target
(
test_permute
)
add_gtest_executable
(
test_permute_scale test_permute_scale.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_permute_scale PRIVATE utility device_permute_scale_instance
)
add_dependencies
(
test_permute test_permute_scale
)
endif
()
target_link_libraries
(
test_permute_scale PRIVATE utility device_permute_scale_instance
)
add_dependencies
(
test_permute test_permute_scale
)
test/transpose/CMakeLists.txt
View file @
9f1b4276
list
(
APPEND gpu_list gfx908 gfx90a gfx940 gfx941 gfx942
)
set
(
target 0
)
foreach
(
gpu IN LISTS GPU_TARGETS
)
if
(
gpu IN_LIST gpu_list AND target EQUAL 0
)
add_gtest_executable
(
test_transpose test_transpose.cpp
)
target_link_libraries
(
test_transpose PRIVATE utility device_transpose_instance
)
set
(
target 1
)
endif
()
endforeach
()
add_gtest_executable
(
test_transpose test_transpose_xdl.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_transpose PRIVATE utility device_transpose_instance
)
endif
()
test/transpose/test_transpose.cpp
→
test/transpose/test_transpose
_xdl
.cpp
View file @
9f1b4276
File moved
test/wrapper/CMakeLists.txt
View file @
9f1b4276
...
...
@@ -12,10 +12,8 @@ add_dependencies(test_wrapper test_wrapper_copy)
add_gtest_executable
(
test_wrapper_partition test_wrapper_partition.cpp
)
target_link_libraries
(
test_wrapper_partition PRIVATE utility
)
add_dependencies
(
test_wrapper test_wrapper_partition
)
if
(
GPU_TARGETS MATCHES
"gfx908"
OR GPU_TARGETS MATCHES
"gfx90a"
OR
GPU_TARGETS MATCHES
"gfx940"
OR GPU_TARGETS MATCHES
"gfx941"
OR
GPU_TARGETS MATCHES
"gfx942"
)
add_gtest_executable
(
test_wrapper_gemm test_wrapper_gemm.cpp
)
add_gtest_executable
(
test_wrapper_gemm test_wrapper_gemm_xdl.cpp
)
if
(
result EQUAL 0
)
target_link_libraries
(
test_wrapper_gemm PRIVATE utility
)
add_dependencies
(
test_wrapper test_wrapper_gemm
)
endif
()
test/wrapper/test_wrapper_gemm.cpp
→
test/wrapper/test_wrapper_gemm
_xdl
.cpp
View file @
9f1b4276
File moved
Prev
1
…
6
7
8
9
10
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment
