Pin update

mila-iqia · Feb 9, 2024 · b2c8c13 · b2c8c13
1 parent e75f56f
commit b2c8c13
Show file tree

Hide file tree

Showing 33 changed files with 3,221 additions and 302 deletions.
diff --git a/.pin/constraints-cuda-torch.txt b/.pin/constraints-cuda-torch.txt
diff --git a/.pin/constraints-rocm-torch.txt b/.pin/constraints-rocm-torch.txt
@@ -1,8 +1,8 @@
 #
-# This file is autogenerated by pip-compile with Python 3.11
+# This file is autogenerated by pip-compile with Python 3.9
 # by the following command:
 #
-#    pip-compile --config=pyproject.toml --output-file=.pin/constraints-rocm-torch.txt --resolver=backtracking .pin/tmp-constraints.txt benchmarks/accelerate_opt/requirements.in benchmarks/dlrm/requirements.in benchmarks/flops/requirements.in benchmarks/huggingface/requirements.in benchmarks/rwkv/requirements.in benchmarks/stargan/requirements.in benchmarks/super-slomo/requirements.in benchmarks/timm/requirements.in benchmarks/torchvision/requirements.in
+#    pip-compile --output-file=.pin/constraints-rocm-torch.txt .pin/tmp-constraints.txt benchmarks/accelerate_opt/requirements.in benchmarks/dlrm/requirements.in benchmarks/flops/requirements.in benchmarks/huggingface/requirements.in benchmarks/llama/requirements.in benchmarks/rwkv/requirements.in benchmarks/stargan/requirements.in benchmarks/super-slomo/requirements.in benchmarks/timm/requirements.in benchmarks/torchvision/requirements.in
 #
 --extra-index-url https://download.pytorch.org/whl/rocm5.6/
 
@@ -39,6 +39,7 @@ codefind==0.1.3
 datasets==2.14.6
     # via
     #   -r benchmarks/accelerate_opt/requirements.in
+    #   -r benchmarks/llama/requirements.in
     #   evaluate
 deepspeed==0.12.2
     # via
@@ -57,6 +58,8 @@ evaluate==0.4.1
     # via -r benchmarks/accelerate_opt/requirements.in
 executing==1.2.0
     # via varname
+fairscale==0.4.13
+    # via -r benchmarks/llama/requirements.in
 fbgemm-gpu==0.5.0
     # via torchrec
 filelock==3.13.1
@@ -66,6 +69,8 @@ filelock==3.13.1
     #   torch
     #   torchx
     #   transformers
+fire==0.5.0
+    # via -r benchmarks/llama/requirements.in
 frozenlist==1.4.0
     # via
     #   aiohttp
@@ -109,7 +114,9 @@ idna==3.4
     #   requests
     #   yarl
 importlib-metadata==6.8.0
-    # via torchx
+    # via
+    #   markdown
+    #   torchx
 jinja2==3.1.2
     # via torch
 joblib==1.3.2
@@ -152,10 +159,13 @@ numpy==1.26.1
     # via
     #   -r benchmarks/dlrm/requirements.in
     #   -r benchmarks/rwkv/requirements.in
+    #   -r benchmarks/stargan/requirements.in
+    #   -r benchmarks/super-slomo/requirements.in
     #   accelerate
     #   datasets
     #   deepspeed
     #   evaluate
+    #   fairscale
     #   fbgemm-gpu
     #   onnx
     #   opencv-python
@@ -284,9 +294,12 @@ scikit-learn==1.3.2
     # via -r benchmarks/dlrm/requirements.in
 scipy==1.11.3
     # via scikit-learn
+sentencepiece==0.1.99
+    # via -r benchmarks/llama/requirements.in
 six==1.16.0
     # via
     #   asttokens
+    #   fire
     #   python-dateutil
     #   tensorboard
 sympy==1.12
@@ -297,16 +310,27 @@ tensorboard==2.15.1
     # via -r benchmarks/dlrm/requirements.in
 tensorboard-data-server==0.7.2
     # via tensorboard
+termcolor==2.4.0
+    # via fire
 threadpoolctl==3.2.0
     # via scikit-learn
 tokenizers==0.14.1
     # via transformers
 torch==2.1.0+rocm5.6
     # via
     #   -r benchmarks/accelerate_opt/requirements.in
+    #   -r benchmarks/dlrm/requirements.in
+    #   -r benchmarks/flops/requirements.in
+    #   -r benchmarks/huggingface/requirements.in
+    #   -r benchmarks/llama/requirements.in
     #   -r benchmarks/rwkv/requirements.in
+    #   -r benchmarks/stargan/requirements.in
+    #   -r benchmarks/super-slomo/requirements.in
+    #   -r benchmarks/timm/requirements.in
+    #   -r benchmarks/torchvision/requirements.in
     #   accelerate
     #   deepspeed
+    #   fairscale
     #   pytorch-lightning
     #   pytorch-triton-rocm
     #   torchaudio
@@ -324,14 +348,20 @@ torchrec==0.5.0
 torchvision==0.16.0+rocm5.6
     # via
     #   -r benchmarks/accelerate_opt/requirements.in
+    #   -r benchmarks/flops/requirements.in
     #   -r benchmarks/stargan/requirements.in
+    #   -r benchmarks/super-slomo/requirements.in
+    #   -r benchmarks/timm/requirements.in
+    #   -r benchmarks/torchvision/requirements.in
 torchviz==0.0.2
     # via -r benchmarks/dlrm/requirements.in
 torchx==0.5.0
     # via -r benchmarks/dlrm/requirements.in
 tqdm==4.66.1
     # via
+    #   -r benchmarks/dlrm/requirements.in
     #   -r benchmarks/flops/requirements.in
+    #   -r benchmarks/super-slomo/requirements.in
     #   -r benchmarks/torchvision/requirements.in
     #   datasets
     #   deepspeed
@@ -344,6 +374,7 @@ transformers==4.35.0
     # via
     #   -r benchmarks/accelerate_opt/requirements.in
     #   -r benchmarks/huggingface/requirements.in
+    #   -r benchmarks/llama/requirements.in
 typing-extensions==4.8.0
     # via
     #   huggingface-hub
@@ -368,8 +399,17 @@ varname==0.10.0
     # via giving
 voir==0.2.11
     # via
+    #   -c .pin/../constraints/rocm.txt
     #   -r benchmarks/accelerate_opt/requirements.in
+    #   -r benchmarks/dlrm/requirements.in
+    #   -r benchmarks/flops/requirements.in
+    #   -r benchmarks/huggingface/requirements.in
+    #   -r benchmarks/llama/requirements.in
     #   -r benchmarks/rwkv/requirements.in
+    #   -r benchmarks/stargan/requirements.in
+    #   -r benchmarks/super-slomo/requirements.in
+    #   -r benchmarks/timm/requirements.in
+    #   -r benchmarks/torchvision/requirements.in
 websocket-client==1.6.4
     # via docker
 werkzeug==3.0.1