diff --git a/docker/common/fw_pyproject.toml b/docker/common/fw_pyproject.toml index aeb4ee5728..391acda765 100644 --- a/docker/common/fw_pyproject.toml +++ b/docker/common/fw_pyproject.toml @@ -69,7 +69,7 @@ override-dependencies = [ "packaging>=25.0", "nvidia-pytriton>=0.7.0", "flashinfer-python==0.6.18.post1", # Match flashinfer-cubin installed by MBridge - "nvidia-cudnn-frontend==1.29.0", # Match MBridge; override PyPI MCore's older dev pin + "nvidia-cudnn-frontend==1.31.0", # Match MBridge; override PyPI MCore's older dev pin "urllib3>=2.0.7", "sh==2.2.4", "jupyter-core==5.8.1", diff --git a/pyproject.toml b/pyproject.toml index 42b77b1a42..1836766642 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -142,7 +142,7 @@ override-dependencies = [ "langchain>=0.3.28", # To address CVE-2025-65106 "langchain-core>=0.3.80", # To address CVE-2025-65106 "onnx>=1.21.0; python_version < '3.13'", # onnx<1.21 has missing lower bound on ml-dtypes, uses float4_e2m1fn (requires ml-dtypes>=0.5.0) - "nvidia-cudnn-frontend==1.29.0", + "nvidia-cudnn-frontend==1.31.0", # Match MCore main: cuDNN Frontend 1.29 needs the newer TileLang/TVM FFI family. "tilelang==0.1.12", "apache-tvm-ffi==0.1.11", diff --git a/uv.lock b/uv.lock index 3c63c9f249..636a1fd19d 100644 --- a/uv.lock +++ b/uv.lock @@ -27,7 +27,7 @@ overrides = [ { name = "langchain", specifier = ">=0.3.28" }, { name = "langchain-core", specifier = ">=0.3.80" }, { name = "mlflow", specifier = ">=3.16.0" }, - { name = "nvidia-cudnn-frontend", specifier = "==1.29.0" }, + { name = "nvidia-cudnn-frontend", specifier = "==1.31.0" }, { name = "nvidia-cutlass-dsl", extras = ["cu13"], marker = "sys_platform == 'never'" }, { name = "nvidia-modelopt", specifier = "==0.46.0rc1", index = "https://pypi.nvidia.com/" }, { name = "onnx", marker = "python_full_version < '3.13'", specifier = ">=1.21.0" }, @@ -2936,16 +2936,16 @@ wheels = [ [[package]] name = "nvidia-cudnn-frontend" -version = "1.29.0" +version = "1.31.0" source = { registry = "https://pypi.nvidia.com/" } dependencies = [ { name = "apache-tvm-ffi" }, { name = "nvidia-cutlass-dsl", extra = ["cu13"], marker = "sys_platform == 'never'" }, ] wheels = [ - { url = "https://pypi.nvidia.com/nvidia-cudnn-frontend/nvidia_cudnn_frontend-1.29.0-cp312-cp312-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:a3fb92256890b3027e59bb25a468ad2b8a474d2910e394edcf0b1c1015463f78" }, - { url = "https://pypi.nvidia.com/nvidia-cudnn-frontend/nvidia_cudnn_frontend-1.29.0-cp312-cp312-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:596c9b7591457b4b49227fa2e22bd5960957a198e4336ce9ada8b5d2a1e4a5a1" }, - { url = "https://pypi.nvidia.com/nvidia-cudnn-frontend/nvidia_cudnn_frontend-1.29.0-cp312-cp312-win_amd64.whl", hash = "sha256:06e46b829c9f1c470e8aa78581a3af3a7041d869d7590a0e55aff6f2638be39e" }, + { url = "https://pypi.nvidia.com/nvidia-cudnn-frontend/nvidia_cudnn_frontend-1.31.0-cp312-cp312-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:47d1a5432f59a2a7d0f903727c0d244e1413a20686b98d8e3890919c5d82a508" }, + { url = "https://pypi.nvidia.com/nvidia-cudnn-frontend/nvidia_cudnn_frontend-1.31.0-cp312-cp312-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:3171f102254136418f24b4470dfa5419f34b03d9d792a4b773cf65c7a37e6bf8" }, + { url = "https://pypi.nvidia.com/nvidia-cudnn-frontend/nvidia_cudnn_frontend-1.31.0-cp312-cp312-win_amd64.whl", hash = "sha256:786c7dc590d436338c55b74e3204fc2adb4739acafb1a5e46ce032282e601e2e" }, ] [[package]]