ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-16 02:54:21 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
4faf3ab5871573fb4a18519c71731069b69ac125
composable_kernel/library/src/tensor_operation_instance/gpu
History
Bartłomiej Kocot 930195c384 Add dynamic elementwise op (#1426) 
* Add dynamic elementwise op

Co-authored-by: ThruptiRajLakshmanaGowda <thruptiraj.lakshmanagowda@amd.com>

* CI issues fix

* Custom parameter value for dynamic functions - Comments addressed

---------

Co-authored-by: ThruptiRajLakshmanaGowda <thruptiraj.lakshmanagowda@amd.com>
Co-authored-by: ThruptiRajLakshmanaGowda <tlakshma@amd.com>

[ROCm/composable_kernel commit: 31bf253aeb]
2024-10-26 15:22:37 +02:00
..
avg_pool2d_bwd
Pool2d max/avg kernel in the BWD version (#1494)
2024-09-12 11:47:52 +02:00
avg_pool3d_bwd
…
batched_gemm
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
batched_gemm_add_relu_gemm_add
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
batched_gemm_bias_permute
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
batched_gemm_gemm
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
batched_gemm_multi_d
…
batched_gemm_reduce
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
batched_gemm_softmax_gemm
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
batched_gemm_softmax_gemm_permute
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
batchnorm
Refactor elementwise kernels (#1222)
2024-04-19 13:31:17 +02:00
column_to_image
…
contraction_bilinear
Extend support for contraction 6D (#1207)
2024-04-09 23:46:21 +02:00
contraction_scale
Extend support for contraction 6D (#1207)
2024-04-09 23:46:21 +02:00
conv1d_bwd_data
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
conv2d_bwd_data
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
conv2d_fwd
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
conv2d_fwd_bias_relu
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
conv2d_fwd_bias_relu_add
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
conv3d_bwd_data
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
elementwise
Refactor elementwise kernels (#1222)
2024-04-19 13:31:17 +02:00
elementwise_normalization
…
gemm
Re-enable fp8 types for all architectures. (#1470)
2024-08-16 16:07:52 -06:00
gemm_ab_scale
[GEMM] gemm_universal related optimization (#1453)
2024-08-14 10:42:30 +08:00
gemm_add
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_add_add_fastgelu
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_add_fastgelu
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_add_multiply
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_add_relu
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_add_relu_add_layernorm
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_add_silu
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_bias_add_reduce
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_bilinear
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_fastgelu
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_multi_abd
Split the gemm_multi_abd instances. (#1306)
2024-05-23 09:17:02 -07:00
gemm_multiply_add
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_multiply_multiply
add int8 gemm multiply multiply a8w8 (#1591)
2024-10-26 16:39:34 +08:00
gemm_reduce
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_splitk
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_streamk
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_universal
Ck profiler instance support (#1575)
2024-10-21 22:47:48 +08:00
gemm_universal_reduce
Universal gemm splitk using reduce (with multi-d) (#1341)
2024-07-19 22:01:22 +08:00
gemm_universal_streamk
Universal streamk with atomics (#1360)
2024-07-05 21:40:30 -07:00
grouped_conv1d_bwd_weight
Enable grouped conv bwd wei bf16 NGCHW (#1589)
2024-10-22 16:18:28 +02:00
grouped_conv1d_fwd
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_conv2d_bwd_data
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_conv2d_bwd_weight
[POST MERGE PR] Enable grouped conv bwd wei bf16 NGCHW (#1594)
2024-10-23 12:02:33 +02:00
grouped_conv2d_fwd
Add support for NGCHW in grouped conv fwd (#1499)
2024-09-20 10:45:46 +02:00
grouped_conv2d_fwd_dynamic_op
Add dynamic elementwise op (#1426)
2024-10-26 15:22:37 +02:00
grouped_conv3d_bwd_data
Re-enable fp8 types for all architectures. (#1470)
2024-08-16 16:07:52 -06:00
grouped_conv3d_bwd_data_bilinear
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_conv3d_bwd_data_scale
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_conv3d_bwd_weight
[POST MERGE PR] Enable grouped conv bwd wei bf16 NGCHW (#1594)
2024-10-23 12:02:33 +02:00
grouped_conv3d_bwd_weight_bilinear
Re-enable fp8 types for all architectures. (#1470)
2024-08-16 16:07:52 -06:00
grouped_conv3d_bwd_weight_scale
Re-enable fp8 types for all architectures. (#1470)
2024-08-16 16:07:52 -06:00
grouped_conv3d_fwd
Revert "Revert "Revert Revert Support access per groups and filter2x3 in grouped conv fwd (#1382) (#1406) (#1415)" (#1455)" (#1490)
2024-09-02 10:39:49 +02:00
grouped_conv3d_fwd_bilinear
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_conv3d_fwd_convinvscale
Add a convinvscale op, related instances and examples (#1307)
2024-06-10 14:48:49 -05:00
grouped_conv3d_fwd_convscale
Adding Instances and Examples for FP8-based Scaled Convolution and AMAX Reduction. (#1473)
2024-08-21 15:22:41 -07:00
grouped_conv3d_fwd_convscale_add
Adding more instances of grouped convolution 3d forward for FP8 with ConvScale+Bias element-wise operation. (#1412)
2024-07-24 15:49:55 -05:00
grouped_conv3d_fwd_convscale_relu
Adding Instances and Examples for FP8-based Scaled Convolution and AMAX Reduction. (#1473)
2024-08-21 15:22:41 -07:00
grouped_conv3d_fwd_dynamic_op
Add dynamic elementwise op (#1426)
2024-10-26 15:22:37 +02:00
grouped_conv3d_fwd_scale
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_conv3d_fwd_scaleadd_ab
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_conv3d_fwd_scaleadd_scaleadd_relu
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_gemm
Add bf16 and bf16@int8 mk_nk_mn instances for grouped gemm two stage (#1228)
2024-04-19 13:16:10 +02:00
grouped_gemm_bias
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_gemm_fastgelu
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_gemm_fixed_nk
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_gemm_fixed_nk_multi_abd
bf16A_Int8B with fastgelu/bias (#1264)
2024-04-26 07:26:30 -05:00
grouped_gemm_tile_loop
Switch to universal gemm in grouped gemm tile loop (#1335)
2024-06-18 09:01:49 -05:00
image_to_column
…
max_pool_bwd
This commit contains implementation of max pool2d for f8 type (#1506)
2024-09-16 10:15:06 +02:00
mha
Add additional instances to device_mha_instance (#1522)
2024-09-24 10:15:30 -06:00
normalization_bwd_data
…
normalization_bwd_gamma_beta
…
normalization_fwd
…
permute_scale
Adding Instances and Examples for FP8-based Scaled Convolution with ReLU Activation and AMAX Reduction. (#1469)
2024-08-20 10:30:56 -05:00
pool2d_fwd
Add pool2d int8 and fp8 instances (#1508)
2024-09-13 10:18:21 -07:00
pool3d_fwd
Extend pool3d fwd avg, max operations by f8_t, int8_t types (#1483)
2024-09-17 15:57:10 +02:00
quantization
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
reduce
Adding Instances and Examples for FP8-based Scaled Convolution with ReLU Activation and AMAX Reduction. (#1469)
2024-08-20 10:30:56 -05:00
softmax
…
transpose
…
CMakeLists.txt
Fix build logic using GRU_ARCHS. (#1536)
2024-10-07 08:18:23 -07:00