Install rocthrust-dev to provide Thrust headers for HIP compilation

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Dockerfile

@@ -12,7 +12,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
         git cmake cmake-curses-gui \
         libopenblas-dev wget \
         python3-dev python3-pip \
-        hipblas-dev miopen-hip-dev \
+        hipblas-dev miopen-hip-dev rocthrust-dev \
     && rm -rf /var/lib/apt/lists/*
 
 # Build oneDNN as a static library (CPU fallback / conv ops)