mirror of
https://github.com/hanzo-ml/trainer.git
synced 2026-09-24 00:31:34 +00:00
Tariq HasanandGitHub
1f57df4484
* ci(boilerplate): enforce copyright headers on YAML files Signed-off-by: tariq-hasan <[email protected]> * ci(boilerplate): enforce year-less header for new files Signed-off-by: tariq-hasan <[email protected]> --------- Signed-off-by: tariq-hasan <[email protected]>
71 lines
2.2 KiB
YAML
71 lines
2.2 KiB
YAML
# Copyright The Kubeflow Authors.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
apiVersion: trainer.kubeflow.org/v1alpha1
|
|
kind: ClusterTrainingRuntime
|
|
metadata:
|
|
name: mlx-distributed
|
|
labels:
|
|
trainer.kubeflow.org/framework: mlx
|
|
spec:
|
|
mlPolicy:
|
|
numNodes: 1
|
|
mpi:
|
|
numProcPerNode: 1
|
|
mpiImplementation: OpenMPI
|
|
sshAuthMountPath: /home/mpiuser/.ssh
|
|
runLauncherAsNode: true
|
|
template:
|
|
spec:
|
|
network:
|
|
publishNotReadyAddresses: true
|
|
successPolicy:
|
|
operator: All
|
|
targetReplicatedJobs:
|
|
- launcher
|
|
replicatedJobs:
|
|
- name: launcher
|
|
template:
|
|
metadata:
|
|
labels:
|
|
trainer.kubeflow.org/trainjob-ancestor-step: trainer
|
|
spec:
|
|
template:
|
|
spec:
|
|
containers:
|
|
- name: node
|
|
image: ghcr.io/kubeflow/trainer/mlx-runtime
|
|
securityContext:
|
|
runAsUser: 1000
|
|
- name: node
|
|
template:
|
|
spec:
|
|
template:
|
|
spec:
|
|
containers:
|
|
- name: node
|
|
image: ghcr.io/kubeflow/trainer/mlx-runtime
|
|
securityContext:
|
|
runAsUser: 1000
|
|
command:
|
|
- /usr/sbin/sshd
|
|
args:
|
|
- -De
|
|
- -f
|
|
- /home/mpiuser/.sshd_config
|
|
readinessProbe:
|
|
tcpSocket:
|
|
port: 2222
|
|
initialDelaySeconds: 5
|