From 1490b75a856f8331e4d9e44f903c8e183a9b0160 Mon Sep 17 00:00:00 2001 From: David Beckingsale Date: Mon, 15 Jul 2024 09:51:39 -0700 Subject: [PATCH] Set CUDA arch --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 4295f7487..2e77b6af3 100644 --- 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