add context.hip

2023-03-08 07:34:19 +01:00
parent 7a3a9b682a
commit bdcb036592
4 changed files with 11 additions and 5 deletions
--- a/src/common/device_helpers.hip.h
+++ b/src/common/device_helpers.hip.h
@@ -23,7 +23,7 @@
 #include <chrono>
 #include <cstddef>  // for size_t
 #include <hipcub/hipcub.hpp>
-#include <cub/util_allocator.cuh>
+#include <hipcub/util_allocator.hpp>
 #include <numeric>
 #include <sstream>
 #include <string>