fix libraries
@@ -8,7 +8,7 @@ RUN apt-get update -y && \
python3-pip \
python3-wheel \
python3-setuptools \
- hipcub hipblas rocrand rocfft rocsolver rccl && \
+ hipblas hipsparse rocsparse rocrand rocthrust rocsolver rocfft hipcub rocprim rccl && \
rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*
RUN pip3 install --no-cache-dir -U install setuptools pip
