From 86cb663040bf4720e7f85a630736504298b79c4d Mon Sep 17 00:00:00 2001
From: wangjialiang <jialiang.wang@xdxct.com>
Date: Wed, 13 Mar 2024 13:24:06 +0800
Subject: [PATCH] fix cuda_runtime not found and assert len(MARKERS) >=
 len(y_keys[index])

---
 CMakeLists.txt            | 2 +-
 scripts/benchmark/plot.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 3cacc545..6f699a04 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -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)
diff --git a/scripts/benchmark/plot.py b/scripts/benchmark/plot.py
index 2cf5d04d..f86eff87 100644
--- a/scripts/benchmark/plot.py
+++ b/scripts/benchmark/plot.py
@@ -18,7 +18,7 @@
 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,