Skip to content

Fix packaged cuDNN loading and prepare qualified v0.1.5 release - #11

Merged
tarsur909 merged 1 commit into
mainfrom
fix/packaged-cudnn-loader
Oct 4, 2026
Merged

tarsur909 merged 1 commit into
mainfrom
fix/packaged-cudnn-loader

Conversation

@tarsur909

Copy link
Copy Markdown
Collaborator

Transformer Engine could not find the cuDNN shared libraries installed by the NVIDIA wheel in a fresh GPU environment, so v0.1.4 qualification correctly blocked publication. Preload the matching CUDA-major distribution's cuDNN9 libraries before importing Transformer Engine, retain their handles, and handle dependency ordering with bounded retries. Doctor, model loading, ops and Userbuffers use the same loader.

The release now checks the installed portable package's native runtime import on CPU before allocating GPUs. Prepare v0.1.5 and document v0.1.4 as unpublished without modifying its tag.

Validation: patched production loader imports Transformer Engine and its ops in the isolated Modal release image with no PYTHONPATH or library-path workaround; four regression tests cover dependency ordering, repeat loading, failure/retry, CPU and missing-distribution behavior. Repository/configuration checks, lock validation, Ruff, actionlint and checkpoint resume pass. GPU doctor, training smoke and correctness qualification remain mandatory before publication.

@tarsur909
tarsur909 merged commit 315db71 into main Oct 4, 2026
5 checks passed
@tarsur909
tarsur909 deleted the fix/packaged-cudnn-loader branch October 4, 2026 10:10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant