From b7c5cba4076820be18a138b9f7a839bb8f3818cd Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Wed, 13 May 2026 23:27:06 +0000
Subject: [PATCH] Fix device_memory include path

---
 .../cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu                    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu b/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu
index f005411f..d559cda8 100644
--- a/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu
+++ b/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu
@@ -20,7 +20,7 @@
 #include <cutlass/gemm/kernel/tile_scheduler_params.h>
 #include <cutlass/float_subbyte.h>
 #include <cutlass/util/packed_stride.hpp>
-#include <cutlass/device_memory.h>
+#include <cutlass/util/device_memory.h>
 
 #if defined(CUTLASS_ARCH_MMA_SM100_SUPPORTED)