Skip to content

Commit 300fcba

Browse files
committed
Fix CuTe header dependency order
1 parent c39a2ce commit 300fcba

1 file changed

Lines changed: 9 additions & 5 deletions

File tree

src/kernels/direct_cute_gemm.cu

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -37,16 +37,20 @@
3737
#include <cstdint>
3838
#include <limits>
3939

40-
#include "cute/algorithm/cooperative_copy.hpp"
41-
#include "cute/arch/cluster_sm90.hpp"
42-
#include "cute/arch/tmem_allocator_sm100.hpp"
43-
#include "cute/numeric/integral_constant.hpp"
44-
#include "cute/tensor.hpp"
4540
#include "cutlass/arch/barrier.h"
4641
#include "cutlass/bfloat16.h"
4742
#include "cutlass/cluster_launch.hpp"
4843
#include "cutlass/cutlass.h"
4944

45+
// Keep the CuTe include order aligned with the SM100 tutorials. Several
46+
// algorithm headers consume Tensor and Copy_Atom declarations but do not own
47+
// those declarations themselves.
48+
#include "cute/tensor.hpp"
49+
#include "cute/arch/cluster_sm90.hpp"
50+
#include "cute/numeric/integral_constant.hpp"
51+
#include "cute/algorithm/cooperative_copy.hpp"
52+
#include "cute/arch/tmem_allocator_sm100.hpp"
53+
5054
namespace blackwell_moe {
5155
namespace {
5256

0 commit comments

Comments
 (0)