From c7b1c0cf8ba95a3796943e2c5d2a8fbf3f37b8ab Mon Sep 17 00:00:00 2001
From: yewentao256 <zhyanwentao@126.com>
Date: Mon, 22 Sep 2025 15:23:46 -0700
Subject: [PATCH] fix cub_helpers

Signed-off-by: yewentao256 <zhyanwentao@126.com>
---
 csrc/quantization/w8a8/fp8/common.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/csrc/quantization/w8a8/fp8/common.cu b/csrc/quantization/w8a8/fp8/common.cu
index 622d559d839c5..ef57902504911 100644
--- a/csrc/quantization/w8a8/fp8/common.cu
+++ b/csrc/quantization/w8a8/fp8/common.cu
@@ -1,7 +1,7 @@
 #include "common.cuh"
 #include "dispatch_utils.h"
+#include "../../cub_helpers.h"
 #include "quantization/vectorization_utils.cuh"
-#include "cub_helpers.h"
 #include <c10/cuda/CUDAGuard.h>
 #include <ATen/cuda/Exceptions.h>