X-Git-Url: https://git.kernel.dk/?p=fio.git;a=blobdiff_plain;f=fio.h;h=6b2b669d3ccc93ac4aafadb8b59da91c537d6102;hp=b67613e6d10bdcb4309e9df6c14f7dbae9cc2f1e;hb=4bd2c8b9251a2c88f44ad52168252ce2de660bf7;hpb=5e7468209ad6956c3d7393110c379dd7d316dd7e

diff --git a/fio.h b/fio.h
index b67613e6..6b2b669d 100644
--- a/fio.h
+++ b/fio.h
@@ -59,6 +59,10 @@
 #define MPOL_LOCAL MPOL_MAX
 #endif
 
+#ifdef CONFIG_CUDA
+#include <cuda.h>
+#endif
+
 /*
  * offset generator types
  */
@@ -408,6 +412,18 @@ struct thread_data {
 	struct steadystate_data ss;
 
 	char verror[FIO_VERROR_SIZE];
+
+#ifdef CONFIG_CUDA
+	/*
+	 * for GPU memory management
+	 */
+	int gpu_dev_cnt;
+	int gpu_dev_id;
+	CUdevice  cu_dev;
+	CUcontext cu_ctx;
+	CUdeviceptr dev_mem_ptr;
+#endif	
+
 };
 
 /*