update rocm

mila-iqia · Oct 17, 2023 · b732226 · b732226
1 parent a595d0a
commit b732226
Show file tree

Hide file tree

Showing 9 changed files with 376 additions and 356 deletions.
diff --git a/.pin/constraints-rocm-torch.txt b/.pin/constraints-rocm-torch.txt
diff --git a/benchmarks/accelerate_opt/requirements.rocm.txt b/benchmarks/accelerate_opt/requirements.rocm.txt
@@ -2,13 +2,13 @@
 # This file is autogenerated by pip-compile with Python 3.9
 # by the following command:
 #
-#    pip-compile --output-file=benchmarks/accelerate_opt/requirements.rocm.txt --resolver=backtracking .pin/tmp-constraints-rocm-opt.txt benchmarks/accelerate_opt/requirements.in
+#    pip-compile --config=pyproject.toml --output-file=benchmarks/accelerate_opt/requirements.rocm.txt --resolver=backtracking .pin/tmp-constraints-rocm-opt.txt benchmarks/accelerate_opt/requirements.in
 #
---extra-index-url https://download.pytorch.org/whl/rocm5.4.2/
+--extra-index-url https://download.pytorch.org/whl/rocm5.6/
 
-accelerate==0.19.0
+accelerate==0.23.0
     # via -r benchmarks/accelerate_opt/requirements.in
-aiohttp==3.8.4
+aiohttp==3.8.6
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   datasets
@@ -17,75 +17,80 @@ aiosignal==1.3.1
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   aiohttp
+annotated-types==0.6.0
+    # via
+    #   -c .pin/../.pin/constraints-rocm-torch.txt
+    #   pydantic
 antlr4-python3-runtime==4.9.3
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   omegaconf
-asttokens==2.2.1
+asttokens==2.4.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   giving
-async-timeout==4.0.2
+async-timeout==4.0.3
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   aiohttp
 attrs==23.1.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   aiohttp
-certifi==2023.5.7
+certifi==2023.7.22
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   requests
-charset-normalizer==3.1.0
+charset-normalizer==3.3.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   aiohttp
     #   requests
-cmake==3.26.3
+cmake==3.27.7
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   pytorch-triton-rocm
 codefind==0.1.3
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   ptera
-datasets==2.12.0
+datasets==2.14.5
     # via
     #   -r benchmarks/accelerate_opt/requirements.in
     #   evaluate
 deepspeed==0.8.3
     # via -r benchmarks/accelerate_opt/requirements.in
-dill==0.3.6
+dill==0.3.7
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   datasets
     #   evaluate
     #   multiprocess
-evaluate==0.4.0
+evaluate==0.4.1
     # via -r benchmarks/accelerate_opt/requirements.in
 executing==1.2.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   varname
-filelock==3.12.0
+filelock==3.12.4
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   huggingface-hub
     #   pytorch-triton-rocm
     #   torch
     #   transformers
-frozenlist==1.3.3
+frozenlist==1.4.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   aiohttp
     #   aiosignal
-fsspec[http]==2023.5.0
+fsspec[http]==2023.6.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   datasets
     #   evaluate
     #   huggingface-hub
+    #   torch
 giving==0.4.2
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
@@ -95,11 +100,13 @@ hjson==3.1.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   deepspeed
-huggingface-hub==0.14.1
+huggingface-hub==0.17.3
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
+    #   accelerate
     #   datasets
     #   evaluate
+    #   tokenizers
     #   transformers
 idna==3.4
     # via
@@ -110,15 +117,15 @@ jinja2==3.1.2
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   torch
-lit==16.0.5
+lit==17.0.2
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   pytorch-triton-rocm
-markdown-it-py==2.2.0
+markdown-it-py==3.0.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   rich
-markupsafe==2.1.2
+markupsafe==2.1.3
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   jinja2
@@ -135,7 +142,7 @@ multidict==6.0.4
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   aiohttp
     #   yarl
-multiprocess==0.70.14
+multiprocess==0.70.15
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   datasets
@@ -144,11 +151,11 @@ networkx==3.1
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   torch
-ninja==1.11.1
+ninja==1.11.1.1
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   deepspeed
-numpy==1.24.3
+numpy==1.26.1
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   accelerate
@@ -167,7 +174,7 @@ ovld==0.3.2
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   voir
-packaging==23.1
+packaging==23.2
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   accelerate
@@ -176,16 +183,16 @@ packaging==23.1
     #   evaluate
     #   huggingface-hub
     #   transformers
-pandas==2.0.2
+pandas==2.1.1
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   datasets
     #   evaluate
-pillow==9.5.0
+pillow==10.1.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   torchvision
-psutil==5.9.5
+psutil==5.9.6
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   accelerate
@@ -198,15 +205,19 @@ py-cpuinfo==9.0.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   deepspeed
-pyarrow==12.0.0
+pyarrow==13.0.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   datasets
-pydantic==1.10.8
+pydantic==2.4.2
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   deepspeed
-pygments==2.15.1
+pydantic-core==2.10.1
+    # via
+    #   -c .pin/../.pin/constraints-rocm-torch.txt
+    #   pydantic
+pygments==2.16.1
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   rich
@@ -218,15 +229,15 @@ python-dateutil==2.8.2
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   pandas
-pytorch-triton-rocm==2.0.2
+pytorch-triton-rocm==2.1.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   torch
-pytz==2023.3
+pytz==2023.3.post1
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   pandas
-pyyaml==6.0
+pyyaml==6.0.1
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   accelerate
@@ -238,7 +249,7 @@ reactivex==4.0.4
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   giving
-regex==2023.5.5
+regex==2023.10.3
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   transformers
@@ -255,12 +266,15 @@ requests==2.31.0
 responses==0.18.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
-    #   datasets
     #   evaluate
-rich==13.3.5
+rich==13.6.0
     # via
     #   -r benchmarks/accelerate_opt/requirements.in
     #   voir
+safetensors==0.4.0
+    # via
+    #   -c .pin/../.pin/constraints-rocm-torch.txt
+    #   transformers
 six==1.16.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
@@ -270,44 +284,45 @@ sympy==1.12
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   torch
-tokenizers==0.13.3
+tokenizers==0.14.1
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   transformers
-torch==2.0.1+rocm5.4.2
+torch==2.1.0+rocm5.6
     # via
     #   -r benchmarks/accelerate_opt/requirements.in
     #   accelerate
     #   deepspeed
     #   pytorch-triton-rocm
     #   torchaudio
     #   torchvision
-torchaudio==2.0.2+rocm5.4.2
+torchaudio==2.1.0+rocm5.6
     # via -r benchmarks/accelerate_opt/requirements.in
-torchvision==0.15.2+rocm5.4.2
+torchvision==0.16.0+rocm5.6
     # via -r benchmarks/accelerate_opt/requirements.in
-tqdm==4.65.0
+tqdm==4.66.1
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   datasets
     #   deepspeed
     #   evaluate
     #   huggingface-hub
     #   transformers
-transformers==4.29.2
+transformers==4.34.0
     # via -r benchmarks/accelerate_opt/requirements.in
-typing-extensions==4.6.2
+typing-extensions==4.8.0
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   huggingface-hub
     #   pydantic
+    #   pydantic-core
     #   reactivex
     #   torch
 tzdata==2023.3
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   pandas
-urllib3==1.26.16
+urllib3==1.26.17
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   requests
@@ -318,7 +333,7 @@ varname==0.10.0
     #   giving
 voir==0.2.10
     # via -r benchmarks/accelerate_opt/requirements.in
-xxhash==3.2.0
+xxhash==3.4.1
     # via
     #   -c .pin/../.pin/constraints-rocm-torch.txt
     #   datasets