diff --git a/cuda/XSbench_header.cuh b/cuda/XSbench_header.cuh
index e96146a..53dee7b 100644
--- a/cuda/XSbench_header.cuh
+++ b/cuda/XSbench_header.cuh
@@ -7,6 +7,7 @@
 #include<assert.h>
 #include<cuda.h>
 #include <thrust/reduce.h>
+#include <thrust/partition.h>
 #include<stdint.h>
 #include <chrono>