arm64 controller

[cyanidium]

Motivation

This allows control plane nodes running arm64/aarch64 CPUs to be able to manage clusters with AMD GPUs. Heterogeneous/multi-arch clusters do not need dedicated amd64/x86_64 control plane nodes just to control AMD GPU scheduling.

Technical Details

Builds the controller/manager container for both amd64 and arm64, in a single multi-platform image.

Not changed:

• Build host still has to be amd64/x86_64
• GPU node still has to be amd64/x86_64
• e2e still only tests amd64/x86_64

Resolves #331

Test Plan

Tested working with a cluster like this:

control-plane   arm64/aarch64   no_gpu
control-plane   arm64/aarch64   no_gpu
control-plane   arm64/aarch64   no_gpu
worker          amd64/x86_64    no_gpu
worker          amd64/x86_64    no_gpu
worker          amd64/x86_64    amd_gpu
worker          amd64/x86_64    amd_gpu
worker          amd64/x86_64    amd_gpu

Helm values:

node-feature-discovery.enabled=false
deviceConfig.spec.driver.enabled=false

Test Result

gpu-operator pod schedules correctly and runs on the arm64/aarch64 control plane nodes. workflow pod schedules and runs correctly on an amd64/x86_64 worker node. GPU resources are available to pods that request them.

Submission Checklist

• Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.

[yansun1996]

Hi @cyanidium , thanks for the PR, really appreciate the contribution, this commit needs to go through internal review and CI test on our side, will get back to you once we decide to move forward with this change.