Install rocthrust-dev to provide Thrust headers for HIP compilation
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -12,7 +12,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
|||||||
git cmake cmake-curses-gui \
|
git cmake cmake-curses-gui \
|
||||||
libopenblas-dev wget \
|
libopenblas-dev wget \
|
||||||
python3-dev python3-pip \
|
python3-dev python3-pip \
|
||||||
hipblas-dev miopen-hip-dev \
|
hipblas-dev miopen-hip-dev rocthrust-dev \
|
||||||
&& rm -rf /var/lib/apt/lists/*
|
&& rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
# Build oneDNN as a static library (CPU fallback / conv ops)
|
# Build oneDNN as a static library (CPU fallback / conv ops)
|
||||||
|
|||||||
Reference in New Issue
Block a user