mirror of
https://github.com/deepseek-ai/FlashMLA
synced 2025-06-26 18:15:54 +00:00
17 lines
337 B
C++
17 lines
337 B
C++
#pragma once
|
|
|
|
#include "cutlass/barrier.h"
|
|
|
|
namespace flash {
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Enumerates the reserved named barriers to avoid potential conflicts
|
|
|
|
enum class NamedBarriers {
|
|
SReady = 1,
|
|
SoftmaxReady = 2,
|
|
TransVReady = 3,
|
|
};
|
|
|
|
} // flash
|