natas_hw
/
jidt
mirror of https://github.com/jlizier/jidt
0
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: remove custom CUB include and disable CUB version check 

- Removed `-I./cub` from the `INCLUDES` as CUB is now included in the CUDA Toolkit.
- Added `-DTHRUST_IGNORE_CUB_VERSION_CHECK` flag to bypass the Thrust and CUB version check.
This commit is contained in:
wenzexu 2024-10-22 20:26:34 +11:00
parent 5d4419d734
commit 82f17abe31
No known key found for this signature in database
GPG Key ID: 34E97B62F134210F
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cuda/Makefile
View File

@ -12,7 +12,7 @@
NVCC ?= nvcc
GCC ?= g++
CCFLAGS ?= -O3 -Wall -Wextra -Wno-unused-parameter
NVCCFLAGS ?= -O3 -D_FORCE_INLINES -Xcompiler -Wall
NVCCFLAGS ?= -O3 -D_FORCE_INLINES -Xcompiler -Wall -DTHRUST_IGNORE_CUB_VERSION_CHECK
BIN =../bin/cuda
UNITBIN =../unittests/bin/cuda
@ -73,7 +73,7 @@ ifdef CUSTOMFLAGS
endif
# Common includes and paths for CUDA. This assumes the CUDA toolkit is in PATH
INCLUDES := -I. -I./cub -I${JNI_INCLUDE} -I${JNI_PLATFORM_INCLUDE}
INCLUDES := -I. -I${JNI_INCLUDE} -I${JNI_PLATFORM_INCLUDE}
NVCCLDFLAGS := -L. -L$(BIN) -lcuda -lcudart
.PHONY: all clean test