From 4d97d3a264301213448712f07e7860a174dff0e9 Mon Sep 17 00:00:00 2001 From: Waf Project Date: Thu, 24 Oct 2024 02:17:12 +0800 Subject: [PATCH] Improve cuda detection logs --- playground/cuda/cuda.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/playground/cuda/cuda.py b/playground/cuda/cuda.py index d4135eb1..aec93b95 100644 --- a/playground/cuda/cuda.py +++ b/playground/cuda/cuda.py @@ -58,6 +58,6 @@ def find_cuda_libs(self): pass # this should not raise any error - self.check_cxx(header='cuda.h', lib='cuda', libpath=_libpath, includes=_includes) - self.check_cxx(header='cuda.h', lib='cudart', libpath=_libpath, includes=_includes) + self.check_cxx(header='cuda.h', lib='cuda', libpath=_libpath, includes=_includes, msg='Checking for header cuda.h and library cuda') + self.check_cxx(header='cuda.h', lib='cudart', libpath=_libpath, includes=_includes, msg='Checking for header cuda.h and library cudart')