yifeifang commented on code in PR #209: URL: https://github.com/apache/tvm-ffi/pull/209#discussion_r2479633522
########## examples/quickstart/raw_compile.sh: ########## @@ -59,3 +59,25 @@ g++ -fvisibility=hidden -O3 \ build/load_cpp # [load_cpp.end] fi + +# Example 4. Load and run `add_one_cuda.so` in C++ +# Before run this example, make sure you have a CUDA-capable GPU and the CUDA toolkit installed. +# See CONTRIBUTING.md to use a pre-built Docker image with CUDA support. + +# if [ -f "$BUILD_DIR/add_one_cuda.so" ] && command -v nvcc >/dev/null 2>&1; then +# # [load_cuda.begin] +# g++ -fvisibility=hidden -O3 \ +# load/load_cuda.cc \ +# $(tvm-ffi-config --cxxflags) \ +# $(tvm-ffi-config --ldflags) \ +# $(tvm-ffi-config --libs) \ +# -I/usr/local/cuda/include \ +# -L/usr/local/cuda/lib64 \ +# -lcudart \ +# -Wl,-rpath,$(tvm-ffi-config --libdir) \ +# -Wl,-rpath,/usr/local/cuda/lib64 \ Review Comment: I would keep it the original form as CUDA_HOME is not part of the official CTK installation (Unlike JAVA_HOME) while `/usr/local/cuda` is always used for a standard CTK installation -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
