composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 19:28:33 +00:00

Files

Andriy Roshchenko 05b10e0e5a Add instances of grouped convolution 3d forward with a ConvScale element-wise op for bf8@bf8->fp8 (#1326 )

We are adding more instances of grouped convolution 3d forward with a ConvScale element-wise operation.
This commit handles bf8@bf8->fp8 data types combination.

* Included an example.
* Added instances.
* Added a client example.

---------

Co-authored-by: Rostyslav Geyyer <rosty.geyyer@amd.com>
Co-authored-by: Bartłomiej Kocot <barkocot@amd.com>

2024-06-21 19:02:57 -06:00

CMakeLists.txt

Add instances of grouped convolution 3d forward with a ConvScale element-wise op for bf8@bf8->fp8 (#1326 )

2024-06-21 19:02:57 -06:00

convnd_fwd_convscale_common.hpp

Add a scale op, related instances and examples (#1242 )

2024-06-04 19:28:15 -05:00

convnd_fwd_xdl_convscale_bf8.cpp

Add instances of grouped convolution 3d forward with a ConvScale element-wise op for bf8@bf8->fp8 (#1326 )

2024-06-21 19:02:57 -06:00

convnd_fwd_xdl_convscale_fp8_bf8.cpp

Add instances for grouped conv fwd 3d with ConvScale for fp8@bf8->fp8 (#1325 )