Change CUDA DOT thread-blocks to 1024

This improves the performance on Ampere (A100) GPUs. Fixes #137.
2023-06-12 15:49:59 +01:00 · 2023-06-12 15:49:59 +01:00 · 092ee67764
commit 092ee67764
parent 893af9f5d0
1 changed files with 1 additions and 1 deletions
--- a/src/cuda/CUDAStream.h
+++ b/src/cuda/CUDAStream.h
@ -22,7 +22,7 @@
 #endif

 #define TBSIZE 1024
-#define DOT_NUM_BLOCKS 256
+#define DOT_NUM_BLOCKS 1024

 template <class T>
 class CUDAStream : public Stream<T>