ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 18:17:44 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
8d4fd1233db5306a6bb7e894913f23d6b4f43918
composable_kernel/include/ck/utility
History
carlushuang 8d4fd1233d embedding fuse layernorm (#405) 
* add gridwise/device sparse embedding

* update code

* update code

* remove useless makefile

* code fix

* workable

* work properly

* emb add

* add more instance

* format

* remove useless code

* fix format

* fix clang-tidy

* clean

* fix a compile error

Co-authored-by: Chao Liu <chao.liu2@amd.com>
Co-authored-by: Chao Liu <lc.roy86@gmail.com>

[ROCm/composable_kernel commit: efd1d25733]
2022-09-09 10:41:15 -05: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 license in file (#303)
2022-06-24 23:32:43 -05: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
add license in file (#303)
2022-06-24 23:32:43 -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
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
add license in file (#303)
2022-06-24 23:32:43 -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