Added user mpirun hostfile slots to NnfDataMovement

Signed-off-by: Blake Devcich <[email protected]>
NearNodeFlash · Sep 8, 2023 · 690068d · 690068d
1 parent 5877923
commit 690068d
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 1 deletion.
diff --git a/api/v1alpha1/nnf_datamovement_types.go b/api/v1alpha1/nnf_datamovement_types.go
@@ -100,6 +100,14 @@ type NnfDataMovementConfig struct {
 	// Note: Enabling this option may degrade performance.
 	// +kubebuilder:default:=false
 	StoreStdout bool `json:"storeStdout,omitempty"`
+
+	// The number of slots specified in the MPI hostfile. A value of 0 disables the use of slots in
+	// the hostfile. Nil will defer to the value specified in the nnf-dm-config ConfigMap.
+	Slots *int `json:"slots,omitempty"`
+
+	// The number of max_slots specified in the MPI hostfile. A value of 0 disables the use of slots
+	// in the hostfile. Nil will defer to the value specified in the nnf-dm-config ConfigMap.
+	MaxSlots *int `json:"maxSlots,omitempty"`
 }
 
 // DataMovementCommandStatus defines the observed status of the underlying data movement

diff --git a/api/v1alpha1/zz_generated.deepcopy.go b/api/v1alpha1/zz_generated.deepcopy.go
diff --git a/config/crd/bases/nnf.cray.hpe.com_nnfdatamovements.yaml b/config/crd/bases/nnf.cray.hpe.com_nnfdatamovements.yaml
@@ -178,6 +178,16 @@ spec:
                       the output is always logged. Note: Enabling this option may
                       degrade performance.'
                     type: boolean
+                  maxSlots:
+                    description: The number of max_slots specified in the MPI hostfile.
+                      A value of 0 disables the use of slots in the hostfile. Nil
+                      will defer to the value specified in the nnf-dm-config ConfigMap.
+                    type: integer
+                  slots:
+                    description: The number of slots specified in the MPI hostfile.
+                      A value of 0 disables the use of slots in the hostfile. Nil
+                      will defer to the value specified in the nnf-dm-config ConfigMap.
+                    type: integer
                   storeStdout:
                     default: false
                     description: 'Similar to LogStdout, store the command''s stdout