Add missing includes for caching_allocator and discard_output_iterator

NVlabs · Mar 30, 2020 · 7513f31 · 7513f31
1 parent 7f0768f
commit 7513f31
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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"