Skip to content

Commit 90e2353

Browse files
committed
Clean code: Limit CUB headers to device_reduce to reduce compile time
1 parent 2f7b90c commit 90e2353

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/preconditioner.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ limitations under the License.
2121
#include <math.h>
2222
#include <cuda_runtime.h>
2323
#include <cublas_v2.h>
24-
#include <cub/cub.cuh>
24+
#include <cub/device/device_reduce.cuh>
2525

2626
#define SCALING_EPSILON 1e-12
2727

0 commit comments

Comments
 (0)