ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 21:21:22 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
dda3a0a10bb62a6d47e3559b89146a9d02361502
composable_kernel/include/ck/tensor_operation/gpu
History
Qianfeng dda3a0a10b Update to the Reduction API and instances (#476) 
* Simplify the macros for declaring and defining the add_device_reduce_instance_xxxx() instances

* Change the types of lengths and strides from std::vector to std::array for the reduction device interfaces

* Remove DeviceSoftmaxImpl's depending on DeviceReduceMultiblock

* Split the cpp and hpp files for reduction instances to enable more parallel compiling

* Remove the using of macros for declaring reduction instances and instance references

* Update to add_device_reduce_instance_xxxx templated functions

* Use ReduceOperation+InElementwiseOp+AccElementwiseOp to repace the ReduceOpId in defining add_reduce_instance_xxxx() templates

* Change return format
2022-10-25 09:39:11 -06:00
..
block
Add batched attention special kernel instances (#424)
2022-09-19 19:20:54 -05:00
device
Update to the Reduction API and instances (#476)
2022-10-25 09:39:11 -06:00
element
Group norm (#417)
2022-09-19 22:30:46 -05:00
grid
Revert "Fused elementwise layernorm (#468)" (#491)
2022-10-25 18:37:12 +08:00
thread
Add 'Permute' device op & example (#408)
2022-09-19 21:30:25 -05:00
warp
Hotfix LDS data hazard in fused attention (#360)
2022-08-15 12:04:20 -05:00