diff --git a/csrc/cub_helpers.h b/csrc/cub_helpers.h
index 470a63a22cab0..18e4e343ad8b7 100644
--- a/csrc/cub_helpers.h
+++ b/csrc/cub_helpers.h
@@ -12,6 +12,7 @@ using CubMaxOp = cub::Max;
   #endif  // CUB_VERSION
 #else
   #include <hipcub/hipcub.hpp>
-using CubAddOp = cub::Sum;
-using CubMaxOp = cub::Max;
+namespace cub = hipcub;
+using CubAddOp = hipcub::Sum;
+using CubMaxOp = hipcub::Max;
 #endif  // USE_ROCM