From 934f488ced5a62db9d78d8c3fc0d16c8dad5aa23 Mon Sep 17 00:00:00 2001
From: Luna Nova <git@lunnova.dev>
Date: Fri, 18 Apr 2025 07:15:43 -0700
Subject: [PATCH] rocmPackages.composable_kernel: make more parts big-parallel

All parts except pool are too slow without the higher core limit on big-parallel builders.

Infra channel discussion: https://matrix.to/#/!RROtHmAaQIkiJzJZZE:nixos.org/
---
 .../rocm-modules/6/composable_kernel/default.nix          | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/pkgs/development/rocm-modules/6/composable_kernel/default.nix b/pkgs/development/rocm-modules/6/composable_kernel/default.nix
index 8671893f3c44..dc8d6c3506f9 100644
--- a/pkgs/development/rocm-modules/6/composable_kernel/default.nix
+++ b/pkgs/development/rocm-modules/6/composable_kernel/default.nix
@@ -36,6 +36,7 @@ let
         "device_grouped_conv2d_fwd_instance"
         "device_grouped_conv2d_fwd_dynamic_op_instance"
       ];
+      requiredSystemFeatures = [ "big-parallel" ];
     };
     grouped_conv_bwd_3d = {
       targets = [
@@ -46,6 +47,7 @@ let
         "device_grouped_conv3d_bwd_weight_bilinear_instance"
         "device_grouped_conv3d_bwd_weight_scale_instance"
       ];
+      requiredSystemFeatures = [ "big-parallel" ];
     };
     grouped_conv_fwd_3d = {
       targets = [
@@ -60,6 +62,7 @@ let
         "device_grouped_conv3d_fwd_scaleadd_ab_instance"
         "device_grouped_conv3d_fwd_scaleadd_scaleadd_relu_instance"
       ];
+      requiredSystemFeatures = [ "big-parallel" ];
     };
     batched_gemm = {
       targets = [
@@ -77,6 +80,7 @@ let
         "device_grouped_gemm_fixed_nk_multi_abd_instance"
         "device_grouped_gemm_tile_loop_instance"
       ];
+      requiredSystemFeatures = [ "big-parallel" ];
     };
     gemm_universal = {
       targets = [
@@ -108,6 +112,7 @@ let
         "device_gemm_splitk_instance"
         "device_gemm_streamk_instance"
       ];
+      requiredSystemFeatures = [ "big-parallel" ];
     };
     conv = {
       targets = [
@@ -118,6 +123,7 @@ let
         "device_conv2d_fwd_bias_relu_add_instance"
         "device_conv3d_bwd_data_instance"
       ];
+      requiredSystemFeatures = [ "big-parallel" ];
     };
     pool = {
       targets = [
@@ -139,6 +145,7 @@ let
         "device_normalization_bwd_gamma_beta_instance"
         "device_normalization_fwd_instance"
       ];
+      requiredSystemFeatures = [ "big-parallel" ];
     };
     other2 = {
       targets = [
@@ -150,6 +157,7 @@ let
         "device_softmax_instance"
         "device_transpose_instance"
       ];
+      requiredSystemFeatures = [ "big-parallel" ];
     };
   };
   tensorOpBuilder =