ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-03-29 19:47:39 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
fa998675fc6def2bc4aa5b9ccdfeefcef476a2b1
composable_kernel/client_example/12_elementwise_normalization
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
..
CMakeLists.txt
Add BF16 tests for batched_gemm_softmax_gemm_permute (#504)
2022-11-15 16:30:23 -06:00
elementwise_layernorm2d.cpp
Add BF16 tests for batched_gemm_softmax_gemm_permute (#504)
2022-11-15 16:30:23 -06:00