[Build issue]: Building wheel for specific arch not supported without GPU but with all ROCm components installed

[caaatch22]

Problem Description

I want to build TE inside docker which has all rocm components installed, but the host machine does not have a GPU.
And I encountered the issue shown in the snapshot:

I think we can add a env variable to detect desired arch and by setting cmake_flags 'AMDGPU_TARGETS' can solve the problem.

Operating System

AlmaLinux

CPU

Intel(R) Xeon(R) Platinum 8269CY CPU @ 2.50GHz

GPU

MI308X

ROCm Version

ROCm 6.2.3

ROCm Component

No response

Steps to Reproduce

No response

(Optional for Linux users) Output of /opt/rocm/bin/rocminfo --support

No response

Additional Information

No response

[wenchenvincent]

@caaatch22 Do you have a reproducer?

[hliuca]

Use NVTE_ROCM_ARCH, such as,

export NVTE_ROCM_ARCH=gfx942