Skip to content

Commit

Permalink
Update pings
Browse files Browse the repository at this point in the history
  • Loading branch information
pierre.delaunay committed Sep 13, 2024
1 parent 0548deb commit f2c4e72
Show file tree
Hide file tree
Showing 23 changed files with 853 additions and 333 deletions.
202 changes: 182 additions & 20 deletions .pin/constraints-cuda-torch.txt
View file

Large diffs are not rendered by default.

6 changes: 3 additions & 3 deletions benchmarks/brax/requirements.cuda.txt
View file

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

10 changes: 5 additions & 5 deletions benchmarks/diffusion/requirements.cuda.txt
View file

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 3 additions & 3 deletions benchmarks/dinov2/requirements.cuda.txt
View file

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 3 additions & 3 deletions benchmarks/flops/requirements.cuda.txt
View file

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

115 changes: 86 additions & 29 deletions benchmarks/geo_gnn/requirements-pre.cuda.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,104 +2,161 @@
# This file is autogenerated by pip-compile with Python 3.10
# by the following command:
#
# pip-compile --output-file=benchmarks/geo_gnn/requirements-pre.cuda.txt .pin/tmp-constraints-cuda-geo_gnn.txt benchmarks/geo_gnn/requirements-pre.in
# pip-compile --output-file=benchmarks/geo_gnn/requirements-pre.cuda.txt .pin/tmp-constraints-cuda-dimenet.txt benchmarks/geo_gnn/requirements-pre.in
#
--extra-index-url https://pypi.ngc.nvidia.com
--extra-index-url https://download.pytorch.org/whl/cu121
--find-links https://data.pyg.org/whl/torch-2.3.0+cu121.html
--find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
--find-links https://data.pyg.org/whl/torch-2.4.0+cu121.html
--trusted-host pypi.ngc.nvidia.com

filelock==3.16.0
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# torch
# triton
fsspec==2024.9.0
fsspec==2024.6.1
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# torch
jax[cuda12]==0.4.31
# via
# -c .pin/../.pin/constraints-cuda-torch.txt
# -r .pin/../constraints/extra/torch.cuda.txt
jax-cuda12-pjrt==0.4.31
# via
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax-cuda12-plugin
jax-cuda12-plugin[with-cuda]==0.4.31
# via
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax
jaxlib==0.4.31
# via
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax
jinja2==3.1.4
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# torch
markupsafe==2.1.5
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# jinja2
ml-dtypes==0.4.0
# via
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax
# jaxlib
mpmath==1.3.0
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# sympy
networkx==3.3
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# torch
numpy==1.26.4
# via
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax
# jaxlib
# ml-dtypes
# opt-einsum
# scipy
# xformers
nvidia-cublas-cu12==12.1.3.1
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax-cuda12-plugin
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
# torch
nvidia-cuda-cupti-cu12==12.1.105
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax-cuda12-plugin
# torch
nvidia-cuda-nvcc-cu12==12.6.68
# via
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax-cuda12-plugin
nvidia-cuda-nvrtc-cu12==12.1.105
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# torch
nvidia-cuda-runtime-cu12==12.1.105
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax-cuda12-plugin
# torch
nvidia-cudnn-cu12==8.9.2.26
nvidia-cudnn-cu12==9.1.0.70
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax-cuda12-plugin
# torch
nvidia-cufft-cu12==11.0.2.54
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax-cuda12-plugin
# torch
nvidia-curand-cu12==10.3.2.106
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# torch
nvidia-cusolver-cu12==11.4.5.107
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax-cuda12-plugin
# torch
nvidia-cusparse-cu12==12.1.0.106
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax-cuda12-plugin
# nvidia-cusolver-cu12
# torch
nvidia-nccl-cu12==2.20.5
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax-cuda12-plugin
# torch
nvidia-nvjitlink-cu12==12.6.68
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax-cuda12-plugin
# nvidia-cusolver-cu12
# nvidia-cusparse-cu12
nvidia-nvtx-cu12==12.1.105
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# torch
opt-einsum==3.3.0
# via
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax
scipy==1.14.1
# via
# -c .pin/../.pin/constraints-cuda-torch.txt
# jax
# jaxlib
sympy==1.13.2
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# torch
torch==2.3.1+cu121
torch==2.4.0+cu121
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -r .pin/../constraints/extra/gnn.cuda.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# -r benchmarks/geo_gnn/requirements-pre.in
triton==2.3.1
# xformers
triton==3.0.0
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# torch
typing-extensions==4.12.2
# via
# -c .pin/../.pin/constraints-cuda-gnn.txt
# -c .pin/../.pin/constraints-cuda-torch.txt
# torch
xformers==0.0.27.post2
# via
# -c .pin/../.pin/constraints-cuda-torch.txt
# -r .pin/../constraints/extra/torch.cuda.txt
Loading

0 comments on commit f2c4e72

Please sign in to comment.