Add ROCm

Dockerfile

@@ -18,27 +18,12 @@ RUN apt-get -y install \
         build-essential \
         wget
 
-# required for DCTFilter
-RUN cd /tmp \
-    && wget http://www.fftw.org/fftw-3.3.8.tar.gz \
-    && tar xvf fftw-3.3.8.tar.gz \
-    && cd fftw-3.3.8 \
-    && ./configure \
-        --enable-shared \
-        --enable-threads \
-        --enable-openmp \
-        --enable-sse2 \
-        --enable-avx2 \
-        --enable-float \
-    && make -j 4 install \
-    && cd .. \
-    && rm -rf fftw-3.3.8
-
 # Vapoursynth plugin build dependencies
 RUN apt-get -y install \
         autoconf \
         cargo \
         git \
+        libfftw3-dev \
         libplacebo-dev \
         libtool \
         meson \
@@ -109,12 +94,21 @@ RUN apt-get -y install \
         x264 \
         x265
 
+# ROCm (AMD Radeon OpenCL)
+# The gdb packages is necessary for opencl to work (why?)
+COPY rocm.gpg /etc/apt/trusted.gpg.d/
+RUN echo 'deb http://repo.radeon.com/rocm/apt/debian/ xenial main' > /etc/apt/sources.list.d/rocm.list \
+    && apt-get update \
+    && apt-get -y install rocm-opencl rocm-gdb \
+    && rm /etc/OpenCL/vendors/amdocl64* \
+    && echo /opt/rocm-*/opencl/lib/libamdocl64.so > /etc/OpenCL/vendors/amdocl64.icd
+
 # models are at wrong location (whyever the package installs it there)
 RUN cp -r /usr/share/vapoursynth-waifu2x-models/ /usr/lib/x86_64-linux-gnu/vapoursynth/models/
 
 RUN pip3 --no-cache-dir install \
         tqdm \
-        yuuno 'prompt-toolkit<2.1.0,>=2.0.0' \
+        yuuno 'pygments>=2.4.1' \
     && yuuno jupyter install
 
 ENV PYTHONPATH=/data/functions/:/usr/local/lib/vapoursynth-scripts/