ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-11 17:00:18 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
6951858221e03e321cb38c55b3a5ef3c68b5b79d
composable_kernel/include/ck/utility
History
Bartlomiej Wroblewski 6951858221 Implement direct loads split-K GEMM kernel (#1137) 
* WIP: Implement direct loads split-K GEMM kernel

* Clean the review

---------

Co-authored-by: Adam Osewski <19374865+aosewski@users.noreply.github.com>
Co-authored-by: Bartłomiej Kocot <barkocot@amd.com>
2024-02-07 01:08:34 +01:00
..
amd_address_space.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
amd_buffer_addressing.hpp
Add support for double buffering in direct load GEMM kernel (#1052)
2023-12-03 23:08:47 +01:00
amd_gemm_dpp.hpp
Redesign the DPP8 GEMM kernel to use warp-wise component (#863)
2023-09-06 11:44:09 -05:00
amd_inline_asm.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
amd_lds.hpp
Implement direct loads split-K GEMM kernel (#1137)
2024-02-07 01:08:34 +01:00
amd_wave_read_first_lane.hpp
Fix incomplete object size (=4n + 3) support of amd_wave_read_first_lane() (#738)
2023-06-12 08:36:40 -05:00
amd_wmma.hpp
Add support for more Navi2x and Navi3x models. (#1152)
2024-02-02 11:35:26 -08:00
amd_xdlops.hpp
Add support for more Navi2x and Navi3x models. (#1152)
2024-02-02 11:35:26 -08:00
array_multi_index.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
array.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
c_style_pointer_cast.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
common_header.hpp
FP8 enablement - add a pseudorandom number generator, add conversion methods (#708)
2023-06-19 11:20:35 -05:00
container_element_picker.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
container_helper.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
data_type.hpp
Fixing most of the cppcheck errors. (#1142)
2024-01-24 13:47:48 -08:00
debug.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
dynamic_buffer.hpp
Add basic support for direct loads from global to LDS (#999)
2023-11-25 13:35:22 +01:00
enable_if.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
f8_utils.hpp
Fix bf8 conversion issues (#1003)
2023-10-20 08:00:45 -05:00
functional2.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
functional3.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
functional4.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
functional.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
generic_memory_space_atomic.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
get_id.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
get_shift.hpp
Batchnorm splitk single kernel (#771)
2023-07-06 10:58:55 -05:00
ignore.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
inner_product_dpp8.hpp
Redesign the DPP8 GEMM kernel to use warp-wise component (#863)
2023-09-06 11:44:09 -05:00
inner_product.hpp
Fix the DL kernel issues on Navi3x. (#998)
2023-10-19 09:34:39 -07:00
integral_constant.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
is_detected.hpp
add vector_type support into thread_copy_v3r1 (#969)
2023-10-13 15:11:43 -05:00
is_known_at_compile_time.hpp
Add optimized copy to ck wrapper (#1126)
2024-01-19 11:29:00 +01:00
loop_scheduler.hpp
Redesign the DPP8 GEMM kernel to use warp-wise component (#863)
2023-09-06 11:44:09 -05:00
magic_division.hpp
initial stream-k implementation with example (#699)
2023-07-26 14:18:15 -05:00
math_v2.hpp
Extend available elementwise operations with conv examples (#995)
2023-10-19 17:23:19 +02:00
math.hpp
Extend available elementwise operations with conv examples (#995)
2023-10-19 17:23:19 +02:00
multi_index.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
number.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
random_gen.hpp
FP8 enablement - add a pseudorandom number generator, add conversion methods (#708)
2023-06-19 11:20:35 -05:00
reduction_common.hpp
Batchnorm splitk single kernel (#771)
2023-07-06 10:58:55 -05:00
reduction_enums.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
reduction_functions_accumulate.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
reduction_operator.hpp
MaxPool & AvgPool bwd instances, test, ckProfiler, client example (#861)
2023-08-31 21:01:50 +08:00
sequence_helper.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
sequence.hpp
Revert "Grouped Gemm with looping over the tiles. (#788)" (#982)
2023-10-11 14:27:29 -05:00
span.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
static_buffer.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
statically_indexed_array_multi_index.hpp
Extend available elementwise operations with conv examples (#995)
2023-10-19 17:23:19 +02:00
statically_indexed_array.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
synchronization.hpp
Add basic support for direct loads from global to LDS (#999)
2023-11-25 13:35:22 +01:00
thread_group.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
transpose_vectors.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
tuple_helper.hpp
Add tensor partition and generic copy for ck wrapper (#1108)
2024-01-03 01:10:57 +01:00
tuple.hpp
Add multiple A/B support (#906)
2023-09-26 21:16:23 -05:00
type_convert.hpp
Add support for more Navi2x and Navi3x models. (#1152)
2024-02-02 11:35:26 -08:00
type.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
workgroup_barrier.hpp
initial stream-k implementation with example (#699)
2023-07-26 14:18:15 -05:00
workgroup_synchronization.hpp
Batchnorm splitk single kernel (#771)
2023-07-06 10:58:55 -05:00