From 3f6ff1c1c52fa3d148b502e465ffb7bc88f7a50e Mon Sep 17 00:00:00 2001 From: Tri Dao Date: Fri, 2 Aug 2024 00:59:35 -0700 Subject: [PATCH] Remove struct : cute::aligned_struct to avoid error with gcc 12 --- hopper/flash_bwd_kernel.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hopper/flash_bwd_kernel.h b/hopper/flash_bwd_kernel.h index d6f9f7e..f4ba0ff 100644 --- a/hopper/flash_bwd_kernel.h +++ b/hopper/flash_bwd_kernel.h @@ -72,14 +72,14 @@ public: // Kernel level shared memory storage struct SharedStorage { - struct : cute::aligned_struct<1024> { + struct { union { typename CollectiveMainloop::TensorStorage mainloop; typename CollectiveEpilogue::TensorStorage epilogue; }; }; - struct : cute::aligned_struct<16> { + struct { alignas(16) cutlass::arch::ClusterTransactionBarrier barrier_KV; alignas(16) cutlass::arch::ClusterBarrier barrier_dKV; alignas(16) typename CollectiveMainloop::MainloopPipeline::SharedStorage pipeline_q;