File tree Expand file tree Collapse file tree 2 files changed +10
-1
lines changed Expand file tree Collapse file tree 2 files changed +10
-1
lines changed Original file line number Diff line number Diff line change @@ -76,7 +76,10 @@ ENV LC_ALL en_US.UTF-8
76
76
77
77
# Install python dependencies for the scripts. ADD will check contents of a file for changes.
78
78
# 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
80
83
RUN pip3 install -r requirements.txt
81
84
82
85
# --------------------- Section 2: Rock dialect setups -----------------
@@ -90,6 +93,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
90
93
&& apt upgrade -y \
91
94
&& python3 -m pip install --upgrade pip \
92
95
&& pip3 install pandas numpy scipy jinja2 tomli \
96
+ && pip3 install -r llvm-requirements.txt \
93
97
&& pip3 install -r requirements.txt
94
98
# ^ pip install pandas... is needed to output performance tests and regenerate test suites
95
99
# Has a copy of the requirements.txt install bit for the new python version
Original file line number Diff line number Diff line change
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
You can’t perform that action at this time.
0 commit comments