-
Notifications
You must be signed in to change notification settings - Fork 60
Add AMD GPU test for ray clusters #670
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
PIP_TRUSTED_HOST: "` + GetPipTrustedHost() + `" | ||
ACCELERATOR: "` + accelerator + `" | ||
`, | ||
RuntimeEnvYAML: fmt.Sprintf(` |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It is also possible to generate the yaml from struct - i.e. using https://pkg.go.dev/sigs.k8s.io/yaml#section-readme.
Though, as this yaml handling is relevant just for this one place - it is ok to keep current implementation.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This looks good to me. Thanks @ChughShilpa
|
||
if accelerator.IsGpu() { | ||
cqSpec.ResourceGroups[0].Flavors[0].Resources[2].NominalQuota = resource.MustParse("1") | ||
if accelerator.ResourceLabel == "amd.com/gpu" { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Same here.
Please adjust https://github.com/project-codeflare/codeflare-operator/blob/main/.github/workflows/e2e_tests.yaml#L92 to skip both CPU and AMD tests. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
/lgtm
Thanks
[APPROVALNOTIFIER] This PR is APPROVED This pull-request has been approved by: jiripetrlik, sutaakar The full list of commands accepted by this bot can be found here. The pull request process is described here
Needs approval from an approver in each of these files:
Approvers can indicate their approval by writing |
a96625a
into
project-codeflare:main
Issue link
Closes RHOAIENG-11047
What changes have been made
numberofGpus
parameter from createClusterQueue, constructRayCluster and constructRayJob methodVerification steps
Manually run the tests on OCP with NVIDIA and AMD gpus
Checks