biondizzle/nvfp4-megamoe-kernel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix device_memory include path

Browse Source
This commit is contained in:
biondizzle
2026-05-13 23:27:06 +00:00
parent 3299d22ad6
commit b7c5cba407
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/cutlass_nvfp4_gemm.cu
View File

@@ -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)