Skip to content

Commit ac51786

Browse files
dorde-anticumangyadav
authored andcommitted
Add requirements.txt file and modify Dockerfile (#1776)
* Add requirements.txt file and modify Dockerfile for installing hip-python
1 parent 594d89e commit ac51786

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

mlir/utils/jenkins/Dockerfile

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,10 @@ ENV LC_ALL en_US.UTF-8
7676

7777
# Install python dependencies for the scripts. ADD will check contents of a file for changes.
7878
# TODO: that should be done during the build as it will pull this repo anyway and will have latest version.
79-
ADD "https://raw.githubusercontent.com/google/llvm-premerge-checks/master/scripts/requirements.txt" requirements.txt
79+
ADD "https://raw.githubusercontent.com/google/llvm-premerge-checks/master/scripts/requirements.txt" llvm-requirements.txt
80+
RUN pip3 install -r llvm-requirements.txt
81+
82+
ADD "https://raw.githubusercontent.com/ROCm/rocMLIR/refs/heads/develop/mlir/requirements.txt" requirements.txt
8083
RUN pip3 install -r requirements.txt
8184

8285
# --------------------- Section 2: Rock dialect setups -----------------
@@ -90,6 +93,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
9093
&& apt upgrade -y \
9194
&& python3 -m pip install --upgrade pip \
9295
&& pip3 install pandas numpy scipy jinja2 tomli \
96+
&& pip3 install -r llvm-requirements.txt \
9397
&& pip3 install -r requirements.txt
9498
# ^ pip install pandas... is needed to output performance tests and regenerate test suites
9599
# Has a copy of the requirements.txt install bit for the new python version

requirements.txt

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
tomli==2.0.1
2+
numpy>=1.19.5, <=2.1.2
3+
ml_dtypes>=0.1.0, <=0.5.0 # provides several NumPy dtype extensions, including the bf16
4+
scipy
5+
hip-python --index-url https://test.pypi.org/simple

0 commit comments

Comments
 (0)