ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-12 01:10:17 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
5bf0475afdf33e823ed8c0247b3a8711326601d3
composable_kernel/include/ck/utility
History
guangzlu 4c4c7328a6 Add BF16 tests for batched_gemm_softmax_gemm_permute (#504) 
* fixed bug in softmax reference & add bf16 examples for batched_gemm_scale_softmax_gemm

* added bf16 tests for batched_gemm_softmax_gemm_permute

* changed format of device_batched_gemm_softmax_gemm_permute_xdl_cshuffle_bf16_bf16_bf16_bf16_gmk_gnk_gno_gmo_instance.cpp

* changed format device_batched_gemm_softmax_gemm_permute_xdl_cshuffle_bf16_bf16_bf16_bf16_gmk_gnk_gno_gmo_instance.cpp

* aligned annotations

* modified CMakeLists for examples

* add common example code of fp16/bf16 version for batched_gemm_scale_softmax_gemm_xdl

* use macro to control the instances

* added macro control into instances

* clang-format some files

* changed error tolerance for bf16

* changed index for 10_elementwise_normalization

* fixed xdlops code bug in amd_xdlops.hpp

Co-authored-by: Po Yen Chen <PoYen.Chen@amd.com>
2022-11-15 16:30:23 -06:00
..
amd_address_space.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
amd_buffer_addressing.hpp
embedding fuse layernorm (#405)
2022-09-09 10:41:15 -05:00
amd_inline_asm.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
amd_llvm_intrinsic.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
amd_xdlops.hpp
Add BF16 tests for batched_gemm_softmax_gemm_permute (#504)
2022-11-15 16:30:23 -06:00
array_multi_index.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
array.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
c_style_pointer_cast.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
common_header.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
container_element_picker.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
container_helper.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
data_type.hpp
Padding for attention: bmm+scale+softmax+bmm kernel (#385)
2022-08-30 11:01:37 -05:00
debug.hpp
Single-kernel GEMM + layernorm (#263)
2022-07-01 01:38:00 -05:00
dynamic_buffer.hpp
CGEMM examples bf16, fp32, int8 (#332)
2022-08-02 14:52:27 -05:00
enable_if.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
functional2.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
functional3.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
functional4.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
functional.hpp
Implement padding and sanity checks for fused GEMM+GEMM (#376)
2022-08-23 10:01:02 -05:00
generic_memory_space_atomic.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
get_id.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
ignore.hpp
Conv bwd data multiple d (#404)
2022-09-19 11:25:28 -05:00
inner_product.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
integral_constant.hpp
N-D Tensor Contraction example, instance, and client example (#270)
2022-07-07 14:31:11 -05:00
is_known_at_compile_time.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
magic_division.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
math_v2.hpp
int4 data type (#364)
2022-08-18 14:53:47 -05:00
math.hpp
Layernorm welford (#346)
2022-08-13 09:43:18 -05:00
multi_index.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
number.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
print.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
reduction_common.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
reduction_enums.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
reduction_functions_accumulate.hpp
Standalone sweep once softmax kernel w/ ckProfiler (#295)
2022-06-30 12:08:50 -05:00
reduction_operator.hpp
Gemm reduce examples int4/int8/fp32/bf16 (#368)
2022-08-30 11:38:26 -05:00
sequence_helper.hpp
N-D Tensor Contraction example, instance, and client example (#270)
2022-07-07 14:31:11 -05:00
sequence.hpp
N-D Tensor Contraction example, instance, and client example (#270)
2022-07-07 14:31:11 -05:00
span.hpp
Add 'Permute' device op & example (#408)
2022-09-19 21:30:25 -05:00
static_buffer.hpp
Fused GEMM+GEMM (#351)
2022-08-13 09:18:58 -05:00
statically_indexed_array_multi_index.hpp
Fused attention (#345)
2022-08-13 00:16:14 -05:00
statically_indexed_array.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
synchronization.hpp
Skip lds of b matrix (#326)
2022-08-13 01:35:49 -05:00
thread_group.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
transpose_vectors.hpp
work around inline asm potential hazard using intrinsic (#416)
2022-09-19 17:28:28 -05:00
tuple_helper.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
tuple.hpp
Clean up conv example, Instances, profiler and test (#324)
2022-07-29 18:19:25 -05:00
type.hpp
N-D Tensor Contraction example, instance, and client example (#270)
2022-07-07 14:31:11 -05:00