Skip to content

Commit

Permalink
minor: add todo marking temporary fix for cuda-enabled ubuntu
Browse files Browse the repository at this point in the history
  • Loading branch information
Peiyang-Song committed Aug 14, 2024
1 parent b7988f9 commit f0e416d
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions lakefile.lean
Original file line number Diff line number Diff line change
Expand Up @@ -211,6 +211,12 @@ def getCt2CmakeFlags : IO (Array String) := do
| .macos => flags := flags ++ #["-DWITH_ACCELERATE=ON", "-DWITH_OPENBLAS=OFF"]
| .linux => flags := flags ++ #["-DWITH_ACCELERATE=OFF", "-DWITH_OPENBLAS=ON", "-DOPENBLAS_INCLUDE_DIR=../../OpenBLAS", "-DOPENBLAS_LIBRARY=../../OpenBLAS/libopenblas.so"]

-- [TODO] Temporary fix: Do not use CUDA even if it is available.
-- if ← useCUDA then
-- flags := flags ++ #["-DWITH_CUDA=ON", "-DWITH_CUDNN=ON"]
-- else
-- flags := flags ++ #["-DWITH_CUDA=OFF", "-DWITH_CUDNN=OFF"]

return flags


Expand Down

0 comments on commit f0e416d

Please sign in to comment.