named_barrier.hpp 722 B

1234567891011121314151617181920212223
  1. /******************************************************************************
  2. * Copyright (c) 2024, Jay Shah, Ganesh Bikshandi, Ying Zhang, Vijay Thakkar, Pradeep Ramani, Tri Dao.
  3. ******************************************************************************/
  4. #pragma once
  5. #include "cutlass/arch/barrier.h"
  6. namespace flash {
  7. ////////////////////////////////////////////////////////////////////////////////////////////////////
  8. // Enumerates the reserved named barriers to avoid potential conflicts
  9. enum class FwdNamedBarriers {
  10. QueryEmpty = 0,
  11. ValueEmpty = 1,
  12. TileCountSmemEmpty = 2,
  13. TileCountSmemFull = 3,
  14. WarpSchedulerWG1 = 4,
  15. WarpSchedulerWG2 = 5,
  16. WarpSchedulerWG3 = 6,
  17. };
  18. } // flash