add --gpu-max-threads-per-block

8f2f74c3 · zhanggzh · d79540ef · 8f2f74c3
Commit 8f2f74c3 authored Jun 10, 2025 by zhanggzh
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

CMakeLists.txt CMakeLists.txt +3 -2

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -86,8 +86,9 @@ if(USE_CUDA)
    -DCUDA_HAS_FP16=1 \
    -D__CUDA_NO_HALF_OPERATORS__ \
    -D__CUDA_NO_HALF_CONVERSIONS__ \
-    -D__CUDA_NO_HALF2_OPERATORS__"
+    -D__CUDA_NO_HALF2_OPERATORS__ \
-  )
+    --gpu-max-threads-per-block=1024" 
+   )
 endif()
 include(cmake/TorchAudioHelper.cmake)