ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 06:49:15 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
4a2a56c22f75263d70c710950ab7313f072a2523
composable_kernel/example/21_gemm_layernorm
History
Po Yen Chen 4a2a56c22f Rangify constructor of HostTensorDescriptor & Tensor<> (#445) 
* Rangify STL algorithms

This commit adapts rangified std::copy(), std::fill() & std::transform()

* Rangify check_err()

By rangifying check_err(), we can not only compare values between
std::vector<>s, but also compare any ranges which have same value
type.

* Allow constructing Tensor<> like a HostTensorDescriptor

* Simplify Tensor<> object construction logics

* Remove more unnecessary 'HostTensorDescriptor' objects

* Re-format example code

* Re-write more HostTensorDescriptor ctor call
2022-11-11 11:36:01 -06:00
..
CMakeLists.txt
Single-kernel GEMM + layernorm (#263)
2022-07-01 01:38:00 -05:00
gemm_bias_relu_add_layernorm_xdl_fp16.cpp
Rangify constructor of HostTensorDescriptor & Tensor<> (#445)
2022-11-11 11:36:01 -06:00
gemm_layernorm_xdl_fp16.cpp
Rangify constructor of HostTensorDescriptor & Tensor<> (#445)
2022-11-11 11:36:01 -06:00
gemm_xdl_layernorm_single_kernel_fp16.cpp
Rangify constructor of HostTensorDescriptor & Tensor<> (#445)
2022-11-11 11:36:01 -06:00