Browse Source

Remove struct : cute::aligned_struct to avoid error with gcc 12

Tri Dao 7 tháng trước cách đây
mục cha
commit
3f6ff1c1c5
1 tập tin đã thay đổi với 2 bổ sung2 xóa
  1. 2 2
      hopper/flash_bwd_kernel.h

+ 2 - 2
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;