Set CUDA arch

LLNL · Jul 15, 2024 · 1490b75 · 1490b75
1 parent 5f9dcc5
commit 1490b75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -41,7 +41,7 @@ FROM ghcr.io/llnl/radiuss:ubuntu-22.04-cuda-12-3 AS cuda
 ENV GTEST_COLOR=1
 COPY . /home/umpire/workspace
 WORKDIR /home/umpire/workspace/build
-RUN cmake -DUMPIRE_ENABLE_DEVELOPER_DEFAULTS=On -DCMAKE_CXX_COMPILER=g++ -DENABLE_CUDA=On .. && \
+RUN cmake -DUMPIRE_ENABLE_DEVELOPER_DEFAULTS=On -DCMAKE_CXX_COMPILER=g++ -DENABLE_CUDA=On -DCMAKE_CUDA_ARCHITECTURES=70 .. && \
     make -j 16
 
 # TODO: switch to ROCM 6