Skip to content

Fix setting Python variables in TensorRT easyblock#4046

Merged
boegel merged 4 commits intoeasybuilders:developfrom
Flamefire:20260115180310_new_pr_pythonpackage
Feb 16, 2026
Merged

Fix setting Python variables in TensorRT easyblock#4046
boegel merged 4 commits intoeasybuilders:developfrom
Flamefire:20260115180310_new_pr_pythonpackage

Conversation

@Flamefire
Copy link
Copy Markdown
Contributor

@Flamefire Flamefire commented Jan 15, 2026

(created using eb --new-pr)

For installing Python packages set_py_env_vars needs to be called which PythonPackage does in its configure_step

TensorRT skips most steps which means installing the extension/wheels fails if the user environment has unexpected settings.

We can further reduce duplication and chances for mistakes by using compose_install_command passing the wheel as the location.

Closes #4050
which is included here

@Flamefire
Copy link
Copy Markdown
Contributor Author

Test report by @Flamefire

Overview of tested easyconfigs (in order)

  • SUCCESS TensorRT-8.6.1-gfbf-2022b-CUDA-12.0.0.eb

Build succeeded for 1 out of 1 (total: 4 mins 13 secs) (1 easyconfigs in total)
n1405.barnard.hpc.tu-dresden.de - Linux RHEL 9.6, x86_64, Intel(R) Xeon(R) Platinum 8470 (sapphirerapids), Python 3.9.21
See https://gist.github.com/Flamefire/2ca3c0e01d230a4b8d4aa5fa0e089db1 for a full test report.

@Thyre Thyre added the bug fix label Jan 19, 2026
@Flamefire
Copy link
Copy Markdown
Contributor Author

Flamefire commented Jan 22, 2026

--sanity-check-only Test report by @Flamefire

Overview of tested easyconfigs (in order)

  • SUCCESS TensorRT-8.6.1-gfbf-2022b-CUDA-12.0.0.eb

Build succeeded for 1 out of 1 (total: 42 secs) (1 easyconfigs in total)
c128 - Linux Rocky Linux 9.6, x86_64, AMD EPYC 9334 32-Core Processor (zen4), 4 x NVIDIA NVIDIA H100, 580.65.06, Python 3.9.21
See https://gist.github.com/Flamefire/de8860ec52c4c0a3aa80129425634ac0 for a full test report.

This now accepts an `install_src` parameter to choose which package to
install.
This can be used to e.g. install wheels without duplicating the logic
used to determine parameters such as `--no-build-isolation`.
@Flamefire
Copy link
Copy Markdown
Contributor Author

Test report by @Flamefire

Overview of tested easyconfigs (in order)

  • SUCCESS TensorRT-8.6.1-gfbf-2022b-CUDA-12.0.0.eb

Build succeeded for 1 out of 1 (total: 2 mins 49 secs) (1 easyconfigs in total)
c128 - Linux Rocky Linux 9.6, x86_64, AMD EPYC 9334 32-Core Processor (zen4), 4 x NVIDIA NVIDIA H100, 580.65.06, Python 3.9.21
See https://gist.github.com/Flamefire/0a355b609a4c94e37167a37afdd35c84 for a full test report.

@Flamefire Flamefire force-pushed the 20260115180310_new_pr_pythonpackage branch from 3b59317 to 6a4dd61 Compare January 22, 2026 10:01
@Flamefire
Copy link
Copy Markdown
Contributor Author

I factored out the changes to PythonPackage to #4050 but kept them included here so they can be tested in conjunction with the TensorRT change and individually.

@boegel boegel added this to the next release (5.2.1?) milestone Jan 28, 2026
@boegel
Copy link
Copy Markdown
Member

boegel commented Feb 12, 2026

Test report by @boegel

Overview of tested easyconfigs (in order)

  • SUCCESS TensorRT-8.6.1-gfbf-2023a-CUDA-12.1.1.eb

  • SUCCESS TensorRT-10.0.1-gfbf-2023a-CUDA-12.1.1.eb

  • SUCCESS TensorRT-10.4.0-gfbf-2023b-CUDA-12.4.0.eb

  • SUCCESS TensorRT-10.7.0-gfbf-2024a-CUDA-12.6.0.eb

Build succeeded for 4 out of 4 (total: 6 mins 8 secs) (4 easyconfigs in total)
node3902.accelgor.os - Linux RHEL 9.6, x86_64, AMD EPYC 7413 24-Core Processor (zen3), 1 x NVIDIA NVIDIA A100-SXM4-80GB, 590.48.01, Python 3.9.21
See https://gist.github.com/boegel/ba8da63f30af091945540554ff803ea9 for a full test report.

@boegel
Copy link
Copy Markdown
Member

boegel commented Feb 12, 2026

Test report by @boegel

Overview of tested easyconfigs (in order)

  • SUCCESS TensorRT-10.7.0-gfbf-2024a-CUDA-12.6.0.eb

Build succeeded for 1 out of 1 (total: 1 min 42 secs) (1 easyconfigs in total)
node3902.accelgor.os - Linux RHEL 9.6, x86_64, AMD EPYC 7413 24-Core Processor (zen3), 1 x NVIDIA NVIDIA A100-SXM4-80GB, 590.48.01, Python 3.9.21
See https://gist.github.com/boegel/7aefa05d69779b2900563cee40041905 for a full test report.

edit: this was mainly to test easybuilders/easybuild-framework#5088 before merging it

Copy link
Copy Markdown
Member

@boegel boegel left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm

@boegel boegel merged commit 27af3bc into easybuilders:develop Feb 16, 2026
39 of 42 checks passed
@Flamefire Flamefire deleted the 20260115180310_new_pr_pythonpackage branch February 16, 2026 08:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants