From c8071faa5d5aad11b8b3a061a5d147d07576b232 Mon Sep 17 00:00:00 2001
From: yewentao256 <zhyanwentao@126.com>
Date: Wed, 24 Sep 2025 20:18:17 +0000
Subject: [PATCH] fix compile error

Signed-off-by: yewentao256 <zhyanwentao@126.com>
---
 csrc/cub_helpers.h | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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