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
aab8c7cc
Commit
aab8c7cc
authored
Feb 23, 2024
by
Adam Osewski
Browse files
Compiler DL & XDL examples only when appropriate arch is available.
parent
c7a9ad98
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
37 additions
and
26 deletions
+37
-26
example/15_grouped_gemm/CMakeLists.txt
example/15_grouped_gemm/CMakeLists.txt
+36
-25
include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_multiple_d_splitk_xdl_cshuffle_tile_loop.hpp
...grouped_gemm_multiple_d_splitk_xdl_cshuffle_tile_loop.hpp
+1
-1
No files found.
example/15_grouped_gemm/CMakeLists.txt
View file @
aab8c7cc
add_custom_target
(
example_grouped_gemm_xdl
)
add_example_executable
(
example_grouped_gemm_xdl_fp32 grouped_gemm_xdl_fp32.cpp
)
add_example_dependencies
(
example_grouped_gemm_xdl example_grouped_gemm_xdl_fp32
)
add_example_executable
(
example_grouped_gemm_xdl_fp16 grouped_gemm_xdl_fp16.cpp
)
add_example_dependencies
(
example_grouped_gemm_xdl example_grouped_gemm_xdl_fp16
)
add_custom_target
(
example_grouped_gemm
)
if
(
DL_KERNELS
)
add_example_executable
(
example_grouped_gemm_multiple_d_dl_fp16 grouped_gemm_multiple_d_dl_fp16.cpp
)
add_example_dependencies
(
example_grouped_gemm_xdl example_grouped_gemm_multiple_d_dl_fp16
)
add_example_dependencies
(
example_grouped_gemm example_grouped_gemm_multiple_d_dl_fp16
)
endif
()
list
(
APPEND gpu_list gfx90a gfx940 gfx941 gfx942
)
set
(
target 0
)
foreach
(
gpu IN LISTS GPU_TARGETS
)
if
(
gpu IN_LIST gpu_list AND target EQUAL 0
)
add_example_executable
(
example_grouped_gemm_xdl_fp32 grouped_gemm_xdl_fp32.cpp
)
add_example_dependencies
(
example_grouped_gemm example_grouped_gemm_xdl_fp32
)
add_example_executable
(
example_grouped_gemm_xdl_
splitk_
fp16 grouped_gemm_xdl_
splitk_
fp16.cpp
)
add_example_dependencies
(
example_grouped_gemm
_xdl
example_grouped_gemm_xdl_
splitk_
fp16
)
add_example_executable
(
example_grouped_gemm_xdl_fp16 grouped_gemm_xdl_fp16.cpp
)
add_example_dependencies
(
example_grouped_gemm example_grouped_gemm_xdl_fp16
)
add_example_executable
(
example_grouped_gemm_xdl_
fixed_n
k_fp16 grouped_gemm_xdl_
fixed_n
k_fp16.cpp
)
add_example_dependencies
(
example_grouped_gemm
_xdl
example_grouped_gemm_xdl_
fixed_n
k_fp16
)
add_example_executable
(
example_grouped_gemm_xdl_
split
k_fp16 grouped_gemm_xdl_
split
k_fp16.cpp
)
add_example_dependencies
(
example_grouped_gemm example_grouped_gemm_xdl_
split
k_fp16
)
add_example_executable
(
example_grouped_gemm_xdl_fixed_nk_
bias_
fp16 grouped_gemm_xdl_fixed_nk_
bias_
fp16.cpp
)
add_example_dependencies
(
example_grouped_gemm
_xdl
example_grouped_gemm_xdl_fixed_nk_
bias_
fp16
)
add_example_executable
(
example_grouped_gemm_xdl_fixed_nk_fp16 grouped_gemm_xdl_fixed_nk_fp16.cpp
)
add_example_dependencies
(
example_grouped_gemm example_grouped_gemm_xdl_fixed_nk_fp16
)
add_example_executable
(
example_grouped_gemm_xdl_
b
f16 grouped_gemm_xdl_
b
f16.cpp
)
add_example_dependencies
(
example_grouped_gemm
_xdl
example_grouped_gemm_xdl_
b
f16
)
add_example_executable
(
example_grouped_gemm_xdl_f
ixed_nk_bias_fp
16 grouped_gemm_xdl_f
ixed_nk_bias_fp
16.cpp
)
add_example_dependencies
(
example_grouped_gemm example_grouped_gemm_xdl_f
ixed_nk_bias_fp
16
)
add_example_executable
(
example_grouped_gemm_xdl_
int8
grouped_gemm_xdl_
int8
.cpp
)
add_example_dependencies
(
example_grouped_gemm
_xdl
example_grouped_gemm_xdl_
int8
)
add_example_executable
(
example_grouped_gemm_xdl_
bf16
grouped_gemm_xdl_
bf16
.cpp
)
add_example_dependencies
(
example_grouped_gemm example_grouped_gemm_xdl_
bf16
)
add_example_executable
(
example_grouped_gemm_xdl_
fixed_nk_fp
8 grouped_gemm_xdl_
fixed_nk_fp
8.cpp
)
add
_dependencies
(
example_grouped_gemm
_xdl
example_grouped_gemm_xdl_
fixed_nk_fp
8
)
add_example_executable
(
example_grouped_gemm_xdl_
int
8 grouped_gemm_xdl_
int
8.cpp
)
add_example
_dependencies
(
example_grouped_gemm example_grouped_gemm_xdl_
int
8
)
add_example_executable
(
example_grouped_gemm_
multiple_d_splitk_xdl_fp16 grouped_gemm_multiple_d_splitk_xdl_fp16
.cpp
)
add_dependencies
(
example_grouped_gemm
_xdl
example_grouped_gemm_
multiple_d_splitk_xdl_fp16
)
add_example_executable
(
example_grouped_gemm_
xdl_fixed_nk_fp8 grouped_gemm_xdl_fixed_nk_fp8
.cpp
)
add_dependencies
(
example_grouped_gemm example_grouped_gemm_
xdl_fixed_nk_fp8
)
add_example_executable
(
example_grouped_gemm_multiple_d_splitk_xdl_fp16 grouped_gemm_multiple_d_splitk_xdl_fp16.cpp
)
add_dependencies
(
example_grouped_gemm example_grouped_gemm_multiple_d_splitk_xdl_fp16
)
if
(
USE_BITINT_EXTENSION_INT4
)
if
(
USE_BITINT_EXTENSION_INT4
)
add_example_executable
(
example_grouped_gemm_xdl_int4 grouped_gemm_xdl_int4.cpp
)
add_example_dependencies
(
example_grouped_gemm_xdl example_grouped_gemm_xdl_int4
)
endif
()
add_example_dependencies
(
example_grouped_gemm example_grouped_gemm_xdl_int4
)
endif
()
set
(
target 1
)
endif
()
endforeach
()
include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_multiple_d_splitk_xdl_cshuffle_tile_loop.hpp
View file @
aab8c7cc
...
...
@@ -72,7 +72,7 @@ __global__ void
const
CDEElementwiseOperation
cde_element_op
)
{
#if(!defined(__HIP_DEVICE_COMPILE__) || defined(__gfx908__) || defined(__gfx90a__) || \
defined(__gfx94
0__) || defined(__gfx941__) || defined(__gfx942
__))
defined(__gfx94__))
constexpr
index_t
shared_size
=
GridwiseGemm
::
GetSharedMemoryNumberOfByte
();
__shared__
uint8_t
p_shared
[
shared_size
];
...
...
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