42 lines
1.0 KiB
C++
42 lines
1.0 KiB
C++
/******************************************************************************
|
|
* Copyright (c) 2024, Jay Shah, Ganesh Bikshandi, Ying Zhang, Vijay Thakkar, Pradeep Ramani, Tri Dao.
|
|
******************************************************************************/
|
|
|
|
#pragma once
|
|
|
|
#include "cutlass/arch/barrier.h"
|
|
|
|
namespace flash {
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Enumerates the reserved named barriers to avoid potential conflicts
|
|
|
|
enum class FwdNamedBarriers {
|
|
QueryEmpty = 0,
|
|
ValueEmpty = 1,
|
|
TileCountSmemEmpty = 2,
|
|
TileCountSmemFull = 3,
|
|
WarpSchedulerWG1 = 4,
|
|
WarpSchedulerWG2 = 5,
|
|
WarpSchedulerWG3 = 6,
|
|
ProducerWG = 7
|
|
};
|
|
|
|
enum class BwdNamedBarriers {
|
|
QueryEmpty = 0,
|
|
KVEmpty = 1,
|
|
TileCountSmemEmpty = 2,
|
|
TileCountSmemFull = 3,
|
|
// WarpSchedulerWG1 = 4,
|
|
// WarpSchedulerWG2 = 5,
|
|
dQEmptyWG1 = 4,
|
|
dQEmptyWG2 = 5,
|
|
dSFull = 6,
|
|
// dSEmptyWG1 = 7,
|
|
// dSEmptyWG2 = 8,
|
|
dQEmpty = 7,
|
|
dQFull = 8,
|
|
};
|
|
|
|
} // flash
|