composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 02:02:46 +00:00

Files

Andriy Roshchenko e05bfee7e5 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>

[ROCm/composable_kernel commit: 05b10e0e5a]

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

grouped_convnd_bwd_data_bilinear

Add bilinear conv fwd and bwd data instances (#1164 )

2024-02-13 11:49:05 +01:00

grouped_convnd_bwd_data_scale

Add conv fwd/bwd data scale instances, extend bilinear instances (#1178 )

2024-03-13 23:09:08 +01:00

grouped_convnd_bwd_weight_bilinear

Add grouped conv bwd weight multi d kernel (#1237 )

2024-04-18 23:35:04 +02:00

grouped_convnd_bwd_weight_scale

Add grouped conv bwd weight multi d kernel (#1237 )