Skip to content

Commit

Permalink
Added command to test nvcc in build script.
Browse files Browse the repository at this point in the history
  • Loading branch information
luisenp committed Nov 7, 2022
1 parent f14d39b commit 6dd81e9
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions build_scripts/build_wheel.sh
Original file line number Diff line number Diff line change
Expand Up @@ -101,6 +101,8 @@ for PYTHON_VERSION in 3.9; do
# --- Install baspacho
RUN git clone https://github.com/facebookresearch/baspacho.git
WORKDIR baspacho
RUN nvcc cmake/detect_cuda.cu -o cuda_test
RUN ./cuda_test
# Note: to use static BLAS the option is really BLA_STATIC (https://cmake.org/cmake/help/latest/module/FindBLAS.html)
RUN /opt/cmake3.24/bin/cmake -S . -B build -DCMAKE_BUILD_TYPE=Release -DBLA_STATIC=ON \
Expand Down

0 comments on commit 6dd81e9

Please sign in to comment.