[CLIENTS]fix cuda_runtime not found and assert len(MARKERS) >= len(y_keys[index]) (#535)

fix cuda_runtime not found and assert len(MARKERS) >= len(y_keys[index])
pull/536/head
engineer1109 2024-03-14 03:25:00 +08:00 committed by GitHub
parent b35737b066
commit e320c204b1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions

View File

@ -518,7 +518,7 @@ if(CLIENTS OR TESTS)
endif()
endif()
if(CUBLAS_FOUND)
set(REF_INCLUDES ${REF_INCLUDES} ${CUDA_INCLUDE_DIRS})
set(REF_INCLUDES ${REF_INCLUDES} ${CUDA_INCLUDE_DIRS} ${CUBLAS_INCLUDE_DIRS})
set(REF_LIBRARIES ${REF_LIBRARIES} ${CUDA_LIBRARIES} ${CUBLAS_LIBRARIES})
set(WRAPPERS ${WRAPPERS} test/wrapper_cuda.hpp test/wrapper_cublas.hpp)
if(MSVC)

View File

@ -18,7 +18,7 @@ REDISH = [c / 255.0 for c in [214, 117, 104]] # #d67568
PURPLISH = [c / 255.0 for c in [85, 0, 119]] # #550077
GREEN = [c / 255.0 for c in [144, 224, 98]] # #90e062
COLORS = [BLUEISH, REDISH, PURPLISH, GREEN]
MARKERS = ["o-", "x-", ".-"]
MARKERS = ["o-", "x-", ".-", "--"]
def plot_graphs(results, file_name, num_rows, num_cols,