test-infra icon indicating copy to clipboard operation
test-infra copied to clipboard

[torch-infra] Support multiple build targets

Open q10 opened this issue 9 months ago • 3 comments

  • Some repos (e.g. FBGEMM) can build multiple targets (i.e. packages with different names and functionality altogether, such as FBGEMM-GPU vs FBGEMM-GPU-GenAI), but there is no generic mechanism inside Nova to specify which target to build other than CU_VERSION environment variable, which is intended for target variants. As such, BUILD_TARGET is provided as an environment variable to the Nova build scripts so that it can be leveraged for specifying a build target if needed

q10 avatar Mar 27 '25 19:03 q10

The latest updates on your projects. Learn more about Vercel for Git ↗︎

1 Skipped Deployment
Name Status Preview Updated (UTC)
torchci ⬜️ Ignored (Inspect) Mar 27, 2025 7:13pm

vercel[bot] avatar Mar 27 '25 19:03 vercel[bot]

HI @q10 looks good. Can you please provide a reference how you use this workflow ? and some testing plan ?

atalman avatar Mar 27 '25 19:03 atalman

HI @q10 looks good. Can you please provide a reference how you use this workflow ? and some testing plan ?

@atalman https://github.com/pytorch/FBGEMM/pull/3890 shows an example of how I plan to use the new workflow input argument. For testing plan, I think it would be to check that both fbgemm_gpu and fbgemm_gpu_genai are published to https://download.pytorch.org/whl/nightly/fbgemm-gpu/ after the PR lands. Since it is a feature addition and not modification, I don't think it will adversely affect existing workflows.

q10 avatar Mar 27 '25 19:03 q10