Added comments on Dl op split_k>1 support

340cb150 · Rosty Geyyer · 9cdaae76 · 340cb150
Commit 340cb150 authored Feb 16, 2023 by Rosty Geyyer
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

example/20_grouped_conv_bwd_weight/run_grouped_conv_bwd_weight_example.inc ...d_conv_bwd_weight/run_grouped_conv_bwd_weight_example.inc +2 -0

No files found.
--- a/example/20_grouped_conv_bwd_weight/run_grouped_conv_bwd_weight_example.inc
+++ b/example/20_grouped_conv_bwd_weight/run_grouped_conv_bwd_weight_example.inc
@@ -16,6 +16,8 @@ bool run_grouped_conv_bwd_weight(const ExecutionConfig& config,
 {
    ck::index_t split_k;
    // Set split_k = 2 for xdl op, split_k = 1 for dl
+    // Dl op doesn't support split_k > 1
+    // TODO: Add Dl op split_k > 1 support
    if(!(ck::get_device_name() == "gfx906" || ck::get_device_name() == "gfx1030"))
    {
        split_k = 2;