From 7513f3138f35e30fdf6dcd540cdadf5fa5d5ec40 Mon Sep 17 00:00:00 2001
From: Felix Kallenborn <kallenborn@uni-mainz.de>
Date: Mon, 2 Mar 2020 09:40:50 +0100
Subject: [PATCH] Add missing includes for caching_allocator and
 discard_output_iterator

---
 cub/cub.cuh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/cub/cub.cuh b/cub/cub.cuh
index 4b8986cebe..a71d78fe0d 100644
--- a/cub/cub.cuh
+++ b/cub/cub.cuh
@@ -83,11 +83,13 @@
 #include "iterator/cache_modified_output_iterator.cuh"
 #include "iterator/constant_input_iterator.cuh"
 #include "iterator/counting_input_iterator.cuh"
+#include "iterator/discard_output_iterator.cuh"
 #include "iterator/tex_obj_input_iterator.cuh"
 #include "iterator/tex_ref_input_iterator.cuh"
 #include "iterator/transform_input_iterator.cuh"
 
 // Util
+#include "util_allocator.cuh"
 #include "util_arch.cuh"
 #include "util_debug.cuh"
 #include "util_device.cuh"