-
Notifications
You must be signed in to change notification settings - Fork 68
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
add nvidia driver for cuda arm wheel #5218
base: main
Are you sure you want to change the base?
Conversation
@tinglvv is attempting to deploy a commit to the Meta Open Source Team on Vercel. A member of the Team first needs to authorize it. |
@@ -8,6 +8,10 @@ inputs: | |||
required: false | |||
type: string | |||
default: "550.54.15" # https://docs.nvidia.com/datacenter/tesla/tesla-release-notes-550-54-15/index.html | |||
GPU_ARCH_TYPE: | |||
required: true | |||
type: string |
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.
can we set a default "x86" and only add if statement for sbsa
GPU_ARCH_TYPE: ${{ inputs.GPU_ARCH_TYPE }} | ||
${{ if eq(inputs.GPU_ARCH_TYPE , 'cuda-aarch64') }}: | ||
ARCH_TYPE: sbsa | ||
${{ else }}: |
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 else would be x86_64 by default, so we can remove this if you set it to default.
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.
Change works, just nitpick to set sth to default.
description: GPU Arch type #"cuda", "cuda-aarch64" | ||
required: false | ||
type: string | ||
ARCH_TYPE: |
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.
How is ARCH_TYPE used?
I think it is safe to assume setup-nvidia is only called for GPU instances, where if it is not cuda, it must be cuda-aarch64.
GPU_ARCH_TYPE is enough?
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
No description provided.